org.restheart.graphql.scalars.bsonCoercing.CoercingUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of restheart-graphql Show documentation
RESTHeart MongoDB - GraphQL plugin
There is a newer version: 8.1.5
/*-
 * ========================LICENSE_START=================================
 * restheart-graphql
 * %%
 * Copyright (C) 2020 - 2024 SoftInstigate
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see .
 * =========================LICENSE_END==================================
 */
package org.restheart.graphql.scalars.bsonCoercing;

import graphql.schema.Coercing;
import graphql.schema.GraphQLScalarType;
import java.util.Map;
import java.util.HashMap;

import org.restheart.utils.LambdaUtils;

import static graphql.Scalars.GraphQLString;
import static graphql.Scalars.GraphQLInt;
import static graphql.Scalars.GraphQLFloat;
import static graphql.Scalars.GraphQLBoolean;

public class CoercingUtils {
    private static final Map builtInScalars = Map.ofEntries(
            Map.entry("String", GraphQLString),
            Map.entry("Int", GraphQLInt),
            Map.entry("Float", GraphQLFloat),
            Map.entry("Boolean", GraphQLBoolean)
    );

    private static final Map> replacements = Map.ofEntries(
            Map.entry("String", new GraphQLBsonStringCoercing()),
            Map.entry("Int", new GraphQLBsonInt32Coercing()),
            Map.entry("Float", new GraphQLBsonDoubleCoerching()),
            Map.entry("Boolean", new GraphQLBsonBooleanCoercing())
    );

    public static final Map> builtInCoercing = new HashMap<>();

    static String typeName(Object input) {
        return input == null ? "null" : input.getClass().getSimpleName();
    }

    static void saveBuiltInCoercing(){
        builtInScalars.forEach(((s, graphQLScalarType) -> builtInCoercing.put(s, graphQLScalarType.getCoercing())));
    }

    public static void replaceBuiltInCoercing() throws NoSuchFieldException, IllegalAccessException {
        var coercingField =  GraphQLScalarType.class.getDeclaredField("coercing");
        coercingField.setAccessible(true);
        saveBuiltInCoercing();
        replacements.forEach(((s, coercing) -> {
                try {
                    coercingField.set(builtInScalars.get(s), coercing);
                } catch (IllegalAccessException e) {
                    LambdaUtils.throwsSneakyException(new RuntimeException("Error replacing built-in scalars", e));
                }
            }));
        coercingField.setAccessible(false);
    }
}