Modificar escalares de código fuente GraphQL

Cuando estaba aprendiendo graphql recientemente, había una tabla con datos de tipo de fecha, pero al configurar el tipo de campo, descubrí que graphql solo admite sus tipos básicos.

Como se muestra en la figura:

No contiene datos de tipo de fecha.

Utilice GarphQLString a la fuerza, los datos de salida son los siguientes

Así que pensé en escribir un método para admitir el tipo Date. Eché un vistazo más de cerca al código fuente de Scalars y escribí uno yo mismo.

Primero mire los resultados de la prueba:

El código final es el siguiente:

MyScalars.java

package com.woodie.modifyoriginal;

import graphql.Scalars;
import graphql.schema.Coercing;
import graphql.schema.CoercingParseLiteralException;
import graphql.schema.GraphQLScalarType;

import java.text.SimpleDateFormat;
import java.util.Date;

public class MyScalars extends Scalars {

    /**
     * 定义一个新的方法
     */
    public static final GraphQLScalarType GraphQLDate = new GraphQLScalarType("Date", "Built-in Date", new Coercing<String, String>() {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");

        public String serialize(Object input) {
            return sdf.format(input);
        }

        public String parseValue(Object input) {
            return this.serialize(sdf.format(input));
        }

        public String parseLiteral(Object input) {
            if (!(input instanceof Date)) {
                throw new CoercingParseLiteralException("Expected AST type 'Date' but was ");
            } else {
                return sdf.format(input);
            }
        }
    });
}

Código de clase de entidad:

Employee.java

package com.woodie.entity;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;

import lombok.*;

import java.util.Date;

@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
@TableName("employees")
public class Employee {
    // 指定表的id
    // 通过type = IdType.AUTO设置id的自动增长， 由于表employees没有设置id为自动增长，因此这里不能设置这个参数，否则报错
    //@TableId(value = "emp_no", type = IdType.AUTO)
    @TableId(value = "emp_no")
    Integer empNo;
    Date birthDate;
    String firstName;
    String lastName;
    Date hireDate;
    String gender;
}

Código de prueba

GraphqlDemo.java

Aquí, el campo Tipo de fecha se pasa en el tipo personalizado de MyScalars.GraphQLDate

package com.woodie;

import com.woodie.entity.Employee;
import com.woodie.modifyoriginal.MyScalars;
import graphql.ExecutionResult;
import graphql.GraphQL;
import graphql.Scalars;
import graphql.schema.GraphQLArgument;
import graphql.schema.GraphQLFieldDefinition;
import graphql.schema.GraphQLObjectType;
import graphql.schema.GraphQLSchema;

import java.text.SimpleDateFormat;
import java.util.Date;

public class GraphqlDemo {
    public static void main(String[] args) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");

        /**
         * type User { #定义对象
         */
        GraphQLObjectType employeeObjectType = GraphQLObjectType.newObject()
                .name("Employee")
                .field(GraphQLFieldDefinition.newFieldDefinition().name("empNo").type(Scalars.GraphQLInt))
                .field(GraphQLFieldDefinition.newFieldDefinition().name("birthDate").type(MyScalars.GraphQLDate))
                .field(GraphQLFieldDefinition.newFieldDefinition().name("hireDate").type(MyScalars.GraphQLDate))
//                .field(GraphQLFieldDefinition.newFieldDefinition().name("birthDate").type(Scalars.GraphQLString))
//                .field(GraphQLFieldDefinition.newFieldDefinition().name("hireDate").type(Scalars.GraphQLString))
                .field(GraphQLFieldDefinition.newFieldDefinition().name("gender").type(Scalars.GraphQLString))
                .field(GraphQLFieldDefinition.newFieldDefinition().name("firstName").type(Scalars.GraphQLString))
                .field(GraphQLFieldDefinition.newFieldDefinition().name("lastName").type(Scalars.GraphQLString))
                .build();

        /**
         * user : User #指定对象以及参数类型
         */
        GraphQLFieldDefinition userFieldDefinition = GraphQLFieldDefinition.newFieldDefinition()
                .name("employee")
                .type(employeeObjectType)
//                .dataFetcher(new StaticDataFetcher(new User(1L, "张三", 26)))
                .argument(GraphQLArgument.newArgument().name("empNo").type(Scalars.GraphQLLong).build())
                .dataFetcher(environment -> {
                    Long id = environment.getArgument("empNo");
                    // 查询数据库
                    // 这里需要设置modules中的sources中的language level: 为8-Lambdas.....
                    return Employee.builder().birthDate(new Date()).empNo(10001).gender("M").firstName("Woodie").lastName("Wang").hireDate(sdf.parse("2222- 03-14")).build();
                })
                .build();

        /**
         * type UserQuery { #定义查询的类型
         */
        GraphQLObjectType userQueryObjectType = GraphQLObjectType.newObject().name("UserQuery").field(userFieldDefinition).build();

        GraphQLSchema graphQLSchema= GraphQLSchema.newSchema().query(userQueryObjectType).build();

        GraphQL graphQL = GraphQL.newGraphQL(graphQLSchema).build();

//        user(id:1) 这里是设置参数 的意思
        String query = "{employee(empNo:10001){empNo,hireDate}}";
        ExecutionResult result = graphQL.execute(query);

//        System.out.println(result.getErrors());
//        System.out.println(result.getData());
        // graphql标准化输出数据
        System.out.println(result.toSpecification());
    }

}