cc.voox.graphql.EnhanceGraphQLInvocation 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

There is a newer version: 1.2.1

Show newest version

package cc.voox.graphql;

import graphql.ExecutionInput;
import graphql.ExecutionResult;
import graphql.GraphQL;
import graphql.GraphQLContext;
import graphql.spring.web.servlet.GraphQLInvocationData;
import graphql.spring.web.servlet.components.DefaultGraphQLInvocation;
import org.dataloader.BatchLoader;
import org.dataloader.DataLoader;
import org.dataloader.DataLoaderRegistry;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Primary;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.WebRequest;

import java.util.ArrayList;
import java.util.concurrent.CompletableFuture;

@Primary
@Component
public class EnhanceGraphQLInvocation extends DefaultGraphQLInvocation {
    @Autowired
    private GraphQL graphQL;

    @Autowired
    private GraphqlResolverFactory graphqlResolverFactory;

    @Override
    public CompletableFuture invoke(GraphQLInvocationData invocationData, WebRequest webRequest) {
        DataLoaderRegistry registry = new DataLoaderRegistry();
        graphqlResolverFactory.getDataLoaders().forEach(registry::register);

        ExecutionInput executionInput = ExecutionInput.newExecutionInput().query(invocationData.getQuery()).operationName(invocationData.getOperationName()).variables(invocationData.getVariables())
                .dataLoaderRegistry(registry)
                .context(webRequest)
                .build();
        return this.graphQL.executeAsync(executionInput);
    }
}