spring WebFilter 覆盖 ExceptionHandler header

spring WebFilter overrides ExceptionHandler header

我同时使用 WebFilter 和 WebExceptionHandler。 仅当 ExceptionHandler 未设置时,WebFilter 才应添加新的 header。 但是WebFilter是在WebHttpHandler执行ExceptionHandler之前添加到ServerWebExchange的,所以无法判断ExceptionHandler是否被触发。

@Component
@Order(-2)
class MyErrorWebExceptionHandler(g: MyErrorAttributes, applicationContext: ApplicationContext, serverCodecConfigurer: ServerCodecConfigurer)
    : AbstractErrorWebExceptionHandler(g, ResourceProperties(), applicationContext) {

    init {
        super.setMessageWriters(serverCodecConfigurer.writers)
        super.setMessageReaders(serverCodecConfigurer.readers)
    }

    @Override
    override fun getRoutingFunction(errorAttributes: ErrorAttributes): RouterFunction<ServerResponse> {
        return RouterFunctions.route(RequestPredicates.all(), HandlerFunction<ServerResponse> { renderErrorResponse(it) })
    }

    private fun renderErrorResponse(request: ServerRequest): Mono<ServerResponse> {

        val errorPropertiesMap = getErrorAttributes(request, false)

        return ServerResponse.status(HttpStatus.OK)
                .contentType(MediaType.APPLICATION_JSON_UTF8)
                .headers { x ->
                        x.set(c_ApplicationStatus, errorPropertiesMap[c_ApplicationStatus].toString())
                      }.build()
}

@Component
class ServerResponseHeaderWebFilter : WebFilter {
    override fun filter(exchange: ServerWebExchange, chain: WebFilterChain): Mono<Void> {
        exchange.response.headers.set(c_ApplicationStatus, HttpStatus.OK.value().toString())
        return chain.filter(exchange)

    }
}

在这种情况下,我们可以快速模拟执行顺序,例如:

WebFilter
|- setting the application status header to OK
|- calling chain.filter
   |- finding the handler and calling it
   |- in case of an error, the exception handler is called
|- after chain.filter

过滤器链完成交换后,响应已提交并处理,因此此时无法更改响应 headers。使用此代码示例,异常处理程序(如果执行)将覆盖 Web 过滤器设置的任何 header。

所以从技术上讲,您的问题的答案是,一旦处理程序链处理完响应,就无法修改响应。这是 Spring WebFlux 设计的预期行为。

但听起来我们需要退后一步,谈谈您想要实现的目标。

您是否正在尝试检查您的异常处理程序是否在开发功能时被调用? 您是否正在尝试以某种目前无法实现的方式调整 Spring 引导中的错误处理机制?

如果您想谈谈您想要实现的目标,请提出另一个问题,以便 SO 用户可以从这个问题中受益。

最终我发现过滤器中设置的 header 可以在 request.exchange() object 中看到。它必须在那里被删除,以便新的 header 集来替换它。

private fun renderErrorResponse(request: ServerRequest): Mono<ServerResponse> {

        val errorPropertiesMap = getErrorAttributes(request, false)

        request.exchange().response.headers.remove(c_ApplicationStatus)
        return ServerResponse.status(HttpStatus.OK)
                .contentType(MediaType.APPLICATION_JSON_UTF8)
                .headers { x -> x.set(c_ApplicationStatus, value) }
                .build()
}