在TimeoutMainSubscriber的情况下，将反应器上下文复制到MDC

Hooks.onEachOperator("TimeoutMainSubscriber", (publisher, subscriber) -> {
    if (subscriber instanceof TimeoutMainSubscriber) {
        return new CoreSubscriber<>() {
            @Override
            public void onSubscribe(Subscription subscription) {
                subscriber.onSubscribe(subscription);
            }

            @Override
            public void onNext(Object o) {
                subscriber.onNext(o);
            }

            @Override
            public void onError(Throwable throwable) {
                subscriber.onError(throwable);
            }

            @Override
            public void onComplete() {
                subscriber.onComplete();
            }

            @Override
            public Context currentContext() {
                // 获取当前的反应器上下文
                Context reactorContext = subscriber.currentContext();

                // 将反应器上下文复制到MDC
                reactorContext.stream()
                        .forEach(entry -> MDC.put(entry.getKey(), entry.getValue().toString()));

                try {
                    return subscriber.currentContext();
                } finally {
                    // 清理MDC中的上下文信息
                    reactorContext.stream()
                            .forEach(entry -> MDC.remove(entry.getKey()));
                }
            }
        };
    } else {
        return subscriber;
    }
});