Sha256: 47622ef665acb10c4da6acc0d58f6cd8aa01d0aebcb0d0f5b9049166f4349b3c
Contents?: true
Size: 1.91 KB
Versions: 33
Compression:
Stored size: 1.91 KB
Contents
require 'sucker_punch'
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('sucker_punch.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('sucker_punch.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('sucker_punch.perform_in') do |span|
span.resource = "ENQUEUE #{self}"
span.set_tag('sucker_punch.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('sucker_punch.queue', to_s)
yield span
end
end
end
end
end
end
end
end
Version data entries
33 entries across 33 versions & 1 rubygems