如何在Spring webflux中处理异常并返回正确的HTTP代码？

@Component
@Order(-2)
public class GlobalExceptionHandler implements ErrorWebExceptionHandler {

    @Override
    public Mono<Void> handle(ServerWebExchange exchange, Throwable ex) {
        HttpStatus httpStatus;
        String errorMessage;

        if (ex instanceof ResponseStatusException) {
            ResponseStatusException responseStatusException = (ResponseStatusException) ex;
            httpStatus = responseStatusException.getStatus();
            errorMessage = responseStatusException.getReason();
        } else {
            httpStatus = HttpStatus.INTERNAL_SERVER_ERROR;
            errorMessage = "Internal Server Error";
        }

        ServerHttpResponse response = exchange.getResponse();
        response.setStatusCode(httpStatus);
        response.getHeaders().setContentType(MediaType.APPLICATION_JSON);

        String errorDetails = ExceptionUtils.getStackTrace(ex);
        ErrorResponse errorResponse = new ErrorResponse(httpStatus.value(), errorMessage, errorDetails);

        DataBuffer buffer = response.bufferFactory().wrap(JsonUtils.toJsonBytes(errorResponse));
        return response.writeWith(Mono.just(buffer));
    }
}