Added: event_sample_rate tag to DelayedJob integration.

lib/ddtrace/contrib/delayed_job/plugin.rb

@@ -1,4 +1,5 @@
 require 'delayed/plugin'
+require 'ddtrace/contrib/sampling'
 require 'ddtrace/contrib/delayed_job/ext'
 
 module Datadog
@@ -18,6 +19,7 @@ def self.instrument(job, &block)
                      end
 
           tracer.trace(Ext::SPAN_JOB, service: configuration[:service_name], resource: job_name) do |span|
+            Contrib::Sampling.set_event_sample_rate(span, configuration[:event_sample_rate])
             span.set_tag(Ext::TAG_ID, job.id)
             span.set_tag(Ext::TAG_QUEUE, job.queue) if job.queue
             span.set_tag(Ext::TAG_PRIORITY, job.priority)

spec/ddtrace/contrib/delayed_job/plugin_spec.rb

@@ -25,13 +25,15 @@ def job_data
   end
 
   let(:tracer) { get_test_tracer }
+  let(:options) { { tracer: tracer } }
 
   before do
-    Datadog.configure { |c| c.use :delayed_job, tracer: tracer }
-
+    Datadog.configure { |c| c.use :delayed_job, options }
     Delayed::Worker.delay_jobs = false
   end
 
+  after(:each) { Datadog.registry[:delayed_job].reset_configuration! }
+
   describe 'instrumenting worker execution' do
     let(:worker) { double(:worker, name: 'worker') }
     before do
@@ -100,6 +102,23 @@ def job_data
         expect(span.get_tag('delayed_job.attempts')).to eq('0')
       end
 
+      describe 'event sample rate' do
+        context 'when not configured' do
+          it 'is not included in the tags' do
+            expect(span.get_tag(Datadog::Ext::Priority::TAG_EVENT_SAMPLE_RATE)).to be nil
+          end
+        end
+
+        context 'when set' do
+          let(:options) { super().merge(event_sample_rate: event_sample_rate) }
+          let(:event_sample_rate) { 0.5 }
+
+          it 'is included in the tags' do
+            expect(span.get_tag(Datadog::Ext::Priority::TAG_EVENT_SAMPLE_RATE)).to eq(event_sample_rate.to_s)
+          end
+        end
+      end
+
       context 'when queue name is set' do
         let(:queue_name) { 'queue_name' }
         let(:job_params) { { queue: queue_name } }