fluent-plugin-dogstatsd 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 565969556dacf436172ed012b9a5eaed2c29d948
-  data.tar.gz: 491155caaad84b0c9be52a476d2166ac456d3967
+  metadata.gz: bb0069527436a3f8212e8656e72708cb57050244
+  data.tar.gz: 121fc2d3d1faf0dc90cafde23438776e2f921fe0
 SHA512:
-  metadata.gz: ddc614f843d34a7ae87f40638fc8894ddc7cb65059c83c67a94f8e3f774cd7aec7f24b118628328270e6488279363c8553e4390a212ad3e5c1a4706e86832005
-  data.tar.gz: 33c2369532d6ed1d1437becf77f89904968f90e37c91c021dcae3b0a891491aac9b1775a0d7367a7f61721e446bf0f163aea660fb0d7addd0e5d0cf6f096bc04
+  metadata.gz: 76b195e133d9cb9bb1c2bfb1d496cbcd8b558218fa96c49131671269f9f0e01217ac28eba702f36696dbeeef9c88a217335d0426e0d9723e14e2d2485c66c05f
+  data.tar.gz: d6fd640212d1a9e5a7825f3e00947dbd21e67909dd07e894e8eec38565c4bb6db9b17c52bdf62c07689eb2af521cb0d10e58accb1b967b2e1e8ae2d6487f096e

data/lib/fluent/plugin/dogstatsd/version.rb

@@ -1,7 +1,7 @@
 module Fluent
   module Plugin
     module Dogstatsd
-      VERSION = "0.0.3"
+      VERSION = "0.0.4"
     end
   end
 end

data/lib/fluent/plugin/out_dogstatsd.rb

@@ -6,9 +6,10 @@ module Fluent
     config_param :port, :integer, :default => nil
     config_param :use_tag_as_key, :bool, :default => false
     config_param :flat_tags, :bool, :default => false
-    config_param :flat_tag, :bool, :default => false # obsolute
+    config_param :flat_tag, :bool, :default => false # obsolete
     config_param :metric_type, :string, :default => nil
     config_param :value_key, :string, :default => nil
+    config_param :sample_rate, :float, :default => nil
 
     unless method_defined?(:log)
       define_method(:log) { $log }
@@ -52,17 +53,20 @@ module Fluent
           value = record.delete(@value_key || 'value')
 
           options = {}
+          title = record.delete('title')
+          text  = record.delete('text')
+          type  = @metric_type || record.delete('type')
+          sample_rate = @sample_rate || record.delete('sample_rate')
+
+          if sample_rate
+            options[:sample_rate] = sample_rate
+          end
 
           tags = if @flat_tags || @flat_tag
                    record
                  else
                    record['tags']
                  end
-
-          title = record.delete('title')
-          text  = record.delete('text')
-          type  = @metric_type || record.delete('type')
-
           if tags
             options[:tags] = tags.map do |k, v|
               "#{k}:#{v}"

data/test/plugin/test_out_dogstatsd.rb

@@ -117,6 +117,30 @@ use_tag_as_key true
     ])
   end
 
+  def test_sample_rate_config
+    d = create_driver(<<-EOC)
+#{default_config}
+sample_rate .5
+    EOC
+
+    d.emit({'type' => 'increment'}, Time.now.to_i)
+    d.run
+
+    assert_equal(d.instance.statsd.messages, [
+      [:increment, 'dogstatsd.tag', {sample_rate: 0.5}],
+    ])
+  end
+
+  def test_sample_rate
+    d = create_driver
+    d.emit({'type' => 'increment', 'sample_rate' => 0.5}, Time.now.to_i)
+    d.run
+
+    assert_equal(d.instance.statsd.messages, [
+      [:increment, 'dogstatsd.tag', {sample_rate: 0.5}],
+    ])
+  end
+
   private
   def default_config
     <<-EOC

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fluent-plugin-dogstatsd
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Ryota Arai
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-10-29 00:00:00.000000000 Z
+date: 2015-11-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: fluentd