cc.voox.graphql.GraphQLContextUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of graphql-spring-boot-starter Show documentation

graphql tool for java

The newest version!

package cc.voox.graphql;

import graphql.schema.DataFetcher;
import graphql.schema.DataFetchingEnvironment;
import org.dataloader.DataLoader;

import java.util.Map;

public final class GraphQLContextUtil {
    private static ThreadLocal threadLocal = new ThreadLocal<>();

    protected final static void add(Object value) {
        threadLocal.set(value);
    }

    public final static DataFetchingEnvironment get() {
        return (DataFetchingEnvironment) threadLocal.get();
    }

    public final static DataLoader getDataLoader(String key) {
        return isFromGraphQL()? get().getDataLoader(key) : null;
    }

    public final static  DataLoader getDataLoader(Class dataLoaderClass) {
        return isFromGraphQL()? get().getDataLoader(dataLoaderClass.getSimpleName()) : null;
    }

    public final static  T getSource(Class src) {
        return (T) get().getSource();
    }

    public final static Map getArguments() {
        return get().getArguments();
    }

    protected final static void clear() {
        threadLocal.remove();
    }

    public final static boolean isFromGraphQL() {
        return get() != null;
    }
}