lib/statsd/instrument.rb in statsd-instrument-1.3.0 vs lib/statsd/instrument.rb in statsd-instrument-1.3.1

@@ -13,11 +13,11 @@
   class << self
     attr_accessor :host, :port, :mode, :logger, :enabled, :default_sample_rate,
                   :prefix, :implementation
   end
   self.enabled = true
-  self.default_sample_rate = 1
+  self.default_sample_rate = 1.0
   self.implementation = :statsd
 
   TimeoutClass = defined?(::SystemTimer) ? ::SystemTimer : ::Timeout
 
   # StatsD.server = 'localhost:1234'
@@ -98,16 +98,16 @@
       alias_method method, method_name_with_statsd
     end
   end
 
   # glork:320|ms
-  def self.measure(key, milli = nil)
+  def self.measure(key, milli = nil, sample_rate = default_sample_rate)
     result = nil
     ms = milli || Benchmark.ms do
       result = yield
     end
 
-    write(key, ms, :ms)
+    write(key, ms, :ms, sample_rate)
     result
   end
 
   # gorets:1|c
   def self.increment(key, delta = 1, sample_rate = default_sample_rate)