Spring WebFilter覆盖ExceptionHandler标头

时间:2018-12-20 13:04:04

标签: java kotlin spring-webflux

我同时使用了WebFilter和WebExceptionHandler。 仅当ExceptionHandler未设置标头时,WebFilter才应添加新标头。 但是,WebFilter是在WebHttpHandler执行ExceptionHandler之前将WebFilter添加到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)

    }
}

2 个答案:

答案 0 :(得分:0)

在这种情况下,我们可以使用以下方法快速为执行顺序建模:

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

一旦交换完成了过滤器链,就已经提交并处理了响应,因此此时无法更改响应头。对于此代码示例,异常处理程序(如果执行)将覆盖Web过滤器设置的任何标头。

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

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

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

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

答案 1 :(得分:0)

最终我发现可以在request.exchange()对象中看到过滤器中设置的标头。必须将其删除以替换新的标头集。

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()
}