com.arangodb.impl.InternalDocumentDriverImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver Show documentation
ArangoDB Java Driver
There is a newer version: 7.15.0
/*
 * Copyright (C) 2012 tamtam180
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.arangodb.impl;

import java.util.List;
import java.util.ListIterator;
import java.util.Locale;

import com.arangodb.ArangoConfigure;
import com.arangodb.ArangoException;
import com.arangodb.entity.*;
import com.arangodb.http.HttpManager;
import com.arangodb.http.HttpResponseEntity;
import com.arangodb.util.CollectionUtils;
import com.arangodb.util.MapBuilder;
import com.google.gson.JsonElement;

/**
 * @author tamtam180 - kirscheless at gmail.com
 *
 */
public class InternalDocumentDriverImpl extends BaseArangoDriverImpl implements com.arangodb.InternalDocumentDriver {

  InternalDocumentDriverImpl(ArangoConfigure configure, HttpManager httpManager) {
    super(configure , httpManager);
  }

  private  DocumentEntity _createDocument(String database, String collectionName, String documentKey, Object value, Boolean createCollection, Boolean waitForSync, boolean raw) throws ArangoException {
    
    validateCollectionName(collectionName);
    
    String body;
    if (raw) {
      body = value.toString();
    } else if (documentKey != null) {
      JsonElement elem = EntityFactory.toJsonElement(value, false);
      if (elem.isJsonObject()) {
        elem.getAsJsonObject().addProperty("_key", documentKey);
      }
      body = EntityFactory.toJsonString(elem);
    } else {
      body = EntityFactory.toJsonString(value);
    }
    
    HttpResponseEntity res = httpManager.doPost(
        createEndpointUrl(baseUrl, database, "/_api/document"), 
        new MapBuilder()
          .put("collection", collectionName)
          .put("createCollection", createCollection)
          .put("waitForSync", waitForSync)
          .get(),
          body);
    
    return createEntity(res, DocumentEntity.class);
    
  }

  @Override
  public  DocumentEntity createDocument(String database, String collectionName, String documentKey, Object value, Boolean createCollection, Boolean waitForSync) throws ArangoException {
    return _createDocument(database, collectionName, documentKey, value, createCollection, waitForSync, false);
  }

  @Override
  public  DocumentEntity createDocumentRaw(String database, String collectionName, String documentKey, String rawJsonString, Boolean createCollection, Boolean waitForSync) throws ArangoException {
    return _createDocument(database, collectionName, documentKey, rawJsonString, createCollection, waitForSync, true);
  }

  @Override
  public  DocumentEntity replaceDocument(String database, String documentHandle, Object value, Long rev, Policy policy, Boolean waitForSync) throws ArangoException {
    
    validateDocumentHandle(documentHandle);
    HttpResponseEntity res = httpManager.doPut(
        createEndpointUrl(baseUrl, database, "/_api/document", documentHandle), 
        new MapBuilder()
          .put("rev", rev)
          .put("policy", policy == null ? null : policy.name())
          .put("waitForSync", waitForSync)
          .get(),
        EntityFactory.toJsonString(value));
    
    return createEntity(res, DocumentEntity.class);
    
  }

  @Override
  public  DocumentEntity updateDocument(String database, String documentHandle, Object value, Long rev, Policy policy, Boolean waitForSync, Boolean keepNull) throws ArangoException {
    
    validateDocumentHandle(documentHandle);
    HttpResponseEntity res = httpManager.doPatch(
        createEndpointUrl(baseUrl, database, "/_api/document", documentHandle), 
        new MapBuilder()
          .put("rev", rev)
          .put("policy", policy == null ? null : policy.name())
          .put("waitForSync", waitForSync)
          .put("keepNull", keepNull)
          .get(),
        EntityFactory.toJsonString(value, keepNull != null && !keepNull));
    
    DocumentEntity entity = createEntity(res, DocumentEntity.class);
    return entity;
    
  }

  private static final String API_DOCUMENT_PREFIX = "/_api/document/";
  @Override
  public List getDocuments(String database, String collectionName, boolean handleConvert) throws ArangoException {
    
    HttpResponseEntity res = httpManager.doGet(
        createEndpointUrl(baseUrl, database, "/_api/document"), 
        new MapBuilder("collection", collectionName).get()
        );
    
    DocumentsEntity entity = createEntity(res, DocumentsEntity.class);
    List documents = CollectionUtils.safety(entity.getDocuments());
    
    if (handleConvert && !documents.isEmpty()) {
      ListIterator lit = documents.listIterator();
      while (lit.hasNext()) {
        String d = lit.next();
        if (d.startsWith(API_DOCUMENT_PREFIX)) {
          lit.set(d.substring(API_DOCUMENT_PREFIX.length()));
        }
      }
    }
    return documents;
  }
  
  @Override
  public long checkDocument(String database, String documentHandle) throws ArangoException {
    validateDocumentHandle(documentHandle);
    HttpResponseEntity res = httpManager.doHead(
        createEndpointUrl(baseUrl, database, "/_api/document", documentHandle),
        null
        );
    
    DefaultEntity entity = createEntity(res, DefaultEntity.class);
    return entity.getEtag();
    
  }

  @Override
  public  DocumentEntity getDocument(
    String database, 
		String documentHandle, 
		Class clazz, 
		Long ifNoneMatchRevision, 
		Long ifMatchRevision) throws ArangoException {
    
    validateDocumentHandle(documentHandle);
	  HttpResponseEntity res = httpManager.doGet(
			  createEndpointUrl(baseUrl, database, "/_api/document", documentHandle),
        new MapBuilder().put("If-None-Match", ifNoneMatchRevision, true).put("If-Match", ifMatchRevision).get(),
        null);
    DocumentEntity entity = createEntity(res, DocumentEntity.class, clazz);
    if (entity == null) {
      entity = new DocumentEntity();
    }
    return entity;
  }

  @Override
  public DocumentEntity deleteDocument(String database, String documentHandle, Long rev, Policy policy) throws ArangoException {
    
    validateDocumentHandle(documentHandle);
    HttpResponseEntity res = httpManager.doDelete(
        createEndpointUrl(baseUrl, database, "/_api/document", documentHandle), 
        new MapBuilder()
        .put("rev", rev)
        .put("policy", policy == null ? null : policy.name().toLowerCase(Locale.US))
        .get());
    
    try {
      DocumentEntity entity = createEntity(res, DocumentEntity.class);
      return entity;
    } catch (ArangoException e) {
      throw e;
    }
    
  }

}