Spring反应器中的上下文参考

Question

我用的是Spring项目Reactor堆芯3.1.8。释放我正在为我的微服务实现一个日志框架，使其具有JSON审计日志，因此使用上下文来存储某些字段，如userID、collaboration ID、component Name和其他几个在请求生命周期中常见的字段。由于Threadlocal不能在反应式服务中用于存储这些元素，因此我必须使用上下文。但是，要了解上下文显然非常困难。我可以通过doOnEach函数调用从信号中获取对上下文的引用，就这样。如果我使用doOnEach，它会被所有信号类型调用，我无法隔离错误、成功等。此外，如果其间发生错误，那么所有后续的doOnEach都会被调用，因此日志会重复使用多个onError日志类型。

关于如何在Spring reactor中获取对上下文对象的引用，文档非常有限。任何关于更好地生成审计日志的方法的帮助，都将在函数调用和外部调用之间存储并传播协作ID和其他特定于请求的ID，我们将不胜感激。

代码片段——在WebFilter中，我设置了几个键值对，如下所示-

override fun filter(exchange: ServerWebExchange, filterChain: WebFilterChain): Mono<Void> {
        // add the context variables at the end of the chain as the context moves from
        // downstream to upstream.
        return filterChain.filter(exchange)
                .subscriberContext { context ->
                    var ctx = context.put(RestRequestInfo::class.java, restRequestInfo(exchange))
                    ctx = ctx.put(COLLABORATION_ID, UUID.randomUUID().toString())
                    ctx=ctx.put(COMPONENT_NAME, "sample-component-name")
                    ctx=ctx.put(USER_NAME, "POSTMAN")
                     ctx
                }
    }

然后我想在所有后续日志中使用上面添加的键值对，这样日志聚合器（如Splunk）就可以根据协作ID获取与这个特定请求相关的所有JSON日志。现在，从上下文中获取值的唯一方法是通过doOnEach函数调用，在这里，我们得到信号的句柄，通过它我们得到上下文的句柄。但所有doOnEach都会在每个事件中被调用，无论每个函数调用是成功还是失败

return Mono.just(request)
            .doOnEach(**Code to log with context data**)
            .map(RequestValidations::validateRequest)
            .doOnEach(**Code to log with context data**)
            .map(RequestValidations::buildRequest)
            .map(RequestValidations::validateQueryParameters)
            .doOnEach(**Code to log with context data**)
            .flatMap(coverageSummariesGateway::getCoverageSummaries)
            .doOnEach(**Code to log with context data**)
            .map({ coverageSummaries -> 
getCoverageSummariesResponse(coverageSummaries, serviceReferenceId) })
            .doOnEach(**Code to log with context data**)
            .flatMap(this::renderSuccess)
            .doOnEach(**Code to log with context data**)
            .doOnError { logger.info("ERROR OCCURRED") }

非常感谢。

Answer 1

你可以这样做：

return Mono.just(request)
            .doOnEach(**Code to log with context data**)
            .flatMap( r -> withMDC(r, RequestValidations::validateRequest))

下面的方法将填充mapping diagnostic context（MDC），以便您的日志中自动包含它（取决于您的日志模式）。例如，logback有%X{traceId}其中traceId是tracingContext映射中的一个键。

java prettyprint-override">public static <T, R> Mono<R> withMDC(T value, Function<T, Mono<R>> f) {
    return Mono.subscriberContext()
               .flatMap( ctx -> {
                    Optional<Map> tracingContext = ctx.getOrEmpty("tracing-context-key");
                    if (tracingContext.isPresent()) {
                        try {
                            MDC.setContextMap(tracingContext.get());
                            return f.apply(value);
                        } finally {
                            MDC.clear();
                        }
                    } else{
                        return f.apply(value);
                    }
               });

}

这不是很好，希望它最终能通过日志框架得到改进，并自动注入上下文。