Add connection logging in non-Rails enviroments

instacart · Nov 20, 2018 · 01195ef · 01195ef
1 parent a506055
commit 01195ef
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/lib/makara.rb b/lib/makara.rb
@@ -1,3 +1,4 @@
+require 'active_support'
 require 'makara/version'
 require 'makara/railtie' if defined?(Rails)
 module Makara
@@ -31,3 +32,9 @@ module Strategies
   end
 
 end
+
+ActiveSupport.on_load(:active_record) do
+  ActiveRecord::LogSubscriber.log_subscribers.each do |subscriber|
+    subscriber.extend ::Makara::Logging::Subscriber
+  end
+end
diff --git a/lib/makara/railtie.rb b/lib/makara/railtie.rb
@@ -5,11 +5,5 @@ class Railtie < ::Rails::Railtie
       app.middleware.use Makara::Middleware
     end
 
-    initializer "makara.initialize_logger" do |app|
-      ActiveRecord::LogSubscriber.log_subscribers.each do |subscriber|
-        subscriber.extend ::Makara::Logging::Subscriber
-      end
-    end
-
   end
 end