All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.arangodb.internal.InternalArangoCollection Maven / Gradle / Ivy

    There is a newer version: 7.15.0
    Show newest version
    /*
 * DISCLAIMER
 *
 * Copyright 2016 ArangoDB GmbH, Cologne, Germany
 *
 * 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.
 *
 * Copyright holder is ArangoDB GmbH, Cologne, Germany
 */

package com.arangodb.internal;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import com.arangodb.ArangoDBException;
import com.arangodb.entity.DocumentCreateEntity;
import com.arangodb.entity.DocumentDeleteEntity;
import com.arangodb.entity.DocumentField;
import com.arangodb.entity.DocumentUpdateEntity;
import com.arangodb.entity.ErrorEntity;
import com.arangodb.entity.IndexEntity;
import com.arangodb.entity.MultiDocumentEntity;
import com.arangodb.entity.Permissions;
import com.arangodb.internal.ArangoExecutor.ResponseDeserializer;
import com.arangodb.internal.util.ArangoSerializationFactory.Serializer;
import com.arangodb.internal.util.DocumentUtil;
import com.arangodb.model.CollectionPropertiesOptions;
import com.arangodb.model.CollectionRenameOptions;
import com.arangodb.model.DocumentCreateOptions;
import com.arangodb.model.DocumentDeleteOptions;
import com.arangodb.model.DocumentExistsOptions;
import com.arangodb.model.DocumentImportOptions;
import com.arangodb.model.DocumentReadOptions;
import com.arangodb.model.DocumentReplaceOptions;
import com.arangodb.model.DocumentUpdateOptions;
import com.arangodb.model.FulltextIndexOptions;
import com.arangodb.model.GeoIndexOptions;
import com.arangodb.model.HashIndexOptions;
import com.arangodb.model.ImportType;
import com.arangodb.model.OptionsBuilder;
import com.arangodb.model.PersistentIndexOptions;
import com.arangodb.model.SkiplistIndexOptions;
import com.arangodb.model.UserAccessOptions;
import com.arangodb.util.ArangoSerializer;
import com.arangodb.velocypack.Type;
import com.arangodb.velocypack.VPackSlice;
import com.arangodb.velocypack.exception.VPackException;
import com.arangodb.velocystream.Request;
import com.arangodb.velocystream.RequestType;
import com.arangodb.velocystream.Response;

/**
 * @author Mark Vollmary
 *
 */
