上一篇文章简单的介绍了如何把zipkin server、brave插件跑起来,这篇文章介绍一下如何把阿里云Ons/RocketMQ添加到链路跟踪里面来
背景
RocketMQ为阿里巴巴开源的一款消息中间件,阿里云的Ons服务可以看做成商业闭源版RocketMQ。为了叙述方便,下面统称RocketMQ
。
zipkin官方出品的brave已经包含了很多中间件插件了,比如dubbo
、okhttp3
、kafka
等,但是在国内比较常用的RocketMQ没有,尤其是部署在阿里云上的应用大部分都使用了RocketMQ作为消息中间件。
作为链路中非常重要的一环,不能让链路信息在此中断,让我们研究一下如何为RocketMQ增加链路跟踪能力吧。恰好RocketMQ为开发扩展提供了支持。
实现
这一章节就来聊一聊如何给RocketMQ添加链路跟踪支持。不过开始之前先来了解下brave是如何对链路信息就行传播的吧。
brave的传播逻辑
在介绍brave传播逻辑之前,先想一下传递链路信息应该有哪些步骤:
- 上游封装当前链路信息并放到载体中
- 载体将链路信息传递到下游
- 下游解析载体中的链路信息并放到链路上下文
注:载体可以是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.