Keep Logging in the ActiveModelSerializers namespace

lib/active_model/serializer/logging.rb

@@ -1,26 +0,0 @@
-module ActiveModel
-  class Serializer
-    module Logging
-      extend ActiveSupport::Concern
-
-      class LogSubscriber < ActiveSupport::LogSubscriber
-        def render(event)
-          logger.tagged('AMS') do
-            info do
-              serializer = event.payload[:serializer]
-              adapter = event.payload[:adapter]
-              duration = event.duration.round(2)
-              "Rendered #{serializer.name} with #{adapter.class} (#{duration}ms)"
-            end
-          end
-        end
-
-        def logger
-          ActiveModelSerializers.logger
-        end
-      end
-    end
-  end
-end
-
-ActiveModel::Serializer::Logging::LogSubscriber.attach_to :active_model_serializers

lib/active_model/serializer/railtie.rb

@@ -1,9 +1,11 @@
+require 'active_model_serializers'
 require 'rails/railtie'
+
 module ActiveModel
   class Railtie < Rails::Railtie
     initializer 'active_model_serializers.logger' do
-      ActiveSupport.on_load(:action_controller) do
-        ActiveModelSerializers.logger = ActionController::Base.logger
+      ActiveSupport.on_load(:active_model_serializers) do
+        self.logger = ActionController::Base.logger
       end
     end