public abstract class InternalArangoCollection, D extends InternalArangoDatabase, E extends ArangoExecutor>
		extends ArangoExecuteable {

	private static final String COLLECTION = "collection";

	protected static final String PATH_API_COLLECTION = "/_api/collection";
	private static final String PATH_API_DOCUMENT = "/_api/document";
	private static final String PATH_API_INDEX = "/_api/index";
	private static final String PATH_API_IMPORT = "/_api/import";
	private static final String PATH_API_USER = "/_api/user";

	private static final String MERGE_OBJECTS = "mergeObjects";
	private static final String IGNORE_REVS = "ignoreRevs";
	private static final String RETURN_NEW = "returnNew";
	private static final String NEW = "new";
	private static final String RETURN_OLD = "returnOld";
	private static final String OLD = "old";
	private static final String SILENT = "silent";

	private final D db;
	protected volatile String name;

	protected InternalArangoCollection(final D db, final String name) {
		super(db.executor, db.util, db.context);
		this.db = db;
		this.name = name;
	}

	public D db() {
		return db;
	}

	public String name() {
		return name;
	}

	protected  Request insertDocumentRequest(final T value, final DocumentCreateOptions options) {
		final Request request = request(db.name(), RequestType.POST, PATH_API_DOCUMENT, name);
		final DocumentCreateOptions params = (options != null ? options : new DocumentCreateOptions());
		request.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync());
		request.putQueryParam(RETURN_NEW, params.getReturnNew());
		request.putQueryParam(RETURN_OLD, params.getReturnOld());
		request.putQueryParam(SILENT, params.getSilent());
		request.putQueryParam("overwrite", params.getOverwrite());
		request.setBody(util(Serializer.CUSTOM).serialize(value));
		return request;
	}

	protected  ResponseDeserializer> insertDocumentResponseDeserializer(
		final T value,
		final DocumentCreateOptions options) {
		return new ResponseDeserializer>() {
			@SuppressWarnings("unchecked")
			@Override
			public DocumentCreateEntity deserialize(final Response response) throws VPackException {
				final VPackSlice body = response.getBody();
				final DocumentCreateEntity doc = util().deserialize(body, DocumentCreateEntity.class);
				final VPackSlice newDoc = body.get(NEW);
				if (newDoc.isObject()) {
					doc.setNew((T) util(Serializer.CUSTOM).deserialize(newDoc, value.getClass()));
				}
				final VPackSlice oldDoc = body.get(OLD);
				if (oldDoc.isObject()) {
					doc.setOld((T) util(Serializer.CUSTOM).deserialize(oldDoc, value.getClass()));
				}
				if (options == null || Boolean.TRUE != options.getSilent()) {
					final Map values = new HashMap();
					values.put(DocumentField.Type.ID, doc.getId());
					values.put(DocumentField.Type.KEY, doc.getKey());
					values.put(DocumentField.Type.REV, doc.getRev());
					executor.documentCache().setValues(value, values);
				}
				return doc;
			}
		};
	}

	protected  Request insertDocumentsRequest(final Collection values, final DocumentCreateOptions params) {
		final Request request = request(db.name(), RequestType.POST, PATH_API_DOCUMENT, name);
		request.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync());
		request.putQueryParam(RETURN_NEW, params.getReturnNew());
		request.putQueryParam(SILENT, params.getSilent());
		request.setBody(util(Serializer.CUSTOM).serialize(values,
			new ArangoSerializer.Options().serializeNullValues(false).stringAsJson(true)));
		return request;
	}

	@SuppressWarnings("unchecked")
	protected  ResponseDeserializer>> insertDocumentsResponseDeserializer(
		final Collection values,
		final DocumentCreateOptions params) {
		return new ResponseDeserializer>>() {
			@Override
			public MultiDocumentEntity> deserialize(final Response response)
					throws VPackException {
				Class type = null;
				if (Boolean.TRUE == params.getReturnNew()) {
					if (!values.isEmpty()) {
						type = (Class) values.iterator().next().getClass();
					}
				}
				final MultiDocumentEntity> multiDocument = new MultiDocumentEntity>();
				final Collection> docs = new ArrayList>();
				final Collection errors = new ArrayList();
				final Collection documentsAndErrors = new ArrayList();
				final VPackSlice body = response.getBody();
				for (final Iterator iterator = body.arrayIterator(); iterator.hasNext();) {
					final VPackSlice next = iterator.next();
					if (next.get(ArangoResponseField.ERROR).isTrue()) {
						final ErrorEntity error = (ErrorEntity) util().deserialize(next, ErrorEntity.class);
						errors.add(error);
						documentsAndErrors.add(error);
					} else {
						final DocumentCreateEntity doc = util().deserialize(next, DocumentCreateEntity.class);
						final VPackSlice newDoc = next.get(NEW);
						if (newDoc.isObject()) {
							doc.setNew((T) util(Serializer.CUSTOM).deserialize(newDoc, type));
						}
						docs.add(doc);
						documentsAndErrors.add(doc);
					}
				}
				multiDocument.setDocuments(docs);
				multiDocument.setErrors(errors);
				multiDocument.setDocumentsAndErrors(documentsAndErrors);
				return multiDocument;
			}
		};
	}

	protected Request importDocumentsRequest(final String values, final DocumentImportOptions options) {
		return importDocumentsRequest(options).putQueryParam("type", ImportType.auto).setBody(util().serialize(values));
	}

	protected Request importDocumentsRequest(final Collection values, final DocumentImportOptions options) {
		return importDocumentsRequest(options).putQueryParam("type", ImportType.list).setBody(util(Serializer.CUSTOM)
				.serialize(values, new ArangoSerializer.Options().serializeNullValues(false).stringAsJson(true)));
	}

	protected Request importDocumentsRequest(final DocumentImportOptions options) {
		final DocumentImportOptions params = options != null ? options : new DocumentImportOptions();
		return request(db.name(), RequestType.POST, PATH_API_IMPORT).putQueryParam(COLLECTION, name)
				.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync())
				.putQueryParam("fromPrefix", params.getFromPrefix()).putQueryParam("toPrefix", params.getToPrefix())
				.putQueryParam("overwrite", params.getOverwrite()).putQueryParam("onDuplicate", params.getOnDuplicate())
				.putQueryParam("complete", params.getComplete()).putQueryParam("details", params.getDetails());
	}

	protected Request getDocumentRequest(final String key, final DocumentReadOptions options) {
		final Request request = request(db.name(), RequestType.GET, PATH_API_DOCUMENT,
			DocumentUtil.createDocumentHandle(name, key));
		final DocumentReadOptions params = (options != null ? options : new DocumentReadOptions());
		request.putHeaderParam(ArangoRequestParam.IF_NONE_MATCH, params.getIfNoneMatch());
		request.putHeaderParam(ArangoRequestParam.IF_MATCH, params.getIfMatch());
		return request;
	}

	protected Request getDocumentsRequest(final Collection keys, final DocumentReadOptions options) {
		return request(db.name(), RequestType.PUT, PATH_API_DOCUMENT, name).putQueryParam("onlyget", true)
				.putHeaderParam(ArangoRequestParam.IF_NONE_MATCH, options.getIfNoneMatch())
				.putHeaderParam(ArangoRequestParam.IF_MATCH, options.getIfMatch()).setBody(util().serialize(keys));
	}

	protected  ResponseDeserializer> getDocumentsResponseDeserializer(
		final Class type,
		final DocumentReadOptions options) {
		return new ResponseDeserializer>() {
			@SuppressWarnings("unchecked")
			@Override
			public MultiDocumentEntity deserialize(final Response response) throws VPackException {
				final MultiDocumentEntity multiDocument = new MultiDocumentEntity();
				final Collection docs = new ArrayList();
				final Collection errors = new ArrayList();
				final Collection documentsAndErrors = new ArrayList();
				final VPackSlice body = response.getBody();
				for (final Iterator iterator = body.arrayIterator(); iterator.hasNext();) {
					final VPackSlice next = iterator.next();
					if (next.get(ArangoResponseField.ERROR).isTrue()) {
						final ErrorEntity error = (ErrorEntity) util().deserialize(next, ErrorEntity.class);
						errors.add(error);
						documentsAndErrors.add(error);
					} else {
						final T doc = (T) util(Serializer.CUSTOM).deserialize(next, type);
						docs.add(doc);
						documentsAndErrors.add(doc);
					}
				}
				multiDocument.setDocuments(docs);
				multiDocument.setErrors(errors);
				multiDocument.setDocumentsAndErrors(documentsAndErrors);
				return multiDocument;
			}
		};
	}

	protected  Request replaceDocumentRequest(
		final String key,
		final T value,
		final DocumentReplaceOptions options) {
		final Request request = request(db.name(), RequestType.PUT, PATH_API_DOCUMENT,
			DocumentUtil.createDocumentHandle(name, key));
		final DocumentReplaceOptions params = (options != null ? options : new DocumentReplaceOptions());
		request.putHeaderParam(ArangoRequestParam.IF_MATCH, params.getIfMatch());
		request.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync());
		request.putQueryParam(IGNORE_REVS, params.getIgnoreRevs());
		request.putQueryParam(RETURN_NEW, params.getReturnNew());
		request.putQueryParam(RETURN_OLD, params.getReturnOld());
		request.putQueryParam(SILENT, params.getSilent());
		request.setBody(util(Serializer.CUSTOM).serialize(value));
		return request;
	}

	protected  ResponseDeserializer> replaceDocumentResponseDeserializer(
		final T value,
		final DocumentReplaceOptions options) {
		return new ResponseDeserializer>() {
			@SuppressWarnings("unchecked")
			@Override
			public DocumentUpdateEntity deserialize(final Response response) throws VPackException {
				final VPackSlice body = response.getBody();
				final DocumentUpdateEntity doc = util().deserialize(body, DocumentUpdateEntity.class);
				final VPackSlice newDoc = body.get(NEW);
				if (newDoc.isObject()) {
					doc.setNew((T) util(Serializer.CUSTOM).deserialize(newDoc, value.getClass()));
				}
				final VPackSlice oldDoc = body.get(OLD);
				if (oldDoc.isObject()) {
					doc.setOld((T) util(Serializer.CUSTOM).deserialize(oldDoc, value.getClass()));
				}
				if (options == null || Boolean.TRUE != options.getSilent()) {
					final Map values = new HashMap();
					values.put(DocumentField.Type.REV, doc.getRev());
					executor.documentCache().setValues(value, values);
				}
				return doc;
			}
		};
	}

	protected  Request replaceDocumentsRequest(final Collection values, final DocumentReplaceOptions params) {
		final Request request = request(db.name(), RequestType.PUT, PATH_API_DOCUMENT, name);
		request.putHeaderParam(ArangoRequestParam.IF_MATCH, params.getIfMatch());
		request.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync());
		request.putQueryParam(IGNORE_REVS, params.getIgnoreRevs());
		request.putQueryParam(RETURN_NEW, params.getReturnNew());
		request.putQueryParam(RETURN_OLD, params.getReturnOld());
		request.putQueryParam(SILENT, params.getSilent());
		request.setBody(util(Serializer.CUSTOM).serialize(values,
			new ArangoSerializer.Options().serializeNullValues(false).stringAsJson(true)));
		return request;
	}

	@SuppressWarnings("unchecked")
	protected  ResponseDeserializer>> replaceDocumentsResponseDeserializer(
		final Collection values,
		final DocumentReplaceOptions params) {
		return new ResponseDeserializer>>() {
			@Override
			public MultiDocumentEntity> deserialize(final Response response)
					throws VPackException {
				Class type = null;
				if (Boolean.TRUE == params.getReturnNew() || Boolean.TRUE == params.getReturnOld()) {
					if (!values.isEmpty()) {
						type = (Class) values.iterator().next().getClass();
					}
				}
				final MultiDocumentEntity> multiDocument = new MultiDocumentEntity>();
				final Collection> docs = new ArrayList>();
				final Collection errors = new ArrayList();
				final Collection documentsAndErrors = new ArrayList();
				final VPackSlice body = response.getBody();
				for (final Iterator iterator = body.arrayIterator(); iterator.hasNext();) {
					final VPackSlice next = iterator.next();
					if (next.get(ArangoResponseField.ERROR).isTrue()) {
						final ErrorEntity error = (ErrorEntity) util().deserialize(next, ErrorEntity.class);
						errors.add(error);
						documentsAndErrors.add(error);
					} else {
						final DocumentUpdateEntity doc = util().deserialize(next, DocumentUpdateEntity.class);
						final VPackSlice newDoc = next.get(NEW);
						if (newDoc.isObject()) {
							doc.setNew((T) util(Serializer.CUSTOM).deserialize(newDoc, type));
						}
						final VPackSlice oldDoc = next.get(OLD);
						if (oldDoc.isObject()) {
							doc.setOld((T) util(Serializer.CUSTOM).deserialize(oldDoc, type));
						}
						docs.add(doc);
						documentsAndErrors.add(doc);
					}
				}
				multiDocument.setDocuments(docs);
				multiDocument.setErrors(errors);
				multiDocument.setDocumentsAndErrors(documentsAndErrors);
				return multiDocument;
			}
		};
	}

	protected  Request updateDocumentRequest(final String key, final T value, final DocumentUpdateOptions options) {
		final Request request = request(db.name(), RequestType.PATCH, PATH_API_DOCUMENT,
			DocumentUtil.createDocumentHandle(name, key));
		final DocumentUpdateOptions params = (options != null ? options : new DocumentUpdateOptions());
		request.putHeaderParam(ArangoRequestParam.IF_MATCH, params.getIfMatch());
		request.putQueryParam(ArangoRequestParam.KEEP_NULL, params.getKeepNull());
		request.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync());
		request.putQueryParam(MERGE_OBJECTS, params.getMergeObjects());
		request.putQueryParam(IGNORE_REVS, params.getIgnoreRevs());
		request.putQueryParam(RETURN_NEW, params.getReturnNew());
		request.putQueryParam(RETURN_OLD, params.getReturnOld());
		request.putQueryParam(SILENT, params.getSilent());
		request.setBody(util(Serializer.CUSTOM).serialize(value, new ArangoSerializer.Options()
				.serializeNullValues(params.getSerializeNull() == null || params.getSerializeNull())));
		return request;
	}

	protected  ResponseDeserializer> updateDocumentResponseDeserializer(
		final T value,
		final DocumentUpdateOptions options) {
		return new ResponseDeserializer>() {
			@SuppressWarnings("unchecked")
			@Override
			public DocumentUpdateEntity deserialize(final Response response) throws VPackException {
				final VPackSlice body = response.getBody();
				final DocumentUpdateEntity doc = util().deserialize(body, DocumentUpdateEntity.class);
				final VPackSlice newDoc = body.get(NEW);
				if (newDoc.isObject()) {
					doc.setNew((T) util(Serializer.CUSTOM).deserialize(newDoc, value.getClass()));
				}
				final VPackSlice oldDoc = body.get(OLD);
				if (oldDoc.isObject()) {
					doc.setOld((T) util(Serializer.CUSTOM).deserialize(oldDoc, value.getClass()));
				}
				if (options == null || Boolean.TRUE != options.getSilent()) {
					final Map values = new HashMap();
					values.put(DocumentField.Type.REV, doc.getRev());
					executor.documentCache().setValues(value, values);
				}
				return doc;
			}
		};
	}

	protected  Request updateDocumentsRequest(final Collection values, final DocumentUpdateOptions params) {
		final Request request = request(db.name(), RequestType.PATCH, PATH_API_DOCUMENT, name);
		final Boolean keepNull = params.getKeepNull();
		request.putHeaderParam(ArangoRequestParam.IF_MATCH, params.getIfMatch());
		request.putQueryParam(ArangoRequestParam.KEEP_NULL, keepNull);
		request.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync());
		request.putQueryParam(MERGE_OBJECTS, params.getMergeObjects());
		request.putQueryParam(IGNORE_REVS, params.getIgnoreRevs());
		request.putQueryParam(RETURN_NEW, params.getReturnNew());
		request.putQueryParam(RETURN_OLD, params.getReturnOld());
		request.putQueryParam(SILENT, params.getSilent());
		request.setBody(util(Serializer.CUSTOM).serialize(values,
			new ArangoSerializer.Options()
					.serializeNullValues(params.getSerializeNull() == null || params.getSerializeNull())
					.stringAsJson(true)));
		return request;
	}

	@SuppressWarnings("unchecked")
	protected  ResponseDeserializer>> updateDocumentsResponseDeserializer(
		final Collection values,
		final DocumentUpdateOptions params) {
		return new ResponseDeserializer>>() {
			@Override
			public MultiDocumentEntity> deserialize(final Response response)
					throws VPackException {
				Class type = null;
				if (Boolean.TRUE == params.getReturnNew() || Boolean.TRUE == params.getReturnOld()) {
					if (!values.isEmpty()) {
						type = (Class) values.iterator().next().getClass();
					}
				}
				final MultiDocumentEntity> multiDocument = new MultiDocumentEntity>();
				final Collection> docs = new ArrayList>();
				final Collection errors = new ArrayList();
				final Collection documentsAndErrors = new ArrayList();
				final VPackSlice body = response.getBody();
				for (final Iterator iterator = body.arrayIterator(); iterator.hasNext();) {
					final VPackSlice next = iterator.next();
					if (next.get(ArangoResponseField.ERROR).isTrue()) {
						final ErrorEntity error = (ErrorEntity) util().deserialize(next, ErrorEntity.class);
						errors.add(error);
						documentsAndErrors.add(error);
					} else {
						final DocumentUpdateEntity doc = util().deserialize(next, DocumentUpdateEntity.class);
						final VPackSlice newDoc = next.get(NEW);
						if (newDoc.isObject()) {
							doc.setNew((T) util(Serializer.CUSTOM).deserialize(newDoc, type));
						}
						final VPackSlice oldDoc = next.get(OLD);
						if (oldDoc.isObject()) {
							doc.setOld((T) util(Serializer.CUSTOM).deserialize(oldDoc, type));
						}
						docs.add(doc);
						documentsAndErrors.add(doc);
					}
				}
				multiDocument.setDocuments(docs);
				multiDocument.setErrors(errors);
				multiDocument.setDocumentsAndErrors(documentsAndErrors);
				return multiDocument;
			}
		};
	}

	protected Request deleteDocumentRequest(final String key, final DocumentDeleteOptions options) {
		final Request request = request(db.name(), RequestType.DELETE, PATH_API_DOCUMENT,
			DocumentUtil.createDocumentHandle(name, key));
		final DocumentDeleteOptions params = (options != null ? options : new DocumentDeleteOptions());
		request.putHeaderParam(ArangoRequestParam.IF_MATCH, params.getIfMatch());
		request.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync());
		request.putQueryParam(RETURN_OLD, params.getReturnOld());
		request.putQueryParam(SILENT, params.getSilent());
		return request;
	}

	protected  ResponseDeserializer> deleteDocumentResponseDeserializer(
		final Class type) {
		return new ResponseDeserializer>() {
			@SuppressWarnings("unchecked")
			@Override
			public DocumentDeleteEntity deserialize(final Response response) throws VPackException {
				final VPackSlice body = response.getBody();
				final DocumentDeleteEntity doc = util().deserialize(body, DocumentDeleteEntity.class);
				final VPackSlice oldDoc = body.get(OLD);
				if (oldDoc.isObject()) {
					doc.setOld((T) util(Serializer.CUSTOM).deserialize(oldDoc, type));
				}
				return doc;
			}
		};
	}

	protected  Request deleteDocumentsRequest(final Collection keys, final DocumentDeleteOptions options) {
		final Request request = request(db.name(), RequestType.DELETE, PATH_API_DOCUMENT, name);
		final DocumentDeleteOptions params = (options != null ? options : new DocumentDeleteOptions());
		request.putQueryParam(ArangoRequestParam.WAIT_FOR_SYNC, params.getWaitForSync());
		request.putQueryParam(RETURN_OLD, params.getReturnOld());
		request.putQueryParam(SILENT, params.getSilent());
		request.setBody(util().serialize(keys));
		return request;
	}

	protected  ResponseDeserializer>> deleteDocumentsResponseDeserializer(
		final Class type) {
		return new ResponseDeserializer>>() {
			@SuppressWarnings("unchecked")
			@Override
			public MultiDocumentEntity> deserialize(final Response response)
					throws VPackException {
				final MultiDocumentEntity> multiDocument = new MultiDocumentEntity>();
				final Collection> docs = new ArrayList>();
				final Collection errors = new ArrayList();
				final Collection documentsAndErrors = new ArrayList();
				final VPackSlice body = response.getBody();
				for (final Iterator iterator = body.arrayIterator(); iterator.hasNext();) {
					final VPackSlice next = iterator.next();
					if (next.get(ArangoResponseField.ERROR).isTrue()) {
						final ErrorEntity error = (ErrorEntity) util().deserialize(next, ErrorEntity.class);
						errors.add(error);
						documentsAndErrors.add(error);
					} else {
						final DocumentDeleteEntity doc = util().deserialize(next, DocumentDeleteEntity.class);
						final VPackSlice oldDoc = next.get(OLD);
						if (oldDoc.isObject()) {
							doc.setOld((T) util(Serializer.CUSTOM).deserialize(oldDoc, type));
						}
						docs.add(doc);
						documentsAndErrors.add(doc);
					}
				}
				multiDocument.setDocuments(docs);
				multiDocument.setErrors(errors);
				multiDocument.setDocumentsAndErrors(documentsAndErrors);
				return multiDocument;
			}
		};
	}

	protected Request documentExistsRequest(final String key, final DocumentExistsOptions options) {
		final Request request = request(db.name(), RequestType.HEAD, PATH_API_DOCUMENT,
			DocumentUtil.createDocumentHandle(name, key));
		final DocumentExistsOptions params = (options != null ? options : new DocumentExistsOptions());
		request.putHeaderParam(ArangoRequestParam.IF_MATCH, params.getIfMatch());
		request.putHeaderParam(ArangoRequestParam.IF_NONE_MATCH, params.getIfNoneMatch());
		return request;
	}

	protected Request getIndexRequest(final String id) {
		return request(db.name(), RequestType.GET, PATH_API_INDEX, createIndexId(id));
	}

	protected Request deleteIndexRequest(final String id) {
		return request(db.name(), RequestType.DELETE, PATH_API_INDEX, createIndexId(id));
	}

	protected ResponseDeserializer deleteIndexResponseDeserializer() {
		return new ResponseDeserializer() {
			@Override
			public String deserialize(final Response response) throws VPackException {
				return response.getBody().get("id").getAsString();
			}
		};
	}

	private String createIndexId(final String id) {
		final String index;
		if (id.matches(DocumentUtil.REGEX_ID)) {
			index = id;
		} else if (id.matches(DocumentUtil.REGEX_KEY)) {
			index = name + "/" + id;
		} else {
			throw new ArangoDBException(String.format("index id %s is not valid.", id));
		}
		return index;
	}

	protected Request createHashIndexRequest(final Iterable fields, final HashIndexOptions options) {
		final Request request = request(db.name(), RequestType.POST, PATH_API_INDEX);
		request.putQueryParam(COLLECTION, name);
		request.setBody(
			util().serialize(OptionsBuilder.build(options != null ? options : new HashIndexOptions(), fields)));
		return request;
	}

	protected Request createSkiplistIndexRequest(final Iterable fields, final SkiplistIndexOptions options) {
		final Request request = request(db.name(), RequestType.POST, PATH_API_INDEX);
		request.putQueryParam(COLLECTION, name);
		request.setBody(
			util().serialize(OptionsBuilder.build(options != null ? options : new SkiplistIndexOptions(), fields)));
		return request;
	}

	protected Request createPersistentIndexRequest(
		final Iterable fields,
		final PersistentIndexOptions options) {
		final Request request = request(db.name(), RequestType.POST, PATH_API_INDEX);
		request.putQueryParam(COLLECTION, name);
		request.setBody(
			util().serialize(OptionsBuilder.build(options != null ? options : new PersistentIndexOptions(), fields)));
		return request;
	}

	protected Request createGeoIndexRequest(final Iterable fields, final GeoIndexOptions options) {
		final Request request = request(db.name(), RequestType.POST, PATH_API_INDEX);
		request.putQueryParam(COLLECTION, name);
		request.setBody(
			util().serialize(OptionsBuilder.build(options != null ? options : new GeoIndexOptions(), fields)));
		return request;
	}

	protected Request createFulltextIndexRequest(final Iterable fields, final FulltextIndexOptions options) {
		final Request request = request(db.name(), RequestType.POST, PATH_API_INDEX);
		request.putQueryParam(COLLECTION, name);
		request.setBody(
			util().serialize(OptionsBuilder.build(options != null ? options : new FulltextIndexOptions(), fields)));
		return request;
	}

	protected Request getIndexesRequest() {
		final Request request = request(db.name(), RequestType.GET, PATH_API_INDEX);
		request.putQueryParam(COLLECTION, name);
		return request;
	}

	protected ResponseDeserializer> getIndexesResponseDeserializer() {
		return new ResponseDeserializer>() {
			@Override
			public Collection deserialize(final Response response) throws VPackException {
				return util().deserialize(response.getBody().get("indexes"), new Type>() {
				}.getType());
			}
		};
	}

	protected Request truncateRequest() {
		return request(db.name(), RequestType.PUT, PATH_API_COLLECTION, name, "truncate");
	}

	protected Request countRequest() {
		return request(db.name(), RequestType.GET, PATH_API_COLLECTION, name, "count");
	}

	protected Request dropRequest(final Boolean isSystem) {
		return request(db.name(), RequestType.DELETE, PATH_API_COLLECTION, name).putQueryParam("isSystem", isSystem);
	}

	protected Request loadRequest() {
		return request(db.name(), RequestType.PUT, PATH_API_COLLECTION, name, "load");
	}

	protected Request unloadRequest() {
		return request(db.name(), RequestType.PUT, PATH_API_COLLECTION, name, "unload");
	}

	protected Request getInfoRequest() {
		return request(db.name(), RequestType.GET, PATH_API_COLLECTION, name);
	}

	protected Request getPropertiesRequest() {
		return request(db.name(), RequestType.GET, PATH_API_COLLECTION, name, "properties");
	}

	protected Request changePropertiesRequest(final CollectionPropertiesOptions options) {
		final Request request = request(db.name(), RequestType.PUT, PATH_API_COLLECTION, name, "properties");
		request.setBody(util().serialize(options != null ? options : new CollectionPropertiesOptions()));
		return request;
	}

	protected Request renameRequest(final String newName) {
		final Request request = request(db.name(), RequestType.PUT, PATH_API_COLLECTION, name, "rename");
		request.setBody(util().serialize(OptionsBuilder.build(new CollectionRenameOptions(), newName)));
		return request;
	}

	protected Request getRevisionRequest() {
		return request(db.name(), RequestType.GET, PATH_API_COLLECTION, name, "revision");
	}

	protected Request grantAccessRequest(final String user, final Permissions permissions) {
		return request(ArangoRequestParam.SYSTEM, RequestType.PUT, PATH_API_USER, user, ArangoRequestParam.DATABASE,
			db.name(), name).setBody(util().serialize(OptionsBuilder.build(new UserAccessOptions(), permissions)));
	}

	protected Request resetAccessRequest(final String user) {
		return request(ArangoRequestParam.SYSTEM, RequestType.DELETE, PATH_API_USER, user, ArangoRequestParam.DATABASE,
			db.name(), name);
	}

	protected Request getPermissionsRequest(final String user) {
		return request(ArangoRequestParam.SYSTEM, RequestType.GET, PATH_API_USER, user, ArangoRequestParam.DATABASE,
			db.name(), name);
	}

	protected ResponseDeserializer getPermissionsResponseDeserialzer() {
		return new ResponseDeserializer() {
			@Override
			public Permissions deserialize(final Response response) throws VPackException {
				final VPackSlice body = response.getBody();
				if (body != null) {
					final VPackSlice result = body.get(ArangoResponseField.RESULT);
					if (!result.isNone()) {
						return util().deserialize(result, Permissions.class);
					}
				}
				return null;
			}
		};
	}

}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy