2

我正在使用 'io.leangen.graphql' 的 'graphql-spqr-spring-boot-starter' 库版本 0.0.4。我可以自定义错误。请参阅以下代码和屏幕截图以供参考:

楷模:

@Getter
@Setter
@ToString
@Entity
@Accessors
public class Student {
    
    @Id 
    @GraphQLQuery(name = "id", description = "A Student's ID")
    private Long id;
    
    @GraphQLQuery(name = "name", description = "A student's name")
    private String name;    
    
    private String addr;

}

服务等级:

@Service
@GraphQLApi
public class StudentService{
    
    private final StudentRepository studentRepository;
    private final AddressRepository addressRepository;
    
    public StudentService(StudentRepository studentRepository, AddressRepository addressRepository) {
        this.addressRepository = addressRepository;
        this.studentRepository = studentRepository;
    }
    
    @GraphQLQuery(name = "allStudents")
    public List<Student> getAllStudents() {
        return studentRepository.findAll();
    }

    @GraphQLQuery(name = "student")
    public Optional<Student> getStudentById(@GraphQLArgument(name = "id") Long id) {
        if(studentRepository.findById(id) != null)
            return studentRepository.findById(id);
        throw new StudentNotFoundException("We were unable to find a student with the provided id", "id");
    }
    
    @GraphQLMutation(name = "saveStudent")
    public Student saveStudent(@GraphQLArgument(name = "student") Student student) {
        if(student.getId() == null)
            throw new NoIdException("Please provide an Id to create a Student entry.");
        return studentRepository.save(student);
    }
}

自定义异常类:

import java.util.List;

import graphql.ErrorType;
import graphql.GraphQLError;
import graphql.language.SourceLocation;

public class NoIdException extends RuntimeException implements GraphQLError {
    
    private String noIdMsg;
    
    public NoIdException(String noIdMsg) {
        this.noIdMsg = noIdMsg;
    }

    @Override
    public List<SourceLocation> getLocations() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public ErrorType getErrorType() {
        // TODO Auto-generated method stub
        return ErrorType.ValidationError;
    }

    @Override
    public String getMessage() {
        // TODO Auto-generated method stub
        return noIdMsg;
    }
}

帖子图片

但是,我不确定如何摆脱Exception while fetching data (/saveStudent)上面屏幕截图中的message字段。我知道我们可以拥有GraphQLExceptionHandler实现GraphQLErrorHandler (graphql-java-kickstart). 但是有什么选择sqpr-spring-boot-starter呢?

import graphql.*;
import graphql.kickstart.execution.error.*;
import org.springframework.stereotype.*;

import java.util.*;
import java.util.stream.*;

@Component
public class GraphQLExceptionHandler implements GraphQLErrorHandler {

    @Override
    public List<GraphQLError> processErrors(List<GraphQLError> list) {
        return list.stream().map(this::getNested).collect(Collectors.toList());
    }

    private GraphQLError getNested(GraphQLError error) {
        if (error instanceof ExceptionWhileDataFetching) {
            ExceptionWhileDataFetching exceptionError = (ExceptionWhileDataFetching) error;
            if (exceptionError.getException() instanceof GraphQLError) {
                return (GraphQLError) exceptionError.getException();
            }
        }
        return error;
    }
}

有人可以帮助我如何删除此声明并仅发送特定消息吗?

4

1 回答 1

1

您可以创建一个Bean并覆盖DataFetcherExceptionHandler. 要覆盖它,您还必须覆盖执行策略:

@Bean
public GraphQL graphQL(GraphQLSchema schema) {
    return GraphQL.newGraphQL(schema)
            .queryExecutionStrategy(new AsyncExecutionStrategy(new CustomDataFetcherExceptionHandler()))
            .mutationExecutionStrategy(new AsyncSerialExecutionStrategy(new CustomDataFetcherExceptionHandler()))
            .build();
}

private static class CustomDataFetcherExceptionHandler implements DataFetcherExceptionHandler {

    @Override
    public DataFetcherExceptionHandlerResult onException(DataFetcherExceptionHandlerParameters handlerParameters) {
        Throwable exception = handlerParameters.getException();
        SourceLocation sourceLocation = handlerParameters.getSourceLocation();
        CustomExceptionWhileDataFetching error = new CustomExceptionWhileDataFetching(exception, sourceLocation);

        return DataFetcherExceptionHandlerResult.newResult().error(error).build();
    }
}

private static class CustomExceptionWhileDataFetching implements GraphQLError {

    private final String message;
    private final List<SourceLocation> locations;

    public CustomExceptionWhileDataFetching(Throwable exception, SourceLocation sourceLocation) {
        this.locations = Collections.singletonList(sourceLocation);
        this.message = exception.getMessage();
    }

    @Override
    public String getMessage() {
        return this.message;
    }

    @Override
    public List<SourceLocation> getLocations() {
        return this.locations;
    }

    @Override
    public ErrorClassification getErrorType() {
        return ErrorType.DataFetchingException;
    }
}
于 2021-04-23T05:21:51.347 回答