Added: event_sample_rate tag to SuckerPunch integration.

DataDog · Jan 3, 2019 · f520e88 · f520e88
1 parent 05a7831
commit f520e88
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/ddtrace/contrib/sucker_punch/instrumentation.rb b/lib/ddtrace/contrib/sucker_punch/instrumentation.rb
@@ -1,4 +1,5 @@
 require 'sucker_punch'
+require 'ddtrace/contrib/sampling'
 require 'ddtrace/contrib/sucker_punch/ext'
 
 module Datadog
@@ -18,6 +19,7 @@ def __run_perform(*args)
 
               __with_instrumentation(Ext::SPAN_PERFORM) do |span|
                 span.resource = "PROCESS #{self}"
+                Contrib::Sampling.set_event_sample_rate(span, datadog_configuration[:event_sample_rate])
                 __run_perform_without_datadog(*args)
               end
             rescue => e
@@ -43,6 +45,10 @@ def perform_in(interval, *args)
 
             private
 
+            def datadog_configuration
+              Datadog.configuration[:sucker_punch]
+            end
+
             def __with_instrumentation(name)
               pin = Datadog::Pin.get_from(::SuckerPunch)