graphql.execution.instrumentation.dataloader.DataLoaderDispatcherInstrumentationState Maven / Gradle / Ivy

Go to download
package graphql.execution.instrumentation.dataloader;

import org.dataloader.DataLoaderRegistry;
import org.slf4j.Logger;

import graphql.execution.instrumentation.InstrumentationState;

/**
 * A base class that keeps track of whether aggressive batching can be used
 */
public class DataLoaderDispatcherInstrumentationState implements InstrumentationState {

    private final FieldLevelTrackingApproach approach;

    private final DataLoaderRegistry dataLoaderRegistry;

    private final InstrumentationState state;

    private boolean aggressivelyBatching = true;

    public DataLoaderDispatcherInstrumentationState(Logger log, DataLoaderRegistry dataLoaderRegistry) {

        this.dataLoaderRegistry = dataLoaderRegistry;
        this.approach = new FieldLevelTrackingApproach(log, dataLoaderRegistry);
        this.state = approach.createState();
    }

    boolean isAggressivelyBatching() {
        return aggressivelyBatching;
    }

    void setAggressivelyBatching(boolean aggressivelyBatching) {
        this.aggressivelyBatching = aggressivelyBatching;
    }

    FieldLevelTrackingApproach getApproach() {
        return approach;
    }

    DataLoaderRegistry getDataLoaderRegistry() {
        return dataLoaderRegistry;
    }

    InstrumentationState getState() {
        return state;
    }
    
    

}