All Downloads are FREE. Search and download functionalities are using the official Maven repository.

graphql.execution.instrumentation.SimpleInstrumentationContext Maven / Gradle / Ivy

Go to download

GraphqL Java

There is a newer version: 230521-nf-execution
Show newest version
package graphql.execution.instrumentation;

import graphql.PublicApi;

import java.util.concurrent.CompletableFuture;
import java.util.function.BiConsumer;
import java.util.function.Consumer;

/**
 * A simple implementation of {@link InstrumentationContext}
 */
@PublicApi
public class SimpleInstrumentationContext implements InstrumentationContext {

    private static final InstrumentationContext NO_OP = new SimpleInstrumentationContext<>();

    /**
     * A context that does nothing
     *
     * @param  the type needed
     *
     * @return a context that does nothing
     */
    @SuppressWarnings("unchecked")
    public static  InstrumentationContext noOp() {
        return (InstrumentationContext) NO_OP;
    }

    private final BiConsumer codeToRunOnComplete;
    private final Consumer> codeToRunOnDispatch;

    public SimpleInstrumentationContext() {
        this(null, null);
    }

    private SimpleInstrumentationContext(Consumer> codeToRunOnDispatch, BiConsumer codeToRunOnComplete) {
        this.codeToRunOnComplete = codeToRunOnComplete;
        this.codeToRunOnDispatch = codeToRunOnDispatch;
    }

    @Override
    public void onDispatched(CompletableFuture result) {
        if (codeToRunOnDispatch != null) {
            codeToRunOnDispatch.accept(result);
        }
    }

    @Override
    public void onCompleted(T result, Throwable t) {
        if (codeToRunOnComplete != null) {
            codeToRunOnComplete.accept(result, t);
        }
    }

    /**
     * Allows for the more fluent away to return an instrumentation context that runs the specified
     * code on instrumentation step dispatch.
     *
     * @param codeToRun the code to run on dispatch
     * @param        the generic type
     *
     * @return an instrumentation context
     */
    public static  SimpleInstrumentationContext whenDispatched(Consumer> codeToRun) {
        return new SimpleInstrumentationContext<>(codeToRun, null);
    }

    /**
     * Allows for the more fluent away to return an instrumentation context that runs the specified
     * code on instrumentation step completion.
     *
     * @param codeToRun the code to run on completion
     * @param        the generic type
     *
     * @return an instrumentation context
     */
    public static  SimpleInstrumentationContext whenCompleted(BiConsumer codeToRun) {
        return new SimpleInstrumentationContext<>(null, codeToRun);
    }

    public static  BiConsumer completeInstrumentationCtxCF(
            InstrumentationContext instrumentationContext, CompletableFuture targetCF) {
        return (result, throwable) -> {
            if (throwable != null) {
                targetCF.completeExceptionally(throwable);
            } else {
                targetCF.complete(result);
            }
            instrumentationContext.onCompleted(result, throwable);
        };
    }

}
    

    

    
            
    
            







    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy