com.arangodb.internal.cursor.ArangoCursorAsyncImpl 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.18.0

Show newest version

package com.arangodb.internal.cursor;

import com.arangodb.ArangoCursorAsync;
import com.arangodb.ArangoDBException;
import com.arangodb.entity.CursorEntity;
import com.arangodb.internal.ArangoDatabaseAsyncImpl;
import com.arangodb.internal.InternalArangoCursor;
import com.arangodb.internal.net.HostHandle;

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

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

public class ArangoCursorAsyncImpl extends InternalArangoCursor implements ArangoCursorAsync {

    private final ArangoDatabaseAsyncImpl db;
    private final HostHandle hostHandle;
    private final CursorEntity entity;

    public ArangoCursorAsyncImpl(
            final ArangoDatabaseAsyncImpl db,
            final CursorEntity entity,
            final Class type,
            final HostHandle hostHandle,
            final Boolean allowRetry
    ) {
        super(db, db.name(), entity, type, allowRetry);
        this.db = db;
        this.hostHandle = hostHandle;
        this.entity = entity;
    }

    @Override
    public CompletableFuture> nextBatch() {
        if (Boolean.TRUE.equals(hasMore())) {
            return executorAsync().execute(this::queryNextRequest, db.cursorEntityDeserializer(getType()), hostHandle)
                    .thenApply(r -> {
                        // needed because the latest batch does not return the cursor id
                        r.setId(entity.getId());
                        return new ArangoCursorAsyncImpl<>(db, r, getType(), hostHandle, allowRetry());
                    });
        } else {
            CompletableFuture> cf = new CompletableFuture<>();
            cf.completeExceptionally(new NoSuchElementException());
            return cf;
        }
    }

    @Override
    public CompletableFuture close() {
        if (getId() != null && (allowRetry() || Boolean.TRUE.equals(hasMore()))) {
            return executorAsync().execute(this::queryCloseRequest, Void.class, hostHandle)
                    .exceptionally(err -> {
                        Throwable e = err instanceof CompletionException ? err.getCause() : err;
                        if (e instanceof ArangoDBException) {
                            ArangoDBException aEx = (ArangoDBException) e;
                            // ignore errors Response: 404, Error: 1600 - cursor not found
                            if (matches(aEx, 404, 1600)) {
                                return null;
                            }
                        }
                        throw ArangoDBException.of(e);
                    })
                    .thenApply(__ -> null);
        } else {
            return CompletableFuture.completedFuture(null);
        }
    }

}