组入brave配置开关

brave-spring-boot-starter/src/main/java/cn/quantgroup/tech/brave/configuration/BraveAutoConfiguration.java

@@ -18,6 +18,7 @@ import okhttp3.OkHttpClient;
 import org.apache.http.impl.client.HttpClientBuilder;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.autoconfigure.AutoConfigureOrder;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -46,108 +47,157 @@ import java.util.List;
 @Slf4j
 @Configuration
 @AutoConfigureOrder
-@EnableWebMvc
-@EnableConfigurationProperties( BraveProperties.class )
-@Import( { TracingClientHttpRequestInterceptor.class, SpanCustomizingAsyncHandlerInterceptor.class } )
-public class BraveAutoConfiguration extends WebMvcConfigurerAdapter {
-    @Autowired
-    private BraveProperties braveProperties;
+public class BraveAutoConfiguration {
+    @Configuration
+    @EnableWebMvc
+    @EnableConfigurationProperties( BraveProperties.class )
+    @ConditionalOnProperty( prefix = "tech.brave", name = "enabled", havingValue = "false")
+    public static class BraveUntEnabled {
+        /**
+         * 提供HttpClientBuilderBean
+         *
+         * @return
+         */
+        @Bean( name = "techHttpClientBuilder" )
+        public HttpClientBuilder httpClientBuilder() {
+            log.info("注册HttpClientBuilder");
+            return TracingHttpClientBuilder.create();
+        }
+
+        /**
+         * 提供OkHttpClientBean
+         *
+         * @return
+         */
+        @Bean( name = "techOkHttpClientBuilder" )
+        public OkHttpClient.Builder okHttpClientBuilder() {
+            log.info("注册OkHttpClient.Builder");
+            return new OkHttpClient.Builder();
+        }
+
+        /**
+         * 提供RestTemplate
+         *
+         * @return
+         */
+        @Bean( name = "techRestTemplate" )
+        public RestTemplate restTemplate() {
+            log.info("添加restTemplate拦截器");
+            return new RestTemplate();
+        }
 
-    @Autowired
-    private TracingClientHttpRequestInterceptor clientInterceptor;
-
-
-    @Autowired
-    private SpanCustomizingAsyncHandlerInterceptor spanCustomizingAsyncHandlerInterceptor;
-
-    @Bean
-    Sender sender() {
-        return KafkaSender.newBuilder().bootstrapServers(braveProperties.getKafkaHost()).topic(braveProperties.getKafkaTopic()).encoding(Encoding.JSON).build();
-    }
-
-    @Bean
-    AsyncReporter<Span> spanReporter() {
-        return AsyncReporter.create(sender());
-    }
-
-    @Bean
-    Tracing tracing() {
-        return Tracing.newBuilder()
-                .sampler(brave.sampler.Sampler.create(braveProperties.getSample()))
-                .localServiceName(braveProperties.getServiceName())
-                .propagationFactory(ExtraFieldPropagation.newFactory(B3Propagation.FACTORY, "user-name"))
-                .currentTraceContext(ThreadContextCurrentTraceContext.create())
-                .spanReporter(spanReporter()).build();
-    }
-
-    @Bean
-    public HttpTracing httpTracing(Tracing tracing) {
-        return HttpTracing.create(tracing);
-    }
-
-    /**
-     * 提供RabbitTemplateBuilder,业务系统需要使用该Builder生成RabbitTemplate
-     * @param tracing
-     * @return
-     */
-    @Bean
-    public SpringRabbitTracing.Builder springRabbitTracingBuilder(Tracing tracing) {
-        return SpringRabbitTracing.newBuilder(tracing);
-    }
-
-    /**
-     * 提供HttpClientBuilderBean
-     * @param tracing
-     * @return
-     */
-    @Bean( name = "techHttpClientBuilder" )
-    public HttpClientBuilder httpClientBuilder(Tracing tracing) {
-        log.info("注册HttpClientBuilder");
-        return TracingHttpClientBuilder.create(tracing);
-    }
-
-    /**
-     * 提供OkHttpClientBean
-     * @param httpTracing
-     * @return
-     */
-    @Bean( name = "techOkHttpClientBuilder" )
-    public OkHttpClient.Builder okHttpClientBuilder(HttpTracing httpTracing) {
-        log.info("注册OkHttpClient.Builder");
-        return new OkHttpClient.Builder()
-                .dispatcher(new Dispatcher(httpTracing.tracing().currentTraceContext().executorService(new Dispatcher().executorService())))
-                .addNetworkInterceptor(TracingInterceptor.create(httpTracing));
-    }
-
-    /**
-     * 创建一个bean
-     *
-     * @return
-     */
-    @Bean( name = "techDelegatingTracingFilter" )
-    public Filter delegatingTracingFilter() {
-        return new DelegatingTracingFilter();
-    }
-
-    /**
-     * 提供RestTemplate
-     * @return
-     */
-    @Bean( name = "techRestTemplate")
-    public RestTemplate restTemplate() {
-        log.info("添加restTemplate拦截器");
-        RestTemplate restTemplate = new RestTemplate();
-        List<ClientHttpRequestInterceptor> interceptors =
-                new ArrayList<>(restTemplate.getInterceptors());
-        interceptors.add(clientInterceptor);
-        restTemplate.setInterceptors(interceptors);
-        return restTemplate;
     }
 
-    @Override
-    public void addInterceptors(InterceptorRegistry registry) {
-        log.info("添加webmvc拦截器");
-        registry.addInterceptor(spanCustomizingAsyncHandlerInterceptor);
+    @Configuration
+    @EnableWebMvc
+    @EnableConfigurationProperties( BraveProperties.class )
+    @ConditionalOnProperty( prefix = "tech.brave", name = "enabled", havingValue = "true" )
+    @Import( {TracingClientHttpRequestInterceptor.class, SpanCustomizingAsyncHandlerInterceptor.class} )
+    public static class BraveEnabled extends WebMvcConfigurerAdapter {
+        @Autowired
+        private BraveProperties braveProperties;
+
+        @Autowired
+        private TracingClientHttpRequestInterceptor clientInterceptor;
+
+
+        @Autowired
+        private SpanCustomizingAsyncHandlerInterceptor spanCustomizingAsyncHandlerInterceptor;
+
+        @Bean
+        Sender sender() {
+            return KafkaSender.newBuilder().bootstrapServers(braveProperties.getKafkaHost()).topic(braveProperties.getKafkaTopic()).encoding(Encoding.JSON).build();
+        }
+
+        @Bean
+        AsyncReporter<Span> spanReporter() {
+            return AsyncReporter.create(sender());
+        }
+
+        @Bean
+        Tracing tracing() {
+            return Tracing.newBuilder()
+                    .sampler(brave.sampler.Sampler.create(braveProperties.getSample()))
+                    .localServiceName(braveProperties.getServiceName())
+                    .propagationFactory(ExtraFieldPropagation.newFactory(B3Propagation.FACTORY, "user-name"))
+                    .currentTraceContext(ThreadContextCurrentTraceContext.create())
+                    .spanReporter(spanReporter()).build();
+        }
+
+        @Bean
+        public HttpTracing httpTracing(Tracing tracing) {
+            return HttpTracing.create(tracing);
+        }
+
+        /**
+         * 提供RabbitTemplateBuilder,业务系统需要使用该Builder生成RabbitTemplate
+         *
+         * @param tracing
+         * @return
+         */
+        @Bean
+        public SpringRabbitTracing.Builder springRabbitTracingBuilder(Tracing tracing) {
+            return SpringRabbitTracing.newBuilder(tracing);
+        }
+
+        /**
+         * 提供HttpClientBuilderBean
+         *
+         * @param tracing
+         * @return
+         */
+        @Bean( name = "techHttpClientBuilder" )
+        public HttpClientBuilder httpClientBuilder(Tracing tracing) {
+            log.info("注册braveHttpClientBuilder");
+            return TracingHttpClientBuilder.create(tracing);
+        }
+
+        /**
+         * 提供OkHttpClientBean
+         *
+         * @param httpTracing
+         * @return
+         */
+        @Bean( name = "techOkHttpClientBuilder" )
+        public OkHttpClient.Builder okHttpClientBuilder(HttpTracing httpTracing) {
+            log.info("注册braveOkHttpClient.Builder");
+            return new OkHttpClient.Builder()
+                    .dispatcher(new Dispatcher(httpTracing.tracing().currentTraceContext().executorService(new Dispatcher().executorService())))
+                    .addNetworkInterceptor(TracingInterceptor.create(httpTracing));
+        }
+
+        /**
+         * 创建一个bean
+         *
+         * @return
+         */
+        @Bean( name = "techDelegatingTracingFilter" )
+        public Filter delegatingTracingFilter() {
+            log.info("注册DelegatingTracingFilter");
+            return new DelegatingTracingFilter();
+        }
+
+        /**
+         * 提供RestTemplate
+         *
+         * @return
+         */
+        @Bean( name = "techRestTemplate" )
+        public RestTemplate restTemplate() {
+            log.info("添加braveRestTemplate拦截器");
+            RestTemplate restTemplate = new RestTemplate();
+            List<ClientHttpRequestInterceptor> interceptors =
+                    new ArrayList<>(restTemplate.getInterceptors());
+            interceptors.add(clientInterceptor);
+            restTemplate.setInterceptors(interceptors);
+            return restTemplate;
+        }
+
+        @Override
+        public void addInterceptors(InterceptorRegistry registry) {
+            log.info("添加braveWebmvc拦截器");
+            registry.addInterceptor(spanCustomizingAsyncHandlerInterceptor);
+        }
     }
 
 }