lib/remnant/base.rb in remnant-0.2.1 vs lib/remnant/base.rb in remnant-0.2.2

@@ -11,25 +11,33 @@
     def handler
       @handler ||= Statsd.new(Remnant.configuration.hostname, Remnant.configuration.port_number)
     end
 
     def collect
+      @sample_counter ||= 0
+
       extra_remnant_key = Remnant::Discover.results.delete(:extra_remnant_key)
 
       if ::Rails.env.production? || ::Rails.env.staging? || ::Rails.env.demo?
-        # send on
-        Remnant::Discover.results.map do |remnant_key, ms|
-          key = [
-                 Remnant.configuration.tag,
-                 Remnant.configuration.env,
-                 extra_remnant_key,
-                 remnant_key
-                ].compact.join('.')
+        # only log if above sample rate
+        if @sample_counter > configuration.sample_rate
+          Remnant::Discover.results.map do |remnant_key, ms|
+            key = [
+                   Remnant.configuration.tag,
+                   Remnant.configuration.env,
+                   extra_remnant_key,
+                   remnant_key
+                  ].compact.join('.')
 
-          Remnant.handler.timing(key, ms.to_i)
+            Remnant.handler.timing(key, ms.to_i)
+          end
+
+          @sample_counter = 0
+        else
+          @sample_counter += 1
         end
       else
-        # log it
+        # always log in development mode
         Rails.logger.info "--------------Remnants Discovered--------------"
 
         Remnant::Discover.results.map do |remnant_key, ms|
           key = [
                  extra_remnant_key,