使用 Mono 和上下文增强错误处理

2024-03-21 17:20:49

使用 Mono 增强错误处理

在微服务架构中，我们经常需要与其他服务进行交互。这些交互通常是通过 RESTful API 进行的，我们使用 Reactive 框架来处理这些异步调用。Reactive 编程提供了对异步数据流进行建模和转换的工具，这对于处理分布式系统的复杂性非常有用。

Mono 和错误处理

Spring Reactive WebFlux 使用 Mono 和 Flux 来表示反应式数据流。Mono 表示一个单值序列，而 Flux 表示一个多值序列。在处理与 REST 服务的调用时，我们可以使用 Mono 来建模服务器的响应。

Mono 提供了一个 onErrorMap 操作符，允许我们对错误进行映射，从而创建自定义异常。这对于为不同类型的错误提供有意义的消息非常有用。

增强异常处理

我们希望通过添加有关被调用服务的附加信息来增强异常处理。这将使我们能够在必须抛出异常时提供更详细的消息。

为了解决这个问题，我们需要一种方法来将服务信息传递给错误映射。一种方法是在调用周围传递服务信息，然后在错误映射中捕获并抛出自定义异常。但是，这可能会导致代码混乱且难以维护。

使用上下文

一种更优雅的方法是使用上下文。Spring Reactive WebFlux 允许我们使用 Context 存储与请求相关的信息。我们可以将服务信息存储在上下文中，然后在错误映射中访问它。

要将服务信息存储在上下文中，我们可以使用 contextWrite 操作符。它允许我们在流中写入上下文值。

return getMonoForServiceCall(param)
    .contextWrite(context -> context.put("servicePath", path))
    .timeout(Duration.ofSeconds(10))
    .onErrorMap(t -> handleExceptions(t));

在错误映射中，我们可以使用 contextRead 操作符访问上下文值。

Throwable handleExceptions(Throwable t) {
  String servicePath = contextRead("servicePath").orElse(null);
  return new RestClientException("调用 " + servicePath + " 时超时");
}