zipkin：为阿里云Ons/RocketMQ加上链路跟踪（二）

上一篇文章简单的介绍了如何把zipkin server、brave插件跑起来，这篇文章介绍一下如何把阿里云Ons/RocketMQ添加到链路跟踪里面来

背景

RocketMQ为阿里巴巴开源的一款消息中间件，阿里云的Ons服务可以看做成商业闭源版RocketMQ。为了叙述方便，下面统称RocketMQ。

zipkin官方出品的brave已经包含了很多中间件插件了，比如dubbo、okhttp3、kafka等，但是在国内比较常用的RocketMQ没有，尤其是部署在阿里云上的应用大部分都使用了RocketMQ作为消息中间件。

作为链路中非常重要的一环，不能让链路信息在此中断，让我们研究一下如何为RocketMQ增加链路跟踪能力吧。恰好RocketMQ为开发扩展提供了支持。

实现

这一章节就来聊一聊如何给RocketMQ添加链路跟踪支持。不过开始之前先来了解下brave是如何对链路信息就行传播的吧。

brave的传播逻辑

在介绍brave传播逻辑之前，先想一下传递链路信息应该有哪些步骤：

1. 上游封装当前链路信息并放到载体中
2. 载体将链路信息传递到下游
3. 下游解析载体中的链路信息并放到链路上下文

注：载体可以是http的header、rpc的attachment等任何可以传递链路信息的传递途径。

来看一下官方如何为dubbo链路添加追踪信息的吧：
brave.dubbo.rpc.TracingFilter.java

//提取&解析载体链路信息逻辑
TraceContext.Extractor<DubboServerRequest> extractor =  tracing.propagation().extractor(GETTER);
//封装当前链路到载体
TraceContext.Injector<DubboClientRequest> injector = tracing.propagation().injector(SETTER);

@Override 
public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcExcept
  if (!isInit) return invoker.invoke(invocation);
  RpcContext rpcContext = RpcContext.getContext();
  //判断是server还是client端
  Kind kind = rpcContext.isProviderSide() ? Kind.SERVER : Kind.CLIENT;
  final Span span;
  if (kind.equals(Kind.CLIENT)) {
   
  	//若client端，则对链路信息进行提取并封装到attachment中
    Map<String, String> attachments = RpcContext.getContext().getAttachments();
    DubboClientRequest request = new DubboClientRequest(invocation, attachments);
    span = tracer.nextSpan(clientSampler, request);
    //将链路信息注入到载体attachment中
    injector.inject(span.context(), request);
  } else {
   
  	//若是server端，则解析载体中的链路信息并注入到上下文中
    DubboServerRequest request = new DubboServerRequest(invocation, invocation.getAttachme
    TraceContextOrSamplingFlags extracted = extractor.extract(request);
    //解析载体中的链路信息
    span = nextSpan(extracted, request);
  }
  if (!span.isNoop()) {
   
    span.kind(kind);
    String service = invoker.getInterface().getSimpleName();
    String method = RpcUtils.getMethodName(invocation);
    span.name(service + "/" + method);
    parseRemoteAddress(rpcContext, span);
    span.start();
  }
  boolean isOneway = false, deferFinish = false;
  try (CurrentTraceContext.Scope scope = current.newScope(span.context