背景
开发排查系统问题用得最多的手段就是查看系统日志，在分布式环境中一般使用ELK来统一收集日志，但是在并发大时使用日志定位问题还是比较麻烦，由于大量的其他用户/其他线程的日志也一起输出穿行其中导致很难筛选出指定请求的全部相关日志，以及下游线程/服务对应的日志。
解决思路
每个请求都使用一个唯一标识来追踪全部的链路显示在日志中，并且不修改原有的打印方式(代码无入侵) 使用Logback的MDC机制日志模板中加入traceId标识，取值方式为%X{traceId}
MDC（Mapped Diagnostic Context，映射调试上下文）是 log4j 和 logback 提供的一种方便在多线程条件下记录日志的功能。MDC 可以看成是一个与当前线程绑定的Map，可以往其中添加键值对。MDC 中包含的内容可以被同一线程中执行的代码所访问。当前线程的子线程会继承其父线程中的 MDC 的内容。当需要记录日志时，只需要从 MDC 中获取所需的信息即可。MDC 的内容则由程序在适当的时候保存进去。对于一个 Web 应用来说，通常是在请求被处理的最开始保存这些数据。
方案实现
由于MDC内部使用的是ThreadLocal所以只有本线程才有效，子线程和下游的服务MDC里的值会丢失；所以方案主要的难点是解决值的传递问题，主要包括以几下部分：
API网关中的MDC数据如何传递给下游服务
服务如何接收数据，并且调用其他远程服务时如何继续传递
异步的情况下(线程池)如何传给子线程
修改日志模板
logback配置文件模板格式添加标识%X{traceId}
网关添加过滤器
生成traceId并通过header传递给下游服务
@ComponentpublicclassTraceFilterextendsZuulFilter{@AutowiredprivateTraceProperties traceProperties;@OverridepublicStringfilterType(){returnFilterConstants.PRE_TYPE; }@OverridepublicintfilterOrder(){returnFORM_BODY_WRAPPER_FILTER_ORDER -1; }@OverridepublicbooleanshouldFilter(){//根据配置控制是否开启过滤器returntraceProperties.getEnable(); }@OverridepublicObjectrun(){//链路追踪idString traceId = IdUtil.fastSimpleUUID(); MDC.put(CommonConstant.LOG_TRACE_ID, traceId); RequestContext ctx = RequestContext.getCurrentContext(); ctx.addZuulRequestHeader(CommonConstant.TRACE_ID_HEADER, traceId);returnnull; }}
下游服务增加spring拦截器
接收并保存traceId的值拦截器
publicclassTraceInterceptorimplementsHandlerInterceptor{@OverridepublicbooleanpreHandle(HttpServletRequest request, HttpServletResponse response, Objecthandler){ String traceId = reques.getHeader(CommonConstant.TRACE_ID_HEADER);if(StrUti.isNotEmpty(traceId)) { MDC.put(CommonConstant.LOG_TRACE_ID, traceId); }returntrue; }}
注册拦截器
publicclassDefaultWebMvcConfigextendsWebMvcConfigurationSupport{@Overrideprotectedvoid addInterceptors(InterceptorRegistryregistry) {//日志链路追踪拦截器registry.addInterceptor(newTraceInterceptor()).addPathPatterns("/**");super.addInterceptors(registry); }}
下游服务增加feign拦截器
继续把当前服务的traceId值传递给下游服务
publicclassFeignInterceptorConfig{@BeanpublicRequestInterceptorrequestInterceptor(){ RequestInterceptor requestInterceptor =template-> {//传递日志traceIdString traceId = MDC.get(CommonConstant.LOG_TRACE_ID);if(StrUtil.isNotEmpty(traceId)) {template.header(CommonConstant.TRACE_ID_HEADER, traceId); } }; retur requestInterceptor; }}
解决父子线程传递问题
主要针对业务会使用线程池(异步、并行处理)，并且spring自己也有@Async注解来使用线程池，要解决这个问题需要以下两个步骤
重写logback的LogbackMDCAdapter
由于logback的MDC实现内部使用的是ThreadLocal不能传递子线程，所以需要重写替换为阿里的TransmittableThreadLocal
TransmittableThreadLocal 是Alibaba开源的、用于解决 “在使用线程池等会缓存线程的组件情况下传递ThreadLocal” 问题的 InheritableThreadLocal 扩展。若希望 TransmittableThreadLocal 在线程池与主线程间传递，需配合 TtlRunnable 和 TtlCallable 使用。
TtlMDCAdapter类
packageorg.slf4j;importcom.alibaba.ttl.TransmittableThreadLocal;importorg.slf4j.spi.MDCAdapter;publicclassTtlMDCAdapterimplementsMDCAdapter{/*
* * 此处是关键
*/privatefinal ThreadLocal> copyOnInheritThreadLocal = newTransmittableThreadLocal<>();privatestaticTtlMDCAdapter mtcMDCAdapter;static{ mtcMDCAdapter = newTtlMDCAdapter(); MDC.mdcAdapter = mtcMDCAdapter; }publicstaticMDCAdapter getInstance() {returnmtcMDCAdapter; }
其他代码与ch.qos.logback.classic.util.LogbackMDCAdapter一样，只需改为调用copyOnInheritThreadLocal变量
TtlMDCAdapterInitializer类用于程序启动时加载自己的mdcAdapter实现
publicclassTtlMDCAdapterInitializerimplementsApplicationContextInitializer{@Overridepublicvoidinitialize(ConfigurableApplicationContext applicationContext){//加载TtlMDCAdapter实例TtlMDCAdapter.getInstance(); }}
扩展线程池实现
增加TtlRunnable和TtlCallable扩展实现TTL
publicclassCustomThreadPoolTaskExecutorextendsThreadPoolTaskExecutor{@Overridepublicvoid execute(Runnable runnable) { Runnable ttlRunnable = TtlRunnable.get(runnable);super.execute(ttlRunnable); }@Overridepublic Future submit(Callable task) { Callable ttlCallable = TtlCallable.get(task);returnsuper.submit(ttlCallable); }@OverridepublicFuture submit(Runnable task) { Runnable ttlRunnable = TtlRunnable.get(task);returnsuper.submit(ttlRunnable); }@OverridepublicListenableFuture submitListenable(Runnable task) { Runnable ttlRunnable = TtlRunnable.get(task);returnsuper.submitListenable(ttlRunnable); }@Overridepublic ListenableFuture submitListenable(Callable task) { Callable ttlCallable = TtlCallable.get(task);returnsuper.submitListenable(ttlCallable); }}
场景测试
测试代码如下
api网关打印的日志
网关生成traceId值为13d9800c8c7944c78a06ce28c36de670
请求跳转到文件服务时打印的日志
显示的traceId与网关相同，这里特意模拟发生异常的场景
ELK聚合日志通过traceId查询整条链路日志
当系统出现异常时，可直接通过该异常日志的traceId的值，在日志中心中询该请求的所有日志信息
免费分享Java学习资料，Java技术资料，需要的朋友在关注后私信我即可
原文：https://mp.weixin.qq.com/s/NG-RIPDBkJsH3qo6_rJGWQ
作者： 陶陶技术笔记
来源：微信公众号
网友评论