6. 分布式链路追踪RestTemplate拦截器实现设计

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestTemplate;
import org.apache.skywalking.apm.toolkit.activation.opentracing.SkyWalkingOpenTracingActivator;
import org.apache.skywalking.apm.toolkit.activation.trace.SkyWalkingTraceActivation;
import org.apache.skywalking.apm.toolkit.activation.webflux.SkyWalkingWebFluxActivator;
import org.apache.skywalking.apm.toolkit.trace.TraceContext;
 
@Configuration
public class RestTemplateConfig {
 
    @Bean
    public RestTemplate restTemplate() {
        // 创建RestTemplate实例
        RestTemplate restTemplate = new RestTemplate();
        // 添加拦截器
        restTemplate.getInterceptors().add(new TracingInterceptor(TraceContext.getCurrentTraceId()));
        return restTemplate;
    }
 
    static {
        // 激活SkyWalking的WebFlux、OpenTracing和传统跟踪工具包激活器
        SkyWalkingWebFluxActivator.activate();
        SkyWalkingOpenTracingActivator.activate();
        SkyWalkingTraceActivation.active();
    }
}
 
class TracingInterceptor implements ClientHttpRequestInterceptor {
    private String traceId;
 
    public TracingInterceptor(String traceId) {
        this.traceId = traceId;
    }
 
    @Override
    public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException {
        // 在这里可以添加逻辑将traceId注入到请求的header中
        request.getHeaders().add("sw6-trace-id", traceId);
        // 执行请求
        return execution.execute(request, body);
    }
}

这个代码示例展示了如何在Spring应用中配置RestTemplate并添加自定义拦截器，以便在发送的HTTP请求中注入追踪信息。这里的TracingInterceptor类是一个实现ClientHttpRequestInterceptor接口的拦截器，它将当前的追踪ID添加到请求头中。同时，该代码还展示了如何激活SkyWalking的WebFlux、OpenTracing和传统跟踪工具包，使其能够追踪和监控WebFlux应用的请求。