Use ActiveSupport.on_load to defer autoloaded Action Cable constants

Co-authored-by: Rob Brackett <[email protected]>

Author: bensheldon

lib/lograge/rails_ext/action_cable/channel/base.rb

@@ -22,4 +22,7 @@ def notification_payload(method_name)
   end
 end
 
-ActionCable::Channel::Base.prepend(Lograge::ActionCable::ChannelInstrumentation)
+
+ActiveSupport.on_load(:action_cable_channel) do
+  ActionCable::Channel::Base.prepend(Lograge::ActionCable::ChannelInstrumentation)
+end

lib/lograge/rails_ext/action_cable/connection/base.rb

@@ -18,4 +18,6 @@ def notification_payload(method_name)
   end
 end
 
-ActionCable::Connection::Base.prepend(Lograge::ActionCable::ConnectionInstrumentation)
+ActiveSupport.on_load(:action_cable_connection) do
+  ActionCable::Connection::Base.prepend(Lograge::ActionCable::ConnectionInstrumentation)
+end

lib/lograge/rails_ext/action_cable/server/base.rb

@@ -1,10 +1,12 @@
 # frozen_string_literal: true
 
-module ActionCable
-  module Server
-    class Base
-      mattr_accessor :logger
-      self.logger = Lograge::SilentLogger.new(config.logger)
+ActiveSupport.on_load(:action_cable) do
+  module ActionCable
+    module Server
+      class Base
+        mattr_accessor :logger
+        self.logger = Lograge::SilentLogger.new(config.logger)
+      end
     end
   end
 end