Sha256: bdbeb897d7d55ba833ae0ff49f389c6ec2b9f1210920ea21e895ce037f073871
Contents?: true
Size: 1.92 KB
Versions: 8
Compression:
Stored size: 1.92 KB
Contents
require 'sucker_punch'
require 'ddtrace/contrib/sucker_punch/ext'
module Datadog
module Contrib
module SuckerPunch
# Defines instrumentation patches for the `sucker_punch` gem
module Instrumentation
module_function
# rubocop:disable Metrics/MethodLength
def patch!
::SuckerPunch::Job::ClassMethods.class_eval do
alias_method :__run_perform_without_datadog, :__run_perform
def __run_perform(*args)
pin = Datadog::Pin.get_from(::SuckerPunch)
pin.tracer.provider.context = Datadog::Context.new
__with_instrumentation(Ext::SPAN_PERFORM) do |span|
span.resource = "PROCESS #{self}"
__run_perform_without_datadog(*args)
end
rescue => e
::SuckerPunch.__exception_handler.call(e, self, args)
end
alias_method :__perform_async, :perform_async
def perform_async(*args)
__with_instrumentation(Ext::SPAN_PERFORM_ASYNC) do |span|
span.resource = "ENQUEUE #{self}"
__perform_async(*args)
end
end
alias_method :__perform_in, :perform_in
def perform_in(interval, *args)
__with_instrumentation(Ext::SPAN_PERFORM_IN) do |span|
span.resource = "ENQUEUE #{self}"
span.set_tag(Ext::TAG_PERFORM_IN, interval)
__perform_in(interval, *args)
end
end
private
def __with_instrumentation(name)
pin = Datadog::Pin.get_from(::SuckerPunch)
pin.tracer.trace(name) do |span|
span.service = pin.service
span.span_type = pin.app_type
span.set_tag(Ext::TAG_QUEUE, to_s)
yield span
end
end
end
end
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems