Как установить контекст из заголовка запроса, используя graphql-java - PullRequest
Новая
       22

Как установить контекст из заголовка запроса, используя graphql-java

1 голос
/ 11 июня 2019

Я хочу установить контекстную переменную из заголовка http-запроса, который я получаю из запроса. Это будет токен jwt, чтобы я мог идентифицировать своего пользователя по каждому запросу. 

package br.com.b2breservas.api;

import com.google.common.base.Charsets;
import com.google.common.io.Resources;
import graphql.GraphQL;
import graphql.schema.GraphQLSchema;
import graphql.schema.idl.RuntimeWiring;
import graphql.schema.idl.SchemaGenerator;
import graphql.schema.idl.SchemaParser;
import graphql.schema.idl.TypeDefinitionRegistry;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.io.IOException;
import java.net.URL;

import static graphql.schema.idl.TypeRuntimeWiring.newTypeWiring;

@Component
public class GraphQLProvider   {


    @Autowired
    GraphQLDataFetchers graphQLDataFetchers;

    private GraphQL graphQL;

    @PostConstruct
    public void init() throws IOException {
        URL url = Resources.getResource("schema.graphqls");
        String sdl = Resources.toString(url, Charsets.UTF_8);
        GraphQLSchema graphQLSchema = buildSchema(sdl);
        this.graphQL = GraphQL.newGraphQL(graphQLSchema).build();
    }

    private GraphQLSchema buildSchema(String sdl) {
        TypeDefinitionRegistry typeRegistry = new SchemaParser().parse(sdl);
        RuntimeWiring runtimeWiring = buildWiring();
        SchemaGenerator schemaGenerator = new SchemaGenerator();
        return schemaGenerator.makeExecutableSchema(typeRegistry, runtimeWiring);
    }

    private RuntimeWiring buildWiring() {
        return RuntimeWiring.newRuntimeWiring()
                .type(newTypeWiring("Query")
                        .dataFetcher("books", graphQLDataFetchers.getBooks()))
                .type(newTypeWiring("Query")
                        .dataFetcher("bookById", graphQLDataFetchers.getBookByIdDataFetcher()))
                .type(newTypeWiring("Book")
                        .dataFetcher("author", graphQLDataFetchers.getAuthorDataFetcher()))
                .build();
    }

    @Bean
    public GraphQL graphQL() {
        return graphQL;
    }

}

1 Ответ

1 голос
/ 11 июня 2019

Вы можете создать пользовательский объект, который внутренне содержит JWT или просто HttpServletRequest: 

public class GraphQLContext {
    private HttpServletRequest httpServletRequest;
}

При выполнении запроса GraphQL вы создаете этот объект контекста и устанавливаете его на ExecutionInput.Большинство веб-фреймворков должно обеспечивать несколько способов легкого доступа к текущему HttpServletRequest: 

GraphQLContext context = new GraphQLContext(httpServletRequest);
ExecutionInput executionInput = ExecutionInput.newExecutionInput()
                .query(query)
                .context(context)
                .build();

ExecutionResult result = graphQL.execute(executionInput);

Затем в сборщике данных контекст можно получить по: 

@Override
public Object get(DataFetchingEnvironment env) throws Exception {

    GraphQLContext context = env.getContext();
    HttpServletRequest httpServletRequest = context.getHttpServletRequest();

}
...