Spring WebFilter覆蓋ExceptionHandler標頭

Question

我正在同時使用WebFilter和WebExceptionHandler。 僅當ExceptionHandler未設置標頭時，WebFilter才應添加新標頭。 但是，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)

    }
}

Answer 1

在這種情況下，我們可以使用以下方法快速對執行順序進行建模：

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用戶可以從該問題中受益。

Answer 2

最終，我發現可以在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()
}