在TimeoutMain Subcriber的情况下，将Reator上下文复制到MDC

Question

我一直在遵循代码在https://github.com/archie-swif/webflux-mdc/blob/master/src/main/java/com/example/webfluxmdc/MdcContextLifter.java把数据从Reactor上下文到MDC使用钩子。

除了Mono超时和TimeoutMainSubscriber正在使用之外，该解决方案在大多数情况下都能完美工作。

下面是我使用的小测试。

Mono<Integer> integerMono = Mono.just(1)
            .doOnEach(logger.addToContext(i -> ImmutableMap.of("index", i)))
            .flatMap(___ -> Mono.fromSupplier(() -> {
                try {
                    Thread.sleep(6000);
                } catch (InterruptedException e) {
                    return 0;
                }
                return 1;
            }))
            .timeout(Duration.ofMillis(5000), Schedulers.parallel())
            .doOnEach(logger.info("testMDCLogging"))
            .doOnError(logger.error("testMDCErrorLogging"))
            .subscriberContext(logger.initContext());

在通过MDC提升程序验证上下文的传播时，onNext（）onError（）和onComplete（）在TimeoutMainSubscriber（上下文复制到MDC）中不调用，只调用currentContext（）。

如何传播上下文并复制到MDC，以便在Mono超时的情况下在doOnError和doOneach中使用？我不想将副本移动到当前上下文（），因为这个方法被调用了几次，在我看来它不是最佳解决方案。

Answer 1

您是否尝试在MdcContextLifter类中重写onError方法？

    @Override
    public void onError(Throwable t) {
        copyToMdc(coreSubscriber.currentContext()); // similar to onNext()
        coreSubscriber.onError(t);
    }