添加RestTemplate支持透传链路信息

Author: hengboy

api-boot-project/api-boot-autoconfigure/src/main/java/org/minbox/framework/api/boot/autoconfigure/logging/ApiBootLoggingAutoConfiguration.java

@@ -61,6 +61,7 @@
         ApiBootLoggingAdminDiscoveryAutoConfiguration.class,
         ApiBootLoggingAdminAppointAutoConfiguration.class,
         ApiBootLoggingOpenfeignAutoConfiguration.class,
+        ApiBootLoggingRestTemplateAutoConfiguration.class,
         ApiBootLoggingWebAutoConfiguration.class
 })
 public class ApiBootLoggingAutoConfiguration {

api-boot-project/api-boot-autoconfigure/src/main/java/org/minbox/framework/api/boot/autoconfigure/logging/ApiBootLoggingRestTemplateAutoConfiguration.java

@@ -0,0 +1,55 @@
+/*
+ * Copyright [2019] [恒宇少年 - 于起宇]
+ *
+ *      Licensed under the Apache License, Version 2.0 (the "License");
+ *      you may not use this file except in compliance with the License.
+ *      You may obtain a copy of the License at
+ *
+ *          http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *      Unless required by applicable law or agreed to in writing, software
+ *      distributed under the License is distributed on an "AS IS" BASIS,
+ *      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *      See the License for the specific language governing permissions and
+ *      limitations under the License.
+ *
+ */
+
+package org.minbox.framework.api.boot.autoconfigure.logging;
+
+import org.minbox.framework.logging.client.http.rest.LoggingRestTemplateInterceptor;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.http.client.ClientHttpRequestInterceptor;
+import org.springframework.util.ObjectUtils;
+import org.springframework.web.client.RestTemplate;
+
+import java.util.Arrays;
+import java.util.List;
+
+/**
+ * ApiBoot Logging ResTemplate Config
+ * {@link RestTemplate}
+ * Setting Interceptor Transmit Link Information
+ *
+ * @author 恒宇少年
+ */
+@Configuration
+@ConditionalOnClass(RestTemplate.class)
+@ConditionalOnBean(RestTemplate.class)
+public class ApiBootLoggingRestTemplateAutoConfiguration {
+
+    public ApiBootLoggingRestTemplateAutoConfiguration(RestTemplate restTemplate) {
+
+        List<ClientHttpRequestInterceptor> interceptors = restTemplate.getInterceptors();
+
+        LoggingRestTemplateInterceptor interceptor = new LoggingRestTemplateInterceptor();
+
+        if (ObjectUtils.isEmpty(interceptors)) {
+            restTemplate.setInterceptors(Arrays.asList(interceptor));
+        } else {
+            interceptors.add(interceptor);
+        }
+    }
+}

api-boot-samples/api-boot-sample-logging/src/main/java/org/minbox/framework/api/boot/sample/logging/ApiBootLoggingApplication.java

@@ -21,6 +21,9 @@
 import org.slf4j.LoggerFactory;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.context.annotation.Bean;
+import org.springframework.web.client.RestTemplate;
 
 /**
  * 日志示例
@@ -34,7 +37,7 @@
  * GitHub：https://github.com/hengboy
  */
 @SpringBootApplication
-public class ApiBootLoggingApplication{
+public class ApiBootLoggingApplication {
     /**
      * logger instance
      */
@@ -44,4 +47,15 @@ public static void main(String[] args) {
         SpringApplication.run(ApiBootLoggingApplication.class, args);
         logger.info("{}服务启动成功.", "ApiBoot Logging Sample");
     }
+
+    /**
+     * 测试使用RestTemplate来透传链路信息
+     *
+     * @return
+     */
+    @Bean
+    @ConditionalOnMissingBean
+    public RestTemplate restTemplate() {
+        return new RestTemplate();
+    }
 }

api-boot-samples/api-boot-sample-logging/src/main/java/org/minbox/framework/api/boot/sample/logging/IndexController.java

@@ -18,9 +18,11 @@
 package org.minbox.framework.api.boot.sample.logging;
 
 import lombok.Data;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.client.RestTemplate;
 
 /**
  * @author：恒宇少年 - 于起宇
@@ -34,11 +36,12 @@
 @RestController
 public class IndexController {
 
+    @Autowired
+    private RestTemplate restTemplate;
+
     @PostMapping(value = "/index")
     public User index(@RequestBody User user) throws Exception {
-        /*HttpEntity<String> httpEntity = new HttpEntity(JSON.toJSONString(user));
-        ResponseEntity<String> result = restTemplate.postForEntity("http://localhost:8080/index", httpEntity, String.class);
-        System.out.println(result.getBody());*/
+        System.out.println(restTemplate.getInterceptors());
         return user;
     }