Spring Security WebFlux/反应异常处理

Question

我看到了以下关于如何定制spring security webflux的帖子：spring webflux自定义API身份验证

如果我们在ServerSecurityContextRepository.Load中抛出异常，那么Spring会将http头更新为500，而我无法操作此异常。

但是，在controller中抛出的任何错误都可以使用常规的@ControllerAdvision来处理，这只是启动了webflux安全性。

spring webflux安全性中是否存在处理异常的问题？

Answer 1

我找到的解决方案是创建一个实现ErrorWebExceptionHandler的组件。ErrorWebExceptionHandlerbean的实例在Spring Security过滤器之前运行。下面是我使用的一个示例：

@Slf4j
@Component
public class GlobalExceptionHandler implements ErrorWebExceptionHandler {

  @Autowired
  private DataBufferWriter bufferWriter;

  @Override
  public Mono<Void> handle(ServerWebExchange exchange, Throwable ex) {
    HttpStatus status = HttpStatus.INTERNAL_SERVER_ERROR;
    AppError appError = ErrorCode.GENERIC.toAppError();

    if (ex instanceof AppException) {
        AppException ae = (AppException) ex;
        status = ae.getStatusCode();
        appError = new AppError(ae.getCode(), ae.getText());

        log.debug(appError.toString());
    } else {
        log.error(ex.getMessage(), ex);
    }

    if (exchange.getResponse().isCommitted()) {
        return Mono.error(ex);
    }

    exchange.getResponse().setStatusCode(status);
    return bufferWriter.write(exchange.getResponse(), appError);
  }
}

如果您要注入HttpHandler，那么它有点不同，但是思想是相同的。

更新：为了完整起见，这里是我的DataBufferWriter对象，它是@component:

@Component
@RequiredArgsConstructor(onConstructor = @__(@Autowired))
@Slf4j
public class DataBufferWriter {
    private final ObjectMapper objectMapper;

    public <T> Mono<Void> write(ServerHttpResponse httpResponse, T object) {
        return httpResponse
            .writeWith(Mono.fromSupplier(() -> {
                DataBufferFactory bufferFactory = httpResponse.bufferFactory();
                try {
                    return bufferFactory.wrap(objectMapper.writeValueAsBytes(object));
                } catch (Exception ex) {
                    log.warn("Error writing response", ex);
                    return bufferFactory.wrap(new byte[0]);
                }
            }));
    }
}