com.arangodb.internal.ArangoDatabaseImpl Maven / Gradle / Ivy
/*
* 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 com.arangodb.*;
import com.arangodb.entity.*;
import com.arangodb.entity.arangosearch.AnalyzerEntity;
import com.arangodb.internal.cursor.ArangoCursorImpl;
import com.arangodb.internal.net.HostHandle;
import com.arangodb.internal.util.DocumentUtil;
import com.arangodb.model.*;
import com.arangodb.model.arangosearch.AnalyzerDeleteOptions;
import com.arangodb.model.arangosearch.ArangoSearchCreateOptions;
import com.arangodb.util.ArangoCursorInitializer;
import com.arangodb.velocypack.Type;
import com.arangodb.velocystream.Request;
import java.util.Collection;
import java.util.Collections;
import java.util.Map;
/**
* @author Mark Vollmary
* @author Michele Rastelli
*/
public class ArangoDatabaseImpl extends InternalArangoDatabase
implements ArangoDatabase {
private ArangoCursorInitializer cursorInitializer;
protected ArangoDatabaseImpl(final ArangoDBImpl arangoDB, final String name) {
super(arangoDB, name);
}
@Override
public ArangoDBVersion getVersion() throws ArangoDBException {
return executor.execute(getVersionRequest(), ArangoDBVersion.class);
}
@Override
public ArangoDBEngine getEngine() throws ArangoDBException {
return executor.execute(getEngineRequest(), ArangoDBEngine.class);
}
@Override
public boolean exists() throws ArangoDBException {
try {
getInfo();
return true;
} catch (final ArangoDBException e) {
if (ArangoErrors.ERROR_ARANGO_DATABASE_NOT_FOUND.equals(e.getErrorNum())) {
return false;
}
throw e;
}
}
@Override
public Collection getAccessibleDatabases() throws ArangoDBException {
return executor.execute(getAccessibleDatabasesRequest(), getDatabaseResponseDeserializer());
}
@Override
public ArangoCollection collection(final String name) {
return new ArangoCollectionImpl(this, name);
}
@Override
public CollectionEntity createCollection(final String name) throws ArangoDBException {
return executor.execute(createCollectionRequest(name, new CollectionCreateOptions()), CollectionEntity.class);
}
@Override
public CollectionEntity createCollection(final String name, final CollectionCreateOptions options)
throws ArangoDBException {
return executor.execute(createCollectionRequest(name, options), CollectionEntity.class);
}
@Override
public Collection getCollections() throws ArangoDBException {
return executor
.execute(getCollectionsRequest(new CollectionsReadOptions()), getCollectionsResponseDeserializer());
}
@Override
public Collection getCollections(final CollectionsReadOptions options) throws ArangoDBException {
return executor.execute(getCollectionsRequest(options), getCollectionsResponseDeserializer());
}
@Override
public IndexEntity getIndex(final String id) throws ArangoDBException {
DocumentUtil.validateIndexId(id);
final String[] split = id.split("/");
return collection(split[0]).getIndex(split[1]);
}
@Override
public String deleteIndex(final String id) throws ArangoDBException {
DocumentUtil.validateIndexId(id);
final String[] split = id.split("/");
return collection(split[0]).deleteIndex(split[1]);
}
@Override
public Boolean create() throws ArangoDBException {
return arango().createDatabase(name());
}
@Override
public Boolean drop() throws ArangoDBException {
return executor.execute(dropRequest(), createDropResponseDeserializer());
}
@Override
public void grantAccess(final String user, final Permissions permissions) throws ArangoDBException {
executor.execute(grantAccessRequest(user, permissions), Void.class);
}
@Override
public void grantAccess(final String user) throws ArangoDBException {
executor.execute(grantAccessRequest(user, Permissions.RW), Void.class);
}
@Override
public void revokeAccess(final String user) throws ArangoDBException {
executor.execute(grantAccessRequest(user, Permissions.NONE), Void.class);
}
@Override
public void resetAccess(final String user) throws ArangoDBException {
executor.execute(resetAccessRequest(user), Void.class);
}
@Override
public void grantDefaultCollectionAccess(final String user, final Permissions permissions)
throws ArangoDBException {
executor.execute(updateUserDefaultCollectionAccessRequest(user, permissions), Void.class);
}
@Override
public Permissions getPermissions(final String user) throws ArangoDBException {
return executor.execute(getPermissionsRequest(user), getPermissionsResponseDeserialzer());
}
@Override
public ArangoCursor query(
final String query, final Map bindVars, final AqlQueryOptions options, final Class type)
throws ArangoDBException {
final Request request = queryRequest(query, bindVars, options);
final HostHandle hostHandle = new HostHandle();
final CursorEntity result = executor.execute(request, CursorEntity.class, hostHandle);
return createCursor(result, type, options, hostHandle);
}
@Override
public ArangoCursor query(
final String query, final Map bindVars, final Class type) throws ArangoDBException {
return query(query, bindVars, null, type);
}
@Override
public ArangoCursor query(final String query, final AqlQueryOptions options, final Class type)
throws ArangoDBException {
return query(query, null, options, type);
}
@Override
public ArangoCursor query(final String query, final Class type) throws ArangoDBException {
return query(query, null, null, type);
}
@Override
public ArangoCursor cursor(final String cursorId, final Class type) throws ArangoDBException {
final HostHandle hostHandle = new HostHandle();
final CursorEntity result = executor
.execute(queryNextRequest(cursorId, null, null), CursorEntity.class, hostHandle);
return createCursor(result, type, null, hostHandle);
}
private ArangoCursor createCursor(
final CursorEntity result,
final Class type,
final AqlQueryOptions options,
final HostHandle hostHandle) {
final ArangoCursorExecute execute = new ArangoCursorExecute() {
@Override
public CursorEntity next(final String id, Map meta) {
return executor.execute(queryNextRequest(id, options, meta), CursorEntity.class, hostHandle);
}
@Override
public void close(final String id, Map meta) {
executor.execute(queryCloseRequest(id, options, meta), Void.class, hostHandle);
}
};
return cursorInitializer != null ?
cursorInitializer.createInstance(this, execute, type, result) :
new ArangoCursorImpl<>(this, execute, type, result);
}
@Override
public AqlExecutionExplainEntity explainQuery(
final String query, final Map bindVars, final AqlQueryExplainOptions options)
throws ArangoDBException {
return executor.execute(explainQueryRequest(query, bindVars, options), AqlExecutionExplainEntity.class);
}
@Override
public AqlParseEntity parseQuery(final String query) throws ArangoDBException {
return executor.execute(parseQueryRequest(query), AqlParseEntity.class);
}
@Override
public void clearQueryCache() throws ArangoDBException {
executor.execute(clearQueryCacheRequest(), Void.class);
}
@Override
public QueryCachePropertiesEntity getQueryCacheProperties() throws ArangoDBException {
return executor.execute(getQueryCachePropertiesRequest(), QueryCachePropertiesEntity.class);
}
@Override
public QueryCachePropertiesEntity setQueryCacheProperties(final QueryCachePropertiesEntity properties)
throws ArangoDBException {
return executor.execute(setQueryCachePropertiesRequest(properties), QueryCachePropertiesEntity.class);
}
@Override
public QueryTrackingPropertiesEntity getQueryTrackingProperties() throws ArangoDBException {
return executor.execute(getQueryTrackingPropertiesRequest(), QueryTrackingPropertiesEntity.class);
}
@Override
public QueryTrackingPropertiesEntity setQueryTrackingProperties(final QueryTrackingPropertiesEntity properties)
throws ArangoDBException {
return executor.execute(setQueryTrackingPropertiesRequest(properties), QueryTrackingPropertiesEntity.class);
}
@Override
public Collection getCurrentlyRunningQueries() throws ArangoDBException {
return executor.execute(getCurrentlyRunningQueriesRequest(), new Type>() {
}.getType());
}
@Override
public Collection getSlowQueries() throws ArangoDBException {
return executor.execute(getSlowQueriesRequest(), new Type>() {
}.getType());
}
@Override
public void clearSlowQueries() throws ArangoDBException {
executor.execute(clearSlowQueriesRequest(), Void.class);
}
@Override
public void killQuery(final String id) throws ArangoDBException {
executor.execute(killQueryRequest(id), Void.class);
}
@Override
public void createAqlFunction(
final String name, final String code, final AqlFunctionCreateOptions options) throws ArangoDBException {
executor.execute(createAqlFunctionRequest(name, code, options), Void.class);
}
@Override
public Integer deleteAqlFunction(final String name, final AqlFunctionDeleteOptions options)
throws ArangoDBException {
return executor.execute(deleteAqlFunctionRequest(name, options), deleteAqlFunctionResponseDeserializer());
}
@Override
public Collection getAqlFunctions(final AqlFunctionGetOptions options) throws ArangoDBException {
return executor.execute(getAqlFunctionsRequest(options), getAqlFunctionsResponseDeserializer());
}
@Override
public ArangoGraph graph(final String name) {
return new ArangoGraphImpl(this, name);
}
@Override
public GraphEntity createGraph(final String name, final Collection edgeDefinitions)
throws ArangoDBException {
return executor.execute(createGraphRequest(name, edgeDefinitions, new GraphCreateOptions()),
createGraphResponseDeserializer());
}
@Override
public GraphEntity createGraph(
final String name, final Collection edgeDefinitions, final GraphCreateOptions options)
throws ArangoDBException {
return executor.execute(createGraphRequest(name, edgeDefinitions, options), createGraphResponseDeserializer());
}
@Override
public Collection getGraphs() throws ArangoDBException {
return executor.execute(getGraphsRequest(), getGraphsResponseDeserializer());
}
@Override
public T transaction(final String action, final Class type, final TransactionOptions options)
throws ArangoDBException {
return executor.execute(transactionRequest(action, options), transactionResponseDeserializer(type));
}
@Override
public StreamTransactionEntity beginStreamTransaction(StreamTransactionOptions options) throws ArangoDBException {
return executor.execute(beginStreamTransactionRequest(options), streamTransactionResponseDeserializer());
}
@Override
public StreamTransactionEntity abortStreamTransaction(String id) throws ArangoDBException {
return executor.execute(abortStreamTransactionRequest(id), streamTransactionResponseDeserializer());
}
@Override
public StreamTransactionEntity getStreamTransaction(String id) throws ArangoDBException {
return executor.execute(getStreamTransactionRequest(id), streamTransactionResponseDeserializer());
}
@Override
public Collection getStreamTransactions() throws ArangoDBException {
return executor.execute(getStreamTransactionsRequest(), transactionsResponseDeserializer());
}
@Override
public StreamTransactionEntity commitStreamTransaction(String id) throws ArangoDBException {
return executor.execute(commitStreamTransactionRequest(id), streamTransactionResponseDeserializer());
}
@Override
public DatabaseEntity getInfo() throws ArangoDBException {
return executor.execute(getInfoRequest(), getInfoResponseDeserializer());
}
@Override
public TraversalEntity executeTraversal(
final Class vertexClass, final Class edgeClass, final TraversalOptions options)
throws ArangoDBException {
final Request request = executeTraversalRequest(options);
return executor.execute(request, executeTraversalResponseDeserializer(vertexClass, edgeClass));
}
@Override
public T getDocument(final String id, final Class type) throws ArangoDBException {
DocumentUtil.validateDocumentId(id);
final String[] split = id.split("/");
return collection(split[0]).getDocument(split[1], type);
}
@Override
public T getDocument(final String id, final Class type, final DocumentReadOptions options)
throws ArangoDBException {
DocumentUtil.validateDocumentId(id);
final String[] split = id.split("/");
return collection(split[0]).getDocument(split[1], type, options);
}
@Override
public void reloadRouting() throws ArangoDBException {
executor.execute(reloadRoutingRequest(), Void.class);
}
protected ArangoDatabaseImpl setCursorInitializer(final ArangoCursorInitializer cursorInitializer) {
this.cursorInitializer = cursorInitializer;
return this;
}
@Override
public ArangoRoute route(final String... path) {
return new ArangoRouteImpl(this, createPath(path), Collections.emptyMap());
}
@Override
public Collection getViews() throws ArangoDBException {
return executor.execute(getViewsRequest(), getViewsResponseDeserializer());
}
@Override
public ArangoView view(final String name) {
return new ArangoViewImpl(this, name);
}
@Override
public ArangoSearch arangoSearch(final String name) {
return new ArangoSearchImpl(this, name);
}
@Override
public ViewEntity createView(final String name, final ViewType type) throws ArangoDBException {
return executor.execute(createViewRequest(name, type), ViewEntity.class);
}
@Override
public ViewEntity createArangoSearch(final String name, final ArangoSearchCreateOptions options)
throws ArangoDBException {
return executor.execute(createArangoSearchRequest(name, options), ViewEntity.class);
}
@Override
public AnalyzerEntity createAnalyzer(AnalyzerEntity options) throws ArangoDBException {
return executor.execute(createAnalyzerRequest(options), AnalyzerEntity.class);
}
@Override
public AnalyzerEntity getAnalyzer(String name) throws ArangoDBException {
return executor.execute(getAnalyzerRequest(name), AnalyzerEntity.class);
}
@Override
public Collection getAnalyzers() throws ArangoDBException {
return executor.execute(getAnalyzersRequest(), getAnalyzersResponseDeserializer());
}
@Override
public void deleteAnalyzer(String name) throws ArangoDBException {
executor.execute(deleteAnalyzerRequest(name, null), Void.class);
}
@Override
public void deleteAnalyzer(String name, AnalyzerDeleteOptions options) throws ArangoDBException {
executor.execute(deleteAnalyzerRequest(name, options), Void.class);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy