com.arangodb.internal.ArangoCollectionImpl Maven / Gradle / Ivy
/*
* DISCLAIMER
*
* Copyright 2018 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 com.arangodb.ArangoCollection;
import com.arangodb.ArangoDBException;
import com.arangodb.entity.*;
import com.arangodb.internal.util.DocumentUtil;
import com.arangodb.model.*;
import com.arangodb.util.RawData;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Collection;
import static com.arangodb.internal.serde.SerdeUtils.constructParametricType;
/**
* @author Mark Vollmary
* @author Michele Rastelli
*/
public class ArangoCollectionImpl extends InternalArangoCollection
implements ArangoCollection {
private static final Logger LOGGER = LoggerFactory.getLogger(ArangoCollectionImpl.class);
protected ArangoCollectionImpl(final ArangoDatabaseImpl db, final String name) {
super(db, name);
}
@Override
public DocumentCreateEntity insertDocument(final Object value) {
return executor.execute(insertDocumentRequest(value, new DocumentCreateOptions()),
constructParametricType(DocumentCreateEntity.class, Void.class));
}
@Override
@SuppressWarnings("unchecked")
public DocumentCreateEntity insertDocument(final T value, final DocumentCreateOptions options) {
return insertDocument(value, options, (Class) value.getClass());
}
@Override
public DocumentCreateEntity insertDocument(final T value, final DocumentCreateOptions options,
final Class type) {
return executor.execute(insertDocumentRequest(value, options),
constructParametricType(DocumentCreateEntity.class, type));
}
@Override
public MultiDocumentEntity> insertDocuments(RawData values) {
return executor
.execute(insertDocumentsRequest(values, new DocumentCreateOptions()),
insertDocumentsResponseDeserializer(Void.class));
}
@Override
@SuppressWarnings("unchecked")
public MultiDocumentEntity> insertDocuments(RawData values,
DocumentCreateOptions options) {
return executor
.execute(insertDocumentsRequest(values, options),
insertDocumentsResponseDeserializer((Class) values.getClass()));
}
@Override
public MultiDocumentEntity> insertDocuments(final Collection values) {
return executor
.execute(insertDocumentsRequest(values, new DocumentCreateOptions()),
insertDocumentsResponseDeserializer(Void.class));
}
@Override
@SuppressWarnings("unchecked")
public MultiDocumentEntity> insertDocuments(
final Collection values, final DocumentCreateOptions options) {
return insertDocuments(values, options, (Class) getCollectionContentClass(values));
}
@Override
public MultiDocumentEntity> insertDocuments(Collection values,
DocumentCreateOptions options,
Class type) {
return executor
.execute(insertDocumentsRequest(values, options), insertDocumentsResponseDeserializer(type));
}
@Override
public DocumentImportEntity importDocuments(final Collection values) {
return importDocuments(values, new DocumentImportOptions());
}
@Override
public DocumentImportEntity importDocuments(final Collection values, final DocumentImportOptions options) {
return executor.execute(importDocumentsRequest(values, options), DocumentImportEntity.class);
}
@Override
public DocumentImportEntity importDocuments(RawData values) {
return importDocuments(values, new DocumentImportOptions());
}
@Override
public DocumentImportEntity importDocuments(RawData values, DocumentImportOptions options) {
return executor.execute(importDocumentsRequest(values, options), DocumentImportEntity.class);
}
@Override
public T getDocument(final String key, final Class type) {
return getDocument(key, type, new DocumentReadOptions());
}
@Override
public T getDocument(final String key, final Class type, final DocumentReadOptions options) {
DocumentUtil.validateDocumentKey(key);
try {
return executor.execute(getDocumentRequest(key, options), getDocumentResponseDeserializer(type));
} catch (final ArangoDBException e) {
if (LOGGER.isDebugEnabled()) {
LOGGER.debug(e.getMessage(), e);
}
// handle Response: 404, Error: 1655 - transaction not found
if (e.getErrorNum() != null && e.getErrorNum() == 1655) {
throw e;
}
if ((e.getResponseCode() != null && (e.getResponseCode() == 404 || e.getResponseCode() == 304
|| e.getResponseCode() == 412))) {
return null;
}
throw e;
}
}
@Override
public MultiDocumentEntity getDocuments(final Collection keys, final Class type) {
return getDocuments(keys, type, new DocumentReadOptions());
}
@Override
public MultiDocumentEntity getDocuments(
final Collection keys, final Class type, final DocumentReadOptions options) {
return executor.execute(getDocumentsRequest(keys, options), getDocumentsResponseDeserializer(type));
}
@Override
public DocumentUpdateEntity replaceDocument(final String key, final Object value) {
return executor.execute(replaceDocumentRequest(key, value, new DocumentReplaceOptions()),
constructParametricType(DocumentUpdateEntity.class, Void.class));
}
@Override
@SuppressWarnings("unchecked")
public DocumentUpdateEntity replaceDocument(
final String key, final T value, final DocumentReplaceOptions options) {
return replaceDocument(key, value, options, (Class) value.getClass());
}
@Override
public DocumentUpdateEntity replaceDocument(String key, T value, DocumentReplaceOptions options,
Class type) {
return executor.execute(replaceDocumentRequest(key, value, options),
constructParametricType(DocumentUpdateEntity.class, type));
}
@Override
public MultiDocumentEntity> replaceDocuments(RawData values) {
return executor.execute(replaceDocumentsRequest(values, new DocumentReplaceOptions()),
replaceDocumentsResponseDeserializer(Void.class));
}
@Override
@SuppressWarnings("unchecked")
public MultiDocumentEntity> replaceDocuments(RawData values,
DocumentReplaceOptions options) {
return executor.execute(replaceDocumentsRequest(values, options),
replaceDocumentsResponseDeserializer((Class) values.getClass()));
}
@Override
public MultiDocumentEntity> replaceDocuments(final Collection values) {
return executor.execute(replaceDocumentsRequest(values, new DocumentReplaceOptions()),
replaceDocumentsResponseDeserializer(Void.class));
}
@Override
@SuppressWarnings("unchecked")
public MultiDocumentEntity> replaceDocuments(
final Collection values, final DocumentReplaceOptions options) {
return replaceDocuments(values, options, (Class) getCollectionContentClass(values));
}
@Override
public MultiDocumentEntity> replaceDocuments(Collection values,
DocumentReplaceOptions options,
Class type) {
return executor.execute(replaceDocumentsRequest(values, options), replaceDocumentsResponseDeserializer(type));
}
@Override
public DocumentUpdateEntity updateDocument(final String key, final Object value) {
return updateDocument(key, value, new DocumentUpdateOptions(), Void.class);
}
@Override
@SuppressWarnings("unchecked")
public DocumentUpdateEntity updateDocument(
final String key, final T value, final DocumentUpdateOptions options) {
return updateDocument(key, value, options, (Class) value.getClass());
}
@Override
public DocumentUpdateEntity updateDocument(
final String key, final Object value, final DocumentUpdateOptions options, final Class returnType) {
return executor.execute(updateDocumentRequest(key, value, options),
constructParametricType(DocumentUpdateEntity.class, returnType));
}
@Override
public MultiDocumentEntity> updateDocuments(RawData values) {
return executor
.execute(updateDocumentsRequest(values, new DocumentUpdateOptions()),
updateDocumentsResponseDeserializer(Void.class));
}
@Override
@SuppressWarnings("unchecked")
public MultiDocumentEntity> updateDocuments(RawData values,
DocumentUpdateOptions options) {
return executor
.execute(updateDocumentsRequest(values, options),
updateDocumentsResponseDeserializer((Class) values.getClass()));
}
@Override
public MultiDocumentEntity> updateDocuments(final Collection values) {
return updateDocuments(values, new DocumentUpdateOptions(), Void.class);
}
@Override
@SuppressWarnings("unchecked")
public MultiDocumentEntity> updateDocuments(
final Collection values, final DocumentUpdateOptions options) {
return updateDocuments(values, options, (Class) getCollectionContentClass(values));
}
@Override
public MultiDocumentEntity> updateDocuments(
final Collection values, final DocumentUpdateOptions options, final Class returnType) {
return executor
.execute(updateDocumentsRequest(values, options), updateDocumentsResponseDeserializer(returnType));
}
@Override
public DocumentDeleteEntity deleteDocument(final String key) {
return deleteDocument(key, new DocumentDeleteOptions());
}
@Override
public DocumentDeleteEntity deleteDocument(String key, DocumentDeleteOptions options) {
return deleteDocument(key, options, Void.class);
}
@Override
public DocumentDeleteEntity deleteDocument(
final String key, final DocumentDeleteOptions options, final Class type) {
return executor.execute(deleteDocumentRequest(key, options),
constructParametricType(DocumentDeleteEntity.class, type));
}
@Override
public MultiDocumentEntity> deleteDocuments(RawData values) {
return executor.execute(deleteDocumentsRequest(values, new DocumentDeleteOptions()),
deleteDocumentsResponseDeserializer(Void.class));
}
@Override
@SuppressWarnings("unchecked")
public MultiDocumentEntity> deleteDocuments(RawData values,
DocumentDeleteOptions options) {
return executor.execute(deleteDocumentsRequest(values, options),
deleteDocumentsResponseDeserializer((Class) values.getClass()));
}
@Override
public MultiDocumentEntity> deleteDocuments(final Collection values) {
return deleteDocuments(values, new DocumentDeleteOptions(), Void.class);
}
@Override
@SuppressWarnings("unchecked")
public MultiDocumentEntity> deleteDocuments(
final Collection values, final DocumentDeleteOptions options) {
return deleteDocuments(values, options, (Class) getCollectionContentClass(values));
}
@Override
public MultiDocumentEntity> deleteDocuments(
final Collection values, final DocumentDeleteOptions options, final Class type) {
return executor.execute(deleteDocumentsRequest(values, options), deleteDocumentsResponseDeserializer(type));
}
@Override
public Boolean documentExists(final String key) {
return documentExists(key, new DocumentExistsOptions());
}
@Override
public Boolean documentExists(final String key, final DocumentExistsOptions options) {
try {
executor.execute(documentExistsRequest(key, options), Void.class);
return true;
} catch (final ArangoDBException e) {
// handle Response: 404, Error: 1655 - transaction not found
if (e.getErrorNum() != null && e.getErrorNum() == 1655) {
throw e;
}
if ((e.getResponseCode() != null &&
(e.getResponseCode() == 404 || e.getResponseCode() == 304 || e.getResponseCode() == 412))) {
return false;
}
throw e;
}
}
@Override
public IndexEntity getIndex(final String id) {
return executor.execute(getIndexRequest(id), IndexEntity.class);
}
@Override
public InvertedIndexEntity getInvertedIndex(String id) {
return executor.execute(getIndexRequest(id), InvertedIndexEntity.class);
}
@Override
public String deleteIndex(final String id) {
return executor.execute(deleteIndexRequest(id), deleteIndexResponseDeserializer());
}
@Deprecated
@Override
public IndexEntity ensureHashIndex(final Iterable fields, final HashIndexOptions options) {
return executor.execute(createHashIndexRequest(fields, options), IndexEntity.class);
}
@Deprecated
@Override
public IndexEntity ensureSkiplistIndex(final Iterable fields, final SkiplistIndexOptions options) {
return executor.execute(createSkiplistIndexRequest(fields, options), IndexEntity.class);
}
@Override
public IndexEntity ensurePersistentIndex(final Iterable fields, final PersistentIndexOptions options) {
return executor.execute(createPersistentIndexRequest(fields, options), IndexEntity.class);
}
@Override
public InvertedIndexEntity ensureInvertedIndex(final InvertedIndexOptions options) {
return executor.execute(createInvertedIndexRequest(options), InvertedIndexEntity.class);
}
@Override
public IndexEntity ensureGeoIndex(final Iterable fields, final GeoIndexOptions options) {
return executor.execute(createGeoIndexRequest(fields, options), IndexEntity.class);
}
@Deprecated
@Override
public IndexEntity ensureFulltextIndex(final Iterable fields, final FulltextIndexOptions options) {
return executor.execute(createFulltextIndexRequest(fields, options), IndexEntity.class);
}
@Override
public IndexEntity ensureTtlIndex(final Iterable fields, final TtlIndexOptions options) {
return executor.execute(createTtlIndexRequest(fields, options), IndexEntity.class);
}
@Override
public IndexEntity ensureZKDIndex(final Iterable fields, final ZKDIndexOptions options) {
return executor.execute(createZKDIndexRequest(fields, options), IndexEntity.class);
}
@Override
public Collection getIndexes() {
return executor.execute(getIndexesRequest(), getIndexesResponseDeserializer());
}
@Override
public Collection getInvertedIndexes() {
return executor.execute(getIndexesRequest(), getInvertedIndexesResponseDeserializer());
}
@Override
public boolean exists() {
try {
getInfo();
return true;
} catch (final ArangoDBException e) {
if (ArangoErrors.ERROR_ARANGO_DATA_SOURCE_NOT_FOUND.equals(e.getErrorNum())) {
return false;
}
throw e;
}
}
@Override
public CollectionEntity truncate() {
return truncate(null);
}
@Override
public CollectionEntity truncate(CollectionTruncateOptions options) {
return executor.execute(truncateRequest(options), CollectionEntity.class);
}
@Override
public CollectionPropertiesEntity count() {
return count(null);
}
@Override
public CollectionPropertiesEntity count(CollectionCountOptions options) {
return executor.execute(countRequest(options), CollectionPropertiesEntity.class);
}
@Override
public CollectionEntity create() {
return db().createCollection(name());
}
@Override
public CollectionEntity create(final CollectionCreateOptions options) {
return db().createCollection(name(), options);
}
@Override
public void drop() {
executor.execute(dropRequest(null), Void.class);
}
@Override
public void drop(final boolean isSystem) {
executor.execute(dropRequest(isSystem), Void.class);
}
@Override
public CollectionEntity getInfo() {
return executor.execute(getInfoRequest(), CollectionEntity.class);
}
@Override
public CollectionPropertiesEntity getProperties() {
return executor.execute(getPropertiesRequest(), CollectionPropertiesEntity.class);
}
@Override
public CollectionPropertiesEntity changeProperties(final CollectionPropertiesOptions options) {
return executor.execute(changePropertiesRequest(options), CollectionPropertiesEntity.class);
}
@Override
public synchronized CollectionEntity rename(final String newName) {
final CollectionEntity result = executor.execute(renameRequest(newName), CollectionEntity.class);
name = result.getName();
return result;
}
@Override
public ShardEntity getResponsibleShard(final Object value) {
return executor.execute(responsibleShardRequest(value), ShardEntity.class);
}
@Override
public CollectionRevisionEntity getRevision() {
return executor.execute(getRevisionRequest(), CollectionRevisionEntity.class);
}
@Override
public void grantAccess(final String user, final Permissions permissions) {
executor.execute(grantAccessRequest(user, permissions), Void.class);
}
@Override
public void revokeAccess(final String user) {
executor.execute(grantAccessRequest(user, Permissions.NONE), Void.class);
}
@Override
public void resetAccess(final String user) {
executor.execute(resetAccessRequest(user), Void.class);
}
@Override
public Permissions getPermissions(final String user) {
return executor.execute(getPermissionsRequest(user), getPermissionsResponseDeserialzer());
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy