集成AWSJavaSDK 2. x与Spring Sleuth

Question

我第一次使用Spring Boot，正在为我的应用程序设置分布式跟踪。我已经将Spring Cloud Slueth添加到我的应用程序中，当调用我的endpoint时，我可以看到生成的跨度和跟踪，但是我很难让它与Aws Sdk 2. x集成（使用Dynamo异步客户端）。我有几个关于集成的问题：

1. 通过aws sdk跟踪http调用的最佳方式是什么。我能做到这一点的唯一方法是实现ExecutionInterceptor并手动创建一个跨度：

@Component
@Log4j2
public class AwsExecutionInterceptor implements ExecutionInterceptor {
    public static final String CLIENT_CALL_SPAN = "ClientCallSpan";
    public static final ExecutionAttribute<Span> ATTRIBUTE = new ExecutionAttribute<>(
            CLIENT_CALL_SPAN);
    private final Tracer tracer;

    @Autowired
    public AwsExecutionInterceptor(Tracer tracer) {
        this.tracer = tracer;
    }

    @Override
    public void beforeExecution(Context.BeforeExecution context, ExecutionAttributes executionAttributes) {

        Span clientCallSpan = this.tracer.nextSpan().name("clientCall").start();
        log.info("Starting new span");
        executionAttributes.putAttribute(ATTRIBUTE, clientCallSpan);
    }

    @Override
    public void afterExecution(Context.AfterExecution context, ExecutionAttributes executionAttributes) {
        Span span = executionAttributes.getAttribute(ATTRIBUTE);
        span.end();
        log.info("Span closed");
    }
}

public class AwsExecutorConfig {

    @Bean("AwsClientExecutor")
    public Executor executor() {

        ThreadPoolExecutor executor = new ThreadPoolExecutor(50, 50,
                                                             10, TimeUnit.SECONDS,
                                                             new LinkedBlockingQueue<>(10_000));

        executor.allowCoreThreadTimeOut(true);
        return executor;
    }
}

@Autowired
public DynamoConfig(AwsExecutionInterceptor executionInterceptor,
                    @Qualifier("AwsClientExecutor") Executor executor) {

  @Bean(destroyMethod = "close")
  public DynamoDbAsyncClient dynamoDbAsyncClient() {
    var builder = DynamoDbAsyncClient.builder();
    builder.overrideConfiguration(
        ClientOverrideConfiguration.builder()
                                   .addExecutionInterceptor(executionInterceptor).build());

    builder.asyncConfiguration(
            b -> b.advancedOption(SdkAdvancedAsyncClientOption.FUTURE_COMPLETION_EXECUTOR, 
                                  executor));
    return builder.build();
  }
}

有了这个，我可以从sdk中看到如下日志，

2021-07-20 10:26:04.369 DEBUG [,c9ae653fd2254266,c9ae653fd2254266] 2873 --- [pool-2-thread-5] s.a.a.m.internal.DefaultMetricCollector  : Collected metrics records: MetricCollection(...)

但其他日志没有任何痕迹，例如：

2021-07-20 10:26:04.360 DEBUG [,,] 2873 --- [tyEventLoop-0-2] software.amazon.awssdk.request           : Received successful response: 200
2021-07-20 10:26:04.361 TRACE [,,] 2873 --- [tyEventLoop-0-2] software.amazon.awssdk.request           : Parsing service response JSON.
2021-07-20 10:26:04.361 DEBUG [,,] 2873 --- [tyEventLoop-0-2] software.amazon.awssdk.requestId         : x-amzn-RequestId : af0d9968-3202-480f-9e3c-8312ae6963c8
2021-07-20 10:26:04.361 DEBUG [,,] 2873 --- [tyEventLoop-0-2] software.amazon.awssdk.requestId         : x-amz-id-2 : not available
2021-07-20 10:26:04.367 TRACE [,,] 2873 --- [tyEventLoop-0-2] software.amazon.awssdk.request           : Done parsing service response.

spring cloud sleuth文档提到了与Netty Http客户端的集成，但我不确定如何使用AWS SDK实现这一点。如果您有任何帮助，我们将不胜感激

Answer 1

这是一个非常复杂的问题，让我试着给你一些背景知识，并解释你能做什么/寻找什么。

Spring Cloud Sleuth是对分布式跟踪库（如OpenZipkin的Brave）的抽象。这意味着您可以使用Sleuth的API来检测代码，并且可以在后台更改跟踪库。这也意味着您可以使用自己选择的跟踪库的内置工具（例如，您可以使用Sleuth为代码提供工具，但也可以添加Brave的MySQL工具，这两种工具将协同工作）。

（Brave不是唯一一个由Sleuth支持的追踪库。Sleuth也有OTel支持，但其中一个还不是GA，所以我们不建议在生产中使用。）

由于上述原因，您可以在以下几个地方查看仪器支持：

1. Spring Cloud Sleuth工具有一堆Spring框架支持的东西（AWS不是那样的，Spring Cloud AWS现在是一个社区支持的项目），但是正如你所说的netty客户端是：docs
2. Brave有很多库的仪器支持，如果有什么可以重用的，你可以看看那里（例如：在不同的回购中有一些AWS支持）
3. 库本身可以有Brave的支持（我不认为AWS SDK是今天的情况）

我建议您使用Sleuth API来插入代码，在这里您可以找到如何创建和结束跨度。如您所见，如果您想为第三方库提供仪器，这可能会很棘手，但您可以做以下几件事：

1. 类似于上面使用拦截器所做的事情（这假设有这样一个组件，您可以使用它来注入自己的代码）。仅供参考：您的实现不会创建“范围”（参见文档）（创建和结束跨度）

日志集成通常是通过将跟踪数据放入MDC来实现的，跟踪数据通常是通过html" target="_blank">线程本地传播的。因此，如果您在执行过程中更改线程，您的跟踪数据就会丢失，进入线程池就像从头开始一个全新的跨度（没有父跨度）。幸运的是，Sleuth有一个解决方案，您需要将您的执行器包装到Sleuth提供的组件中，Sleuth将为您将上下文传播到您的线程池中，请参见：线程池支持

请检查文档中的HttpClient部分，它需要是一个bean，以便Sleuth可以对其进行检测。