graphql.execution.defer.DeferredCall Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of graphql-java Show documentation

GraphqL Java

There is a newer version: 230521-nf-execution

package graphql.execution.defer;

import graphql.ExecutionResult;
import graphql.ExecutionResultImpl;
import graphql.GraphQLError;
import graphql.Internal;

import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.function.Supplier;

/**
 * This represents a deferred call (aka @defer) to get an execution result sometime after
 * the initial query has returned
 */
@Internal
public class DeferredCall {
    private final Supplier> call;
    private final DeferredErrorSupport errorSupport;

    public DeferredCall(Supplier> call, DeferredErrorSupport deferredErrorSupport) {
        this.call = call;
        this.errorSupport = deferredErrorSupport;
    }

    CompletableFuture invoke() {
        CompletableFuture future = call.get();
        return future.thenApply(this::addErrorsEncountered);
    }

    private ExecutionResult addErrorsEncountered(ExecutionResult executionResult) {
        List errorsEncountered = errorSupport.getErrors();
        if (errorsEncountered.isEmpty()) {
            return executionResult;
        }
        ExecutionResultImpl sourceResult = (ExecutionResultImpl) executionResult;
        ExecutionResultImpl.Builder builder = ExecutionResultImpl.newExecutionResult().from(sourceResult);
        builder.addErrors(errorsEncountered);
        return builder.build();
    }

}