Hystrix MDC日志链路优化

Author: xiaolyuh123

pom.xml

@@ -12,7 +12,7 @@
 	<parent>
 		<groupId>org.springframework.boot</groupId>
 		<artifactId>spring-boot-starter-parent</artifactId>
-		<version>1.5.3.RELEASE</version>
+		<version>2.0.6.RELEASE</version>
 		<relativePath /> <!-- lookup parent from repository -->
 	</parent>
 

spring-boot-student-log/pom.xml

@@ -41,6 +41,17 @@
             <artifactId>fastjson</artifactId>
             <version>1.2.47</version>
         </dependency>
+
+        <dependency>
+            <groupId>com.netflix.hystrix</groupId>
+            <artifactId>hystrix-core</artifactId>
+            <version>1.5.12</version>
+        </dependency>
+        <dependency>
+            <groupId>com.netflix.hystrix</groupId>
+            <artifactId>hystrix-javanica</artifactId>
+            <version>1.5.12</version>
+        </dependency>
     </dependencies>
 
 </project>

spring-boot-student-log/src/main/java/com/xiaolyuh/hystrix/HystrixConfig.java

@@ -0,0 +1,24 @@
+package com.xiaolyuh.hystrix;
+
+import com.netflix.hystrix.contrib.javanica.aop.aspectj.HystrixCommandAspect;
+import com.netflix.hystrix.strategy.HystrixPlugins;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import javax.annotation.PostConstruct;
+
+@Configuration
+public class HystrixConfig {
+
+    //用来拦截处理HystrixCommand注解
+    @Bean
+    public HystrixCommandAspect hystrixAspect() {
+        return new HystrixCommandAspect();
+    }
+
+    @PostConstruct
+    public void init() {
+        HystrixPlugins.getInstance().registerConcurrencyStrategy(new MdcHystrixConcurrencyStrategy());
+    }
+
+}

spring-boot-student-log/src/main/java/com/xiaolyuh/hystrix/MdcHystrixConcurrencyStrategy.java

@@ -0,0 +1,47 @@
+package com.xiaolyuh.hystrix;
+
+import com.netflix.hystrix.strategy.concurrency.HystrixConcurrencyStrategy;
+import org.slf4j.MDC;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.Callable;
+
+/**
+ * Hystrix线程池隔离支持日志链路跟踪
+ *
+ * @author yuhao.wang3
+ */
+public class MdcHystrixConcurrencyStrategy extends HystrixConcurrencyStrategy {
+
+    @Override
+    public <T> Callable<T> wrapCallable(Callable<T> callable) {
+        return new MdcAwareCallable(callable, MDC.getCopyOfContextMap());
+    }
+
+    /**
+     * 装饰原有的Callable，用来实现MDC容器内容的拷贝
+     * @param <T>
+     */
+    private class MdcAwareCallable<T> implements Callable<T> {
+
+        private final Callable<T> delegate;
+
+        private final Map<String, String> contextMap;
+
+        public MdcAwareCallable(Callable<T> callable, Map<String, String> contextMap) {
+            this.delegate = callable;
+            this.contextMap = contextMap != null ? contextMap : new HashMap();
+        }
+
+        @Override
+        public T call() throws Exception {
+            try {
+                MDC.setContextMap(contextMap);
+                return delegate.call();
+            } finally {
+                MDC.clear();
+            }
+        }
+    }
+}