diff options
-rw-r--r-- | activesupport/lib/active_support/notifications.rb | 23 |
1 files changed, 20 insertions, 3 deletions
diff --git a/activesupport/lib/active_support/notifications.rb b/activesupport/lib/active_support/notifications.rb index 2e2314cad5..12bcd6d308 100644 --- a/activesupport/lib/active_support/notifications.rb +++ b/activesupport/lib/active_support/notifications.rb @@ -137,6 +137,23 @@ module ActiveSupport module Notifications @instrumenters = Hash.new { |h,k| h[k] = notifier.listening?(k) } + class Registry # :nodoc: + def self.instance + Thread.current[name] ||= new + end + + attr_reader :notifier, :instrumenter + + def initialize + self.notifier = Fanout.new + end + + def notifier=(notifier) + @notifier = notifier + @instrumenter = Instrumenter.new(notifier) + end + end + class << self def publish(name, *args) notifier.publish(name, *args) @@ -169,15 +186,15 @@ module ActiveSupport end def instrumenter - Thread.current[:"instrumentation_#{notifier.object_id}"] ||= Instrumenter.new(notifier) + Registry.instance.instrumenter end def notifier - Thread.current[:notifier] ||= Fanout.new + Registry.instance.notifier end def notifier=(notifier) - Thread.current[:notifier] = notifier + Registry.instance.notifier = notifier end end end |