Sha256: 93d1b6321294a42867a51112ddb9d4145568017958e141cd15ff0acd6038df20
Contents?: true
Size: 1.17 KB
Versions: 8
Compression:
Stored size: 1.17 KB
Contents
require 'honeybadger/plugin'
require 'honeybadger/ruby'
module Honeybadger
module Plugins
module Shoryuken
class Middleware
def call(_worker, _queue, sqs_msg, body)
begin
yield
rescue => e
if attempt_threshold <= receive_count(sqs_msg)
Honeybadger.notify(e, parameters: notification_params(body))
end
raise e
end
ensure
Honeybadger.context.clear!
end
private
def attempt_threshold
::Honeybadger.config[:'shoryuken.attempt_threshold'].to_i
end
def receive_count(sqs_msg)
return 0 if sqs_msg.is_a?(Array)
sqs_msg.attributes['ApproximateReceiveCount'.freeze].to_i
end
def notification_params(body)
body.is_a?(Array) ? { batch: body } : { body: body }
end
end
Plugin.register do
requirement { defined?(::Shoryuken) }
execution do
::Shoryuken.configure_server do |config|
config.server_middleware do |chain|
chain.add Middleware
end
end
end
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems