graphql.servlet.GraphQLVariables Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graphql-java-servlet Show documentation
relay.js-compatible GraphQL servlet
There is a newer version: 6.1.3
/**
 * Copyright 2016 Yurii Rashkovskii
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 */
package graphql.servlet;

import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import graphql.annotations.GraphQLObjectBackedByClass;
import graphql.language.*;
import graphql.parser.Parser;
import graphql.schema.GraphQLSchema;
import graphql.schema.GraphQLType;
import lombok.SneakyThrows;

import java.util.HashMap;
import java.util.Map;
import java.util.function.Consumer;

public class GraphQLVariables extends HashMap {

    private final GraphQLSchema schema;
    private final String query;

    public GraphQLVariables(GraphQLSchema schema, String query, Map variables) {
        super();
        this.schema = schema;
        this.query = query;
        ObjectMapper objectMapper = new ObjectMapper();
        // this will help combating issues with unknown fields like clientMutationId
        objectMapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        new Parser().parseDocument(query).getDefinitions().stream()
                    .filter(d -> d instanceof OperationDefinition)
                    .map(d -> (OperationDefinition) d)
                    .flatMap(d -> d.getVariableDefinitions().stream())
                    .forEach(new Consumer() {
                        @SneakyThrows
                        @Override public void accept(VariableDefinition d) {
                            GraphQLType type;
                            Type t = d.getType();
                            if (t instanceof TypeName) {
                                type = schema.getType(((TypeName) t).getName());
                            } else if (t instanceof NonNullType) {
                                accept(new VariableDefinition(d.getName(), ((NonNullType) t).getType()));
                                return;
                            } else {
                                type = null;
                            }
                            if (type instanceof GraphQLObjectBackedByClass) {
                                String value = objectMapper.writeValueAsString(variables.get(d.getName()));
                                Object val = objectMapper.readValue(value, ((GraphQLObjectBackedByClass) type).getObjectClass());
                                GraphQLVariables.this.put(d.getName(), val);
                            } else {
                                GraphQLVariables.this.put(d.getName(), variables.get(d.getName()));
                            }
                        }
                    });
    }
}