move subscribers to their own files (635ff5c0) · Commits · GitLab.org / ruby / gems / labkit-ruby

lib/labkit/tracing/rails/action_view.rb

+1 −22

Original line number	Diff line number	Diff line
		@@ -7,30 +7,9 @@ module Labkit
		autoload :RenderCollectionInstrumenter, "labkit/tracing/rails/action_view/render_collection_instrumenter"
		autoload :RenderPartialInstrumenter, "labkit/tracing/rails/action_view/render_partial_instrumenter"
		autoload :RenderTemplateInstrumenter, "labkit/tracing/rails/action_view/render_template_instrumenter"
		autoload :Subscriber, "labkit/tracing/rails/action_view/subscriber"

		COMPONENT_TAG = "ActionView"

		# ActionView bridges action view notifications to
		# the distributed tracing subsystem
		class Subscriber
		include RailsCommon

		RENDER_TEMPLATE_NOTIFICATION_TOPIC = "render_template.action_view"
		RENDER_COLLECTION_NOTIFICATION_TOPIC = "render_collection.action_view"
		RENDER_PARTIAL_NOTIFICATION_TOPIC = "render_partial.action_view"

		# Instruments Rails ActionView events for opentracing.
		# Returns a lambda, which, when called will unsubscribe from the notifications
		def self.instrument
		subscriptions = [
		::ActiveSupport::Notifications.subscribe(RENDER_TEMPLATE_NOTIFICATION_TOPIC, RenderTemplateInstrumenter.new),
		::ActiveSupport::Notifications.subscribe(RENDER_COLLECTION_NOTIFICATION_TOPIC, RenderCollectionInstrumenter.new),
		::ActiveSupport::Notifications.subscribe(RENDER_PARTIAL_NOTIFICATION_TOPIC, RenderPartialInstrumenter.new),
		]

		create_unsubscriber subscriptions
		end
		end
		end
		end
		end

lib/labkit/tracing/rails/action_view/subscriber.rb

0 → 100644

+31 −0

Original line number	Diff line number	Diff line
		# frozen_string_literal: true

		module Labkit
		module Tracing
		module Rails
		module ActionView
		# ActionView bridges action view notifications to
		# the distributed tracing subsystem
		class Subscriber
		include RailsCommon

		RENDER_TEMPLATE_NOTIFICATION_TOPIC = "render_template.action_view"
		RENDER_COLLECTION_NOTIFICATION_TOPIC = "render_collection.action_view"
		RENDER_PARTIAL_NOTIFICATION_TOPIC = "render_partial.action_view"

		# Instruments Rails ActionView events for opentracing.
		# Returns a lambda, which, when called will unsubscribe from the notifications
		def self.instrument
		subscriptions = [
		::ActiveSupport::Notifications.subscribe(RENDER_TEMPLATE_NOTIFICATION_TOPIC, RenderTemplateInstrumenter.new),
		::ActiveSupport::Notifications.subscribe(RENDER_COLLECTION_NOTIFICATION_TOPIC, RenderCollectionInstrumenter.new),
		::ActiveSupport::Notifications.subscribe(RENDER_PARTIAL_NOTIFICATION_TOPIC, RenderPartialInstrumenter.new),
		]

		create_unsubscriber subscriptions
		end
		end
		end
		end
		end
		end

lib/labkit/tracing/rails/active_record.rb

+1 −16

Original line number	Diff line number	Diff line
		@@ -5,24 +5,9 @@ module Labkit
		module Rails
		module ActiveRecord
		autoload :SqlInstrumenter, "labkit/tracing/rails/active_record/sql_instrumenter"
		autoload :Subscriber, "labkit/tracing/rails/active_record/subscriber"

		COMPONENT_TAG = "ActiveRecord"

		# ActiveRecord bridges active record notifications to
		# the distributed tracing subsystem
		class Subscriber
		include RailsCommon

		ACTIVE_RECORD_NOTIFICATION_TOPIC = "sql.active_record"

		# Instruments Rails ActiveRecord events for opentracing.
		# Returns a lambda, which, when called will unsubscribe from the notifications
		def self.instrument
		subscription = ::ActiveSupport::Notifications.subscribe(ACTIVE_RECORD_NOTIFICATION_TOPIC, SqlInstrumenter.new)

		create_unsubscriber [subscription]
		end
		end
		end
		end
		end

lib/labkit/tracing/rails/active_record/subscriber.rb

0 → 100644

+25 −0

Original line number	Diff line number	Diff line
		# frozen_string_literal: true

		module Labkit
		module Tracing
		module Rails
		module ActiveRecord
		# ActiveRecord bridges active record notifications to
		# the distributed tracing subsystem
		class Subscriber
		include RailsCommon

		ACTIVE_RECORD_NOTIFICATION_TOPIC = "sql.active_record"

		# Instruments Rails ActiveRecord events for opentracing.
		# Returns a lambda, which, when called will unsubscribe from the notifications
		def self.instrument
		subscription = ::ActiveSupport::Notifications.subscribe(ACTIVE_RECORD_NOTIFICATION_TOPIC, SqlInstrumenter.new)

		create_unsubscriber [subscription]
		end
		end
		end
		end
		end
		end

lib/labkit/tracing/rails/active_support.rb

+1 −26

Original line number	Diff line number	Diff line
		@@ -9,34 +9,9 @@ module Labkit
		autoload :CacheGenerateInstrumenter, "labkit/tracing/rails/active_support/cache_generate_instrumenter"
		autoload :CacheReadInstrumenter, "labkit/tracing/rails/active_support/cache_read_instrumenter"
		autoload :CacheWriteInstrumenter, "labkit/tracing/rails/active_support/cache_write_instrumenter"
		autoload :Subscriber, "labkit/tracing/rails/active_support/subscriber"

		COMPONENT_TAG = "ActiveSupport"

		# ActiveSupport bridges action active support notifications to
		# the distributed tracing subsystem
		class Subscriber
		include RailsCommon

		CACHE_READ_TOPIC = "cache_read.active_support"
		CACHE_GENERATE_TOPIC = "cache_generate.active_support"
		CACHE_FETCH_HIT_TOPIC = "cache_fetch_hit.active_support"
		CACHE_WRITE_TOPIC = "cache_write.active_support"
		CACHE_DELETE_TOPIC = "cache_delete.active_support"

		# Instruments Rails ActiveSupport events for opentracing.
		# Returns a lambda, which, when called will unsubscribe from the notifications
		def self.instrument
		subscriptions = [
		::ActiveSupport::Notifications.subscribe(CACHE_READ_TOPIC, CacheReadInstrumenter.new),
		::ActiveSupport::Notifications.subscribe(CACHE_GENERATE_TOPIC, CacheGenerateInstrumenter.new),
		::ActiveSupport::Notifications.subscribe(CACHE_FETCH_HIT_TOPIC, CacheFetchHitInstrumenter.new),
		::ActiveSupport::Notifications.subscribe(CACHE_WRITE_TOPIC, CacheWriteInstrumenter.new),
		::ActiveSupport::Notifications.subscribe(CACHE_DELETE_TOPIC, CacheDeleteInstrumenter.new),
		]

		create_unsubscriber subscriptions
		end
		end
		end
		end
		end