Sha256: 317fa3f64e6a70471e8c1abc5c7494e79a9702381e9910c4597e4f43cfe805e1
Contents?: true
Size: 1.99 KB
Versions: 8
Compression:
Stored size: 1.99 KB
Contents
# frozen_string_literal: true
require "active_support/notifications"
module Honeycomb
module ActiveSupport
##
# Included in the configuration object to specify events that should be
# subscribed to
module Configuration
attr_accessor :notification_events
def after_initialize(client)
super(client) if defined?(super)
events = notification_events || []
ActiveSupport::Subscriber.new(client: client).tap do |sub|
events.each do |event|
sub.subscribe(event, &method(:handle_notification_event))
end
end
end
def on_notification_event(&hook)
if block_given?
@on_notification_event = hook
else
@on_notification_event
end
end
def handle_notification_event(name, span, payload)
if on_notification_event
on_notification_event.call(name, span, payload)
else
payload.each do |key, value|
span.add_field("#{name}.#{key}", value.to_s)
end
end
end
end
# Handles ActiveSupport::Notification subscriptions, relaying them to a
# Honeycomb client
class Subscriber
def initialize(client:)
@client = client
@handlers = {}
@key = ["honeycomb", self.class.name, object_id].join("-")
end
def subscribe(event, &block)
return unless block_given?
handlers[event] = block
::ActiveSupport::Notifications.subscribe(event, self)
end
def start(name, id, _payload)
spans[id] << client.start_span(name: name)
end
def finish(name, id, payload)
return unless (span = spans[id].pop)
handlers[name].call(name, span, payload)
span.send
end
private
attr_reader :key, :client, :handlers
def spans
Thread.current[key] ||= Hash.new { |h, id| h[id] = [] }
end
end
end
end
Honeycomb::Configuration.include Honeycomb::ActiveSupport::Configuration
Version data entries
8 entries across 8 versions & 1 rubygems