Merge pull request #79 from theablefew/esmarkowski/Add-instrumentation

Add instrumentation
theablefew · Mar 15, 2024 · d64449f · d64449f
2 parents 8b26a96 + a8c12a5
commit d64449f
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/bin/console b/bin/console
@@ -14,6 +14,9 @@ print "\t    " * 7
 print Rainbow(" v#{Stretchy::VERSION}").white.faint
 print "\n\n"
 
+
+Stretchy.instrument!
+
 ActiveSupport::Notifications.subscribe 'search.stretchy' do |name, start, finish, id, payload|
    message = [
       Rainbow("  #{payload[:klass]}").bright, 

diff --git a/lib/rails/instrumentation/railtie.rb b/lib/rails/instrumentation/railtie.rb
@@ -4,6 +4,8 @@ module Instrumentation
     class Railtie < ::Rails::Railtie
 
       require 'rails/instrumentation/publishers'
+      Stretchy.instrument!
+
       ActiveSupport::Notifications.subscribe 'search.stretchy' do |name, start, finish, id, payload|
         message = [
            Rainbow("  #{payload[:klass]}").bright, 

diff --git a/lib/stretchy.rb b/lib/stretchy.rb
@@ -30,6 +30,10 @@ def self.boot!
     Stretchy::Attributes.register!
   end
 
+  def self.instrument!
+    Stretchy::Delegation::GatewayDelegation.send(:include, Rails::Instrumentation::Publishers::Record)
+  end
+
   module Errors
     class QueryOptionMissing < StandardError; end
   end

diff --git a/lib/stretchy/delegation/gateway_delegation.rb b/lib/stretchy/delegation/gateway_delegation.rb
@@ -18,8 +18,6 @@ module GatewayDelegation
                 :count,
         to: :gateway
 
-      include Rails::Instrumentation::Publishers::Record
-
       def index_name(name=nil, &block)
           if name || block_given?
             return (@index_name = name || block)