如何在GraphQL SPQR中设置标题变量

Question

我正在使用GraphQL SPQR和Spring Boot运行GraphQL API。

目前，我正在抛出RuntimeExceptions以返回GraphQL错误。我有一个CustExceptionHandler，它实现了DataFetcherExceptionHandler，以正确的格式返回错误，如下所示：

class CustomExceptionHandler : DataFetcherExceptionHandler {
    override fun onException(handlerParameters: DataFetcherExceptionHandlerParameters?): DataFetcherExceptionHandlerResult {


        // get exception
        var exception = handlerParameters?.exception
        val locations = listOf(handlerParameters?.sourceLocation)
        val path = listOf(handlerParameters?.path?.segmentName)

        // create a GraphQLError from your exception
        if (exception !is GraphQLError) {
            exception = CustomGraphQLError(exception?.localizedMessage, locations, path)
        }

        // cast to GraphQLError
        exception as CustomGraphQLError

        exception.locations = locations
        exception.path = path

        val errors = listOf<GraphQLError>(exception)

        return DataFetcherExceptionHandlerResult.Builder().errors(errors).build()
    }
}

我使用CustomExceptionHandler如下（在我的主应用程序类中）：

@Bean
    fun graphQL(schema: GraphQLSchema): GraphQL {
        return GraphQL.newGraphQL(schema)
                .queryExecutionStrategy(AsyncExecutionStrategy(CustomExceptionHandler()))
                .mutationExecutionStrategy(AsyncSerialExecutionStrategy(CustomExceptionHandler()))
                .build()
    }

更好的是，是否可以创建一个Spring Bean，将UUID放在所有查询和突变的头中？

谢谢

Answer 1

使用spring boot时，有两种选择：

• 您使用的是spring boot graphql spqr启动器（它提供了自己的控制器来处理所有graphql请求）

无论如何，你有几个选择：

这可能是最简单的方法——而且在任何情况下都可能有效：您可以简单地将CustomExceptionHandler设为Spring bean并让它自动连接HttpServletRequest——在处理程序方法中，您可以将其设置为您想要的任何内容。这是Java中的一些虚拟代码（抱歉，我对静态编程语言不够精通）：

@Component
class CustomExceptionHandler implements DataFetcherExceptionHandler {
    private final HttpServletResponse response; 

    public CustomExceptionHandler(HttpServletResponse response) {
        this.response = response; 
    }

    @Override
    public DataFetcherExceptionHandlerResult onException(DataFetcherExceptionHandlerParameters handlerParameters) {
        response.setHeader("X-Request-ID", UUID.randomUUID().toString());
        // ... your actual error handling code
    }
}

这将起作用，因为Spring会意识到HttpServletRequest对于每个请求都不同。因此，它将向您的错误处理程序注入一个动态代理，该代理将指向每个请求的实际HttpServletACK实例。

我认为，这不是最优雅的方式，但它肯定会解决你的问题。

在使用此启动器的项目中使用了一个默认控制器实现。该控制器将处理您收到的每个图形化请求。您可以通过实现自己的GraphQLExector并将其设为Spring bean来自定义它。该执行器负责调用GraphQL引擎，将参数传入并输出响应。这是默认实现，您可能希望将其作为工作的基础。

与前面的解决方案类似，您可以在该类中自动连接HttpServletResponse，并设置HTTP响应头。

该解决方案将允许您决定是在所有情况下设置请求ID，还是仅在特定的错误情况下设置。（graphql.execute返回一个对象，您可以从中获取是否存在错误以及存在哪些错误的信息）

找到GraphQL控制器，向HttpServletRequest类型的方法添加一个参数，然后根据需要向该方法添加头（请参阅前一节中的一些更具体的建议）