添加采样率配置

jgchen · Mar 6, 2017 · c9f22a8 · c9f22a8
1 parent f7fe731
commit c9f22a8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/README.MD b/README.MD
@@ -13,11 +13,11 @@
 
    * 打印日志方式
 ```xml
-    <bean id="brave" class="com.github.kristofa.brave.dubbo.BraveFactoryBean" p:serviceName="notify" />
+        <bean id="brave" class="com.github.kristofa.brave.dubbo.BraveFactoryBean" p:serviceName="serviceName" p:zipkinHost="" p:rate="1.0" />
 ```
    * http方式
 ```xml
- <bean id="brave" class="com.github.kristofa.brave.dubbo.BraveFactoryBean" p:serviceName="serviceName" p:zipkinHost="http://zipkinip:9411/" />
+        <bean id="brave" class="com.github.kristofa.brave.dubbo.BraveFactoryBean" p:serviceName="serviceName" p:zipkinHost="http://10.168.16.111:9411/" p:rate="1.0" />
 ```
 
 

diff --git a/src/main/java/com/github/kristofa/brave/dubbo/BraveFactoryBean.java b/src/main/java/com/github/kristofa/brave/dubbo/BraveFactoryBean.java
@@ -16,6 +16,8 @@ public class BraveFactoryBean implements FactoryBean<Brave> {
     private String serviceName;
     /**zipkin服务器ip及端口，不配置默认打印日志*/
     private String zipkinHost;
+    /**采样率 0~1 之间*/
+    private float rate = 1.0f;
     /**单例模式*/
     private Brave instance;
 
@@ -31,6 +33,14 @@ public String getZipkinHost() {
         return zipkinHost;
     }
 
+    public String getRate() {
+        return String.valueOf(rate);
+    }
+
+    public void setRate(String rate) {
+        this.rate = Float.parseFloat(rate);
+    }
+
     public void setZipkinHost(String zipkinHost) {
         this.zipkinHost = zipkinHost;
     }
@@ -41,11 +51,11 @@ private void createInstance() {
         }
         Brave.Builder builder = new Brave.Builder(this.serviceName);
         if (this.zipkinHost != null && !"".equals(this.zipkinHost)) {
-            builder.spanCollector(HttpSpanCollector.create(this.zipkinHost, new EmptySpanCollectorMetricsHandler()));
-            LOGGER.info("brave dubbo config collect whith httpSpanColler");
+            builder.spanCollector(HttpSpanCollector.create(this.zipkinHost, new EmptySpanCollectorMetricsHandler())).traceSampler(Sampler.create(rate)).build();
+            LOGGER.info("brave dubbo config collect whith httpSpanColler , rate is "+ rate);
         }else{
-            builder.spanCollector(new LoggingSpanCollector()).traceSampler(Sampler.create(1.0f)).build();
-            LOGGER.info("brave dubbo config collect whith loggingSpanColletor");
+            builder.spanCollector(new LoggingSpanCollector()).traceSampler(Sampler.create(rate)).build();
+            LOGGER.info("brave dubbo config collect whith loggingSpanColletor , rate is "+ rate);
         }
         this.instance = builder.build();
     }