反应式单声道如何将订阅上下文传播到`doOnSubscribe`和`doFinally`

Question

我有以下跟踪方法执行时间的方面：

public Object addMetricsToReactiveMonoTimedMethod(ProceedingJoinPoint pjp, ReactiveTimed reactiveTimed) throws Throwable {
        StopWatch stopWatch = new StopWatch();
        Mono<?> mono = (Mono<?>) pjp.proceed();
        return mono
                .doOnSubscribe(subscription -> stopWatch.start())
                .doFinally(signalType -> {
                    stopWatch.stop();
                    logTimer(pjp, stopWatch, reactiveTimed.name(), signalType);
                });
}

方法本身是这样的：

public Mono<String> sayHi() { 
  return Mono.just("hi")
    .subscriberContext(context -> context.put("requestId", "requestId"));
}

如何在我的方面方法中从订阅者上下文中获取requestId 变量？我想在doFinally 中使用它来了解分析了哪个请求。

Answer 1

免责声明：我一生中从未使用过 Reactor 或类似的东西。由于 aspectj 标签，我发现了这个问题。

在我快速查看Mono Javadoc 之后，您似乎可以在方面再次调用subscriberContext(Function<Context, Context>)，就像在目标方法中一样。您将现有上下文作为函数或 lambda 的输入参数，并可以根据需要使用它。您的虚拟函数/lambda 的结果将是一个新的上下文，但您可以丢弃它。我没有测试过，但我的意思是这样的：

// ...

Mono<?> mono = (Mono<?>) pjp.proceed();

// Alternatively, use a List<Context> with one element, a Stack<Context>, ...
Context[] targetContext = new Context[1];

mono.subscriberContext(context -> {
  targetContext[0] = context;
  // We can also return null, it does not matter because we are not interested
  // in the newly created context, only in the original one we salvaged into the
  // outer array.
  return context;
});

System.out.println("Now do whatever you need to do with " + targetContext[0]);

// ...

我使用单元素数组/列表 - 相反，您可以“滥用”任何其他包装器对象，例如原子引用或线程本地作为包装器，因为您不能从内部直接将 a 分配给 Context 变量拉姆达。代码无法编译，因为从 lambda 内部引用的外部变量必须是有效的 final。