WebFlux 异常处理：全面指南

2024-03-29 17:18:27

在 WebFlux 中优雅地处理异常

在构建异步、非阻塞的 Web 应用程序时，有效处理异常对于确保应用程序的稳定性和用户体验至关重要。WebFlux 框架提供了全面的异常处理机制，本文将深入探讨如何在 WebFlux 中正确地处理异常，涵盖全局异常处理、自定义异常、使用 onErrorResume 处理异常以及遵循最佳实践。

全局异常处理

WebFlux 允许你通过 @RestControllerAdvice 注解集中处理异常。通过创建一个带有此注解的类，你可以定义各种类型的异常处理程序，并在出现异常时返回适当的 HTTP 响应。

@RestControllerAdvice
public class GlobalExceptionHandler {

    @ExceptionHandler(ResourceNotFoundException.class)
    public ResponseEntity<String> handleResourceNotFoundException(ResourceNotFoundException exception, WebRequest request) {
        return ResponseEntity.status(HttpStatus.NOT_FOUND).body(exception.getMessage());
    }
}

自定义异常

为了处理应用程序中特定的错误情况，你可以创建自定义异常类并指定其 HTTP 状态码。这将使你能够提供更详细和有意义的异常消息。

@ResponseStatus(value = HttpStatus.NOT_FOUND)
public class ResourceNotFoundException extends RuntimeException {
    private String resourceName;
    private String fieldName;
    private String fieldValue;

    public ResourceNotFoundException(String resourceName, String fieldName, String fieldValue) {
        super(String.format("%s not found with %s : %s", resourceName, fieldName, fieldValue));
        this.resourceName = resourceName;
        this.fieldName = fieldName;
        this.fieldValue = fieldValue;
    }
}

使用 onErrorResume 处理异常

onErrorResume 操作符允许你捕获异常并返回一个替代响应。这对于处理特定类型的异常并避免应用程序崩溃非常有用。

Flux<ResponseEntity<String>> example() {
    return quoteService.findAllQuotes()
            .map(ResponseEntity::ok)
            .onErrorResume(ResourceNotFoundException.class,
                    ex -> Mono.just(ResponseEntity.notFound().build()));
}