开源地址

• GitHub：https://github.com/lzhpo/logging-tracer
• Gitee：https://gitee.com/lzhpo/logging-tracer

前言

当我们使用Splunk、ELK等日志工具的时候，我们希望有一个traceId就可以把这个请求关联的所有服务日志都搜索出来，并且可以看到上下游服务名称，以及当前请求经历了几层服务，这样子更加方便我们定位问题。

日志样例：

2022-08-18 21:44:17.611 DEBUG [service-sample,webclient-sample,a43972b7c68846ccafa4fc9676869d1a,0.1.1.1.1.1] 15788 --- [nio-9000-exec-1] c.l.t.servlet.TracerServletInterceptor   : Built logging tracer context: {X-B3-TraceId=a43972b7c68846ccafa4fc9676869d1a, X-B3-SpanName=service-sample, X-B3-SpanId=0.1.1.1.1.1, X-B3-Parent-SpanName=webclient-sample}
2022-08-18 21:44:17.622  INFO [service-sample,webclient-sample,a43972b7c68846ccafa4fc9676869d1a,0.1.1.1.1.1] 15788 --- [nio-9000-exec-1] c.l.t.s.service.ServiceSampleController  : Received new request for hello api.
2022-08-18 21:44:17.622  INFO [service-sample,webclient-sample,a43972b7c68846ccafa4fc9676869d1a,0.1.1.1.1.1] 15788 --- [nio-9000-exec-1] c.l.t.s.service.ServiceSampleController  : Request header with [accept-encoding: gzip]

[service-sample,webclient-sample,a43972b7c68846ccafa4fc9676869d1a,0.1.1.1.1.1]：

• service-sample：当前服务名称。
• webclient-sample：上游服务名称。
• a43972b7c68846ccafa4fc9676869d1a：链路ID，即traceId。
• 0.1.1.1.1.1：请求经历的服务层级。
  • 假设是0：表示到达当前位置，没有经历下游服务。
  • 假设是0.1：表示到达当前位置，已经经历了1层下游服务。
  • 假设是0.1.1：表示到达当前位置，已经经历了2层下游服务。
  • 以此类推...

快速使用

导入依赖

依赖已发布至Maven中央仓库，可直接引入依赖。

• Maven：

  <dependency>
    <groupId>com.lzhpo</groupId>
    <artifactId>logging-tracer-spring-boot-starter</artifactId>
    <version>${latest-version}</version>
  </dependency>
• Gradle:

  implementation 'com.lzhpo:logging-tracer-spring-boot-starter:${latest-version}'

PS：如果不需要定制，直接导入相关依赖按照下面组件说明正常使用即可。

使用示例

在这模拟多链路调用服务，按如下顺序逐层call：

1.Feign

导入feign依赖将自动生效。

<dependency>
    <groupId>org.springframework.cloud</groupId>
    <artifactId>spring-cloud-starter-openfeign</artifactId>
</dependency>

参考示例：logging-tracer-feign-sample

2.HttpClient

导入httpclient依赖将自动生效。

<dependency>
    <groupId>org.apache.httpcomponents.client5</groupId>
    <artifactId>httpclient5</artifactId>
    <version>${httpclient.version}</version>
</dependency>

和使用HttpClients一样，不同的是在这使用的TracerHttpClients 的Bean，方法和HttpClients一样。

参考示例：logging-tracer-httpclient-sample

3.Okhttp

导入okhttp依赖将自动生效。

<dependency>
    <groupId>com.squareup.okhttp3</groupId>
    <artifactId>okhttp</artifactId>
    <version>${okhttp.version}</version>
</dependency>

• 之前：

  OkHttpClient okHttpClient = new OkHttpClient();

• 现在：注入OkHttpClient.BuilderBean直接使用，就不是直接new了。

  @Autowired
  private OkHttpClient.Builder okHttpClientBuilder;
  
  // 略...
  OkHttpClient okHttpClient = okHttpClientBuilder.build();

参考示例：logging-tracer-okhttp-sample

4.RestTemplate

直接注入RestTemplateBean使用即可。

参考示例：logging-tracer-resttemplate-sample

5.Webclient

直接注入WebclientBean使用即可。

参考示例：logging-tracer-webclient-sample

自定义配置

自定义配置文件

自定义日志链路输出格式

logging:
  tracer:
    pattern: '%5p [${spring.application.name:},%X{X-B3-Parent-SpanName},%X{X-B3-TraceId},%X{X-B3-SpanId}]'

当然，你也可以按照以前的方式配置，也是一样生效的，如果两个都配置了，优先使用上面的。

logging:
  pattern:
    level: '%5p [${spring.application.name:},%X{X-B3-Parent-SpanName},%X{X-B3-TraceId},%X{X-B3-SpanId}]'

自定义转发到下游服务的请求头

logging:
  tracer:
    proxy-headers:
      - Authorization
      - User-Id

然后，这些请求头将会自动转发到下游服务，比如，我需要在日志中只显示一个User-Id，就可以这样配置：

logging:
  pattern:
    level: '%5p [%X{User-Id}]'

如果是多个，英文逗号分隔即可。

关闭logging-tracer功能

logging:
  tracer:
    enabled: false

多线程环境下使用

可以参考logging-tracer-resttemplate-sample模块中的RestTemplateSampleController中示例代码。

1. ThreadPoolTaskExecutor【推荐】：如果是注入ThreadPoolTaskExecutor线程池执行多线程任务的话，则正常使用即可，logging-tracer会自动处理相关逻辑。

   @Autowired
   private ThreadPoolTaskExecutor threadPoolTaskExecutor;
   
   // 略...
   threadPoolTaskExecutor.execute(() -> log.info("[threadPool] Hello, I'm threadPool."));

2. Thread: 因为logging-tracer依赖的是MDC，所以，如果是使用的Thread，需要手动在子父线程中维持MDC相关上下文。

    Map<String, String> context = MDC.getCopyOfContextMap();
    Thread thread = new Thread(() -> {
      MDC.setContextMap(context);
      log.info("[multiThread] Hello, I'm multiThread.");
    });
    thread.start();

3. 其它的方式也同理。

自定义业务逻辑

实现接口TracerContextCustomizer，加入到Spring容器中即可。

比如，当你需要添加请求头转发到下游服务的时候，又或者是你需要自定义日志链路输出格式，如果请求头中或当前context中没有此变量，就需要将相关的变量添加到context中。

@Component
public class SampleTracerContextCustomizer implements TracerContextCustomizer {

  @Override
  public void customize(Map<String, String> context) {
    context.put("abc", "123");
    context.put("def", "456");
  }
}

鸣谢

感谢Jetbrains提供的License！

公众号

微信	公众号