Sha256: 0b70a67ce7c415d06ca77ef154f992efa293b20c4d995eda5af2a13ad6f5fafd
Contents?: true
Size: 1.82 KB
Versions: 27
Compression:
Stored size: 1.82 KB
Contents
require 'delayed/plugin'
require 'ddtrace/contrib/analytics'
require 'ddtrace/contrib/delayed_job/ext'
module Datadog
module Contrib
module DelayedJob
# DelayedJob plugin that instruments invoke_job hook
class Plugin < Delayed::Plugin
def self.instrument(job, &block)
return block.call(job) unless tracer && tracer.enabled
# When DelayedJob is used through ActiveJob, we need to parse the payload differentely
# to get the actual job name
job_name = if job.payload_object.respond_to?(:job_data)
job.payload_object.job_data['job_class']
else
job.name
end
tracer.trace(Ext::SPAN_JOB, service: configuration[:service_name], resource: job_name) do |span|
# Set analytics sample rate
if Contrib::Analytics.enabled?(configuration[:analytics_enabled])
Contrib::Analytics.set_sample_rate(span, configuration[:analytics_sample_rate])
end
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)
span.set_tag(Ext::TAG_ATTEMPTS, job.attempts)
span.span_type = Datadog::Ext::AppTypes::WORKER
yield job
end
end
def self.flush(worker, &block)
yield worker
tracer.shutdown! if tracer && tracer.enabled
end
def self.configuration
Datadog.configuration[:delayed_job]
end
def self.tracer
configuration[:tracer]
end
callbacks do |lifecycle|
lifecycle.around(:invoke_job, &method(:instrument))
lifecycle.around(:execute, &method(:flush))
end
end
end
end
end
Version data entries
27 entries across 27 versions & 2 rubygems