lib/lograge/sql.rb in lograge-sql-2.0.0 vs lib/lograge/sql.rb in lograge-sql-2.1.0

@@ -9,22 +9,21 @@
     class << self
       # Format SQL log
       attr_accessor :formatter
       # Extract information from SQL event
       attr_accessor :extract_event
+      # Filter SQL events by duration
+      attr_accessor :min_duration_ms
 
       # Initialise configuration with fallback to default values
       def setup(config)
-        Lograge::Sql.formatter     = config.formatter     || default_formatter
-        Lograge::Sql.extract_event = config.extract_event || default_extract_event
+        Lograge::Sql.formatter       = config.formatter       || default_formatter
+        Lograge::Sql.extract_event   = config.extract_event   || default_extract_event
+        Lograge::Sql.min_duration_ms = config.min_duration_ms || 0
 
         # Disable existing ActiveRecord logging
-        unless config.keep_default_active_record_log
-          ActiveSupport::LogSubscriber.log_subscribers.each do |subscriber|
-            Lograge.unsubscribe(:active_record, subscriber) if subscriber.is_a?(ActiveRecord::LogSubscriber)
-          end
-        end
+        unsubscribe_log_subscribers unless config.keep_default_active_record_log
 
         return unless defined?(Lograge::ActiveRecordLogSubscriber)
 
         Lograge::ActiveRecordLogSubscriber.attach_to(:active_record)
       end
@@ -47,9 +46,15 @@
       # By default, only extract values required for the default_formatter and
       # already convert to a string
       def default_extract_event
         proc do |event|
           "#{event.payload[:name]} (#{event.duration.to_f.round(2)}) #{event.payload[:sql]}"
+        end
+      end
+
+      def unsubscribe_log_subscribers
+        ActiveSupport::LogSubscriber.log_subscribers.each do |subscriber|
+          Lograge.unsubscribe(:active_record, subscriber) if subscriber.is_a?(ActiveRecord::LogSubscriber)
         end
       end
     end
   end
 end