Sha256: 77e2a6effa8d41830167933f0a706eb9458fb2f27215912a638a47d9487b6c5e
Contents?: true
Size: 1.79 KB
Versions: 18
Compression:
Stored size: 1.79 KB
Contents
# Copyright (c) 2015 AppNeta, Inc.
# All rights reserved.
module TraceView
class SidekiqClient
def collect_kvs(args)
begin
# Attempt to collect up pertinent info. If we hit something unexpected,
# keep calm and instrument on.
report_kvs = {}
worker_class, msg, queue, _ = args
report_kvs[:Spec] = :msgclient
report_kvs[:Flavor] = :sidekiq
report_kvs[:Queue] = queue
report_kvs[:Retry] = msg['retry']
report_kvs[:JobName] = worker_class
report_kvs[:MsgID] = msg['jid']
report_kvs[:Args] = msg['args'].to_s[0..1024] if TV::Config[:sidekiqclient][:log_args]
report_kvs[:Backtrace] = TV::API.backtrace if TV::Config[:sidekiqclient][:collect_backtraces]
rescue => e
TraceView.logger.warn "[traceview/sidekiq] Non-fatal error capturing KVs: #{e.message}"
end
report_kvs
end
def call(*args)
# args: 0: worker_class, 1: msg, 2: queue, 3: redis_pool
report_kvs = collect_kvs(args)
TraceView::API.log_entry('sidekiq-client', report_kvs)
args[1]['SourceTrace'] = TraceView::Context.toString if TraceView.tracing?
result = yield
report_kvs = { :JobID => result['jid'] }
result
rescue => e
TraceView::API.log_exception('sidekiq-client', e, report_kvs)
raise
ensure
TraceView::API.log_exit('sidekiq-client', report_kvs)
end
end
end
if defined?(::Sidekiq) && RUBY_VERSION >= '2.0' && TraceView::Config[:sidekiqclient][:enabled]
::Sidekiq.configure_client do |config|
config.client_middleware do |chain|
::TraceView.logger.info '[traceview/loading] Adding Sidekiq client middleware' if TraceView::Config[:verbose]
chain.add ::TraceView::SidekiqClient
end
end
end
Version data entries
18 entries across 18 versions & 1 rubygems