com.arangodb.internal.ArangoVertexCollectionAsyncImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver-shaded Show documentation
ArangoDB Java Driver Shaded
There is a newer version: 7.17.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 com.arangodb.ArangoDBException;
import com.arangodb.ArangoGraphAsync;
import com.arangodb.ArangoVertexCollectionAsync;
import com.arangodb.entity.VertexEntity;
import com.arangodb.entity.VertexUpdateEntity;
import com.arangodb.model.*;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionException;

import static com.arangodb.internal.ArangoErrors.*;

/**
 * @author Mark Vollmary
 */
public class ArangoVertexCollectionAsyncImpl extends InternalArangoVertexCollection implements ArangoVertexCollectionAsync {

    private final ArangoGraphAsync graph;

    protected ArangoVertexCollectionAsyncImpl(final ArangoGraphAsyncImpl graph, final String name) {
        super(graph, graph.db().name(), graph.name(), name);
        this.graph = graph;
    }

    @Override
    public ArangoGraphAsync graph() {
        return graph;
    }

    @Deprecated
    @Override
    public CompletableFuture drop() {
        return drop(new VertexCollectionDropOptions());
    }

    @Deprecated
    @Override
    public CompletableFuture drop(final VertexCollectionDropOptions options) {
        return executorAsync().execute(() -> dropRequest(options), Void.class);
    }

    @Override
    public CompletableFuture remove() {
        return remove(new VertexCollectionRemoveOptions());
    }

    @Override
    public CompletableFuture remove(final VertexCollectionRemoveOptions options) {
        return executorAsync().execute(() -> removeVertexCollectionRequest(options), Void.class);
    }

    @Override
    public CompletableFuture insertVertex(final Object value) {
        return executorAsync().execute(() -> insertVertexRequest(value, new VertexCreateOptions()),
                insertVertexResponseDeserializer());
    }

    @Override
    public CompletableFuture insertVertex(final Object value, final VertexCreateOptions options) {
        return executorAsync().execute(() -> insertVertexRequest(value, options), insertVertexResponseDeserializer());
    }

    @Override
    public  CompletableFuture getVertex(final String key, final Class type) {
        return getVertex(key, type, null);
    }

    @Override
    public  CompletableFuture getVertex(final String key, final Class type, final GraphDocumentReadOptions options) {
        return executorAsync().execute(() -> getVertexRequest(key, options), getVertexResponseDeserializer(type))
                .exceptionally(err -> {
                    Throwable e = err instanceof CompletionException ? err.getCause() : err;
                    if (e instanceof ArangoDBException) {
                        ArangoDBException aEx = (ArangoDBException) e;
                        if (matches(aEx, 304)
                                || matches(aEx, 404, ERROR_ARANGO_DOCUMENT_NOT_FOUND)
                                || matches(aEx, 412, ERROR_ARANGO_CONFLICT)
                        ) {
                            return null;
                        }
                    }
                    throw ArangoDBException.of(e);
                });
    }

    @Override
    public CompletableFuture replaceVertex(final String key, final Object value) {
        return executorAsync().execute(() -> replaceVertexRequest(key, value, new VertexReplaceOptions()),
                replaceVertexResponseDeserializer());
    }

    @Override
    public CompletableFuture replaceVertex(final String key, final Object value, final VertexReplaceOptions options) {
        return executorAsync().execute(() -> replaceVertexRequest(key, value, options), replaceVertexResponseDeserializer());
    }

    @Override
    public CompletableFuture updateVertex(final String key, final Object value) {
        return executorAsync().execute(() -> updateVertexRequest(key, value, new VertexUpdateOptions()),
                updateVertexResponseDeserializer());
    }

    @Override
    public CompletableFuture updateVertex(final String key, final Object value, final VertexUpdateOptions options) {
        return executorAsync().execute(() -> updateVertexRequest(key, value, options), updateVertexResponseDeserializer());
    }

    @Override
    public CompletableFuture deleteVertex(final String key) {
        return executorAsync().execute(() -> deleteVertexRequest(key, new VertexDeleteOptions()), Void.class);
    }

    @Override
    public CompletableFuture deleteVertex(final String key, final VertexDeleteOptions options) {
        return executorAsync().execute(() -> deleteVertexRequest(key, options), Void.class);
    }

}