com.arangodb.internal.ArangoEdgeCollectionAsyncImpl 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.20.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.ArangoEdgeCollectionAsync;
import com.arangodb.ArangoGraphAsync;
import com.arangodb.entity.EdgeEntity;
import com.arangodb.entity.EdgeUpdateEntity;
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 ArangoEdgeCollectionAsyncImpl extends InternalArangoEdgeCollection implements ArangoEdgeCollectionAsync {

    private final ArangoGraphAsync graph;

    protected ArangoEdgeCollectionAsyncImpl(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 EdgeCollectionDropOptions());
    }

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

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

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

    @Override
    public CompletableFuture insertEdge(final Object value) {
        return executorAsync().execute(() -> insertEdgeRequest(value, new EdgeCreateOptions()),
                insertEdgeResponseDeserializer());
    }

    @Override
    public CompletableFuture insertEdge(final Object value, final EdgeCreateOptions options) {
        return executorAsync().execute(() -> insertEdgeRequest(value, options), insertEdgeResponseDeserializer());
    }

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

    @Override
    public  CompletableFuture getEdge(final String key, final Class type, final GraphDocumentReadOptions options) {
        return executorAsync().execute(() -> getEdgeRequest(key, options), getEdgeResponseDeserializer(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 replaceEdge(final String key, final Object value) {
        return executorAsync().execute(() -> replaceEdgeRequest(key, value, new EdgeReplaceOptions()),
                replaceEdgeResponseDeserializer());
    }

    @Override
    public CompletableFuture replaceEdge(final String key, final Object value, final EdgeReplaceOptions options) {
        return executorAsync().execute(() -> replaceEdgeRequest(key, value, options), replaceEdgeResponseDeserializer());
    }

    @Override
    public CompletableFuture updateEdge(final String key, final Object value) {
        return executorAsync().execute(() -> updateEdgeRequest(key, value, new EdgeUpdateOptions()),
                updateEdgeResponseDeserializer());
    }

    @Override
    public CompletableFuture updateEdge(final String key, final Object value, final EdgeUpdateOptions options) {
        return executorAsync().execute(() -> updateEdgeRequest(key, value, options), updateEdgeResponseDeserializer());
    }

    @Override
    public CompletableFuture deleteEdge(final String key) {
        return executorAsync().execute(() -> deleteEdgeRequest(key, new EdgeDeleteOptions()), Void.class);
    }

    @Override
    public CompletableFuture deleteEdge(final String key, final EdgeDeleteOptions options) {
        return executorAsync().execute(() -> deleteEdgeRequest(key, options), Void.class);
    }

}