active_record_query_counter 2.1.0 → 2.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9d653eb9665511008bc78023c7e39ae8eb189e051847d83e16244424b5cbe595
-  data.tar.gz: 6e9643c840a635228ea0a04cb02277b91c34b1ea5e69bd5b93753066be8ab4fb
+  metadata.gz: 00cf8b2bba420e0b37944a53f4be128aeac846d6e387b15b03c3393f3a0030c2
+  data.tar.gz: 10c05613e25513f1afab756f9cdd1133a19f7b54e34ca811b7f3ba5ad982ff0e
 SHA512:
-  metadata.gz: df192239b02e3ed039eac91a64393d5332cad36dce242e9e02948b3ec0cbe23e7ae5d4b23dc58ebf01158fa885680135e6719f444a871abd37779e85a9f85d2d
-  data.tar.gz: '069e6959ba3f7ead49824ae91d35774b3b336afe18248c84800eaca65115daa040bdf5871d0c3e591b648cce2b0d5414078fe51ef39bef8d9f0e777be8e7330d'
+  metadata.gz: 4294aac8c3b0b55197ba1d069cc0a57f269ad772c23480b928fcf3f0a01b7c4f806f4969bf32f2d229b8dea65bc976d18cc864f923ec468dbbb43fc2fbe20bdd
+  data.tar.gz: ffddf3b2d87b74a2ad184d0ec55c4b371cda418c262aa2d0889cf40916579f5bff575ef0e631a39764edd58f7d321b71fa1b993dda6c3ff0c9f1ef060ca02769

data/CHANGELOG.md

@@ -4,6 +4,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## 2.2.0
+
+### Added
+
+- Added `ActiveRecordQueryCounter.disable` method to allow disabling query counting behavior within a block.
+- Rails 7.1 compatibility.
+
 ## 2.1.0
 
 ### Added

data/README.md

@@ -60,6 +60,18 @@ Sidekiq.configure_server do |config|
 end
 ```
 
+If you want to disable query counting within a block of code, you can use the `disable` method.
+
+```ruby
+ActiveRecordQueryCounter.count_queries do
+  do_something
+  ActiveRecordQueryCounter.disable do
+    # Queries will not be counted in this block.
+    do_something_else
+  end
+end
+```
+
 ### Notifications
 
 You can also subscribe to ActiveSupport notifications to get notified when query thresholds are exceeded.

data/VERSION

@@ -1 +1 @@
-2.1.0
+2.2.0

data/lib/active_record_query_counter/connection_adapter_extension.rb

@@ -3,14 +3,38 @@
 module ActiveRecordQueryCounter
   # Module to prepend to the connection adapter to inject the counting behavior.
   module ConnectionAdapterExtension
-    def exec_query(sql, name = nil, binds = [], *args, **kwargs)
-      start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
-      result = super
-      if result.is_a?(ActiveRecord::Result)
-        end_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
-        ActiveRecordQueryCounter.add_query(sql, name, binds, result.length, start_time, end_time)
+    class << self
+      def inject(connection_class)
+        # Rails 7.1+ uses internal_exec_query instead of exec_query.
+        mod = (connection_class.instance_methods.include?(:internal_exec_query) ? InternalExecQuery : ExecQuery)
+        unless connection_class.include?(mod)
+          connection_class.prepend(mod)
+        end
+      end
+    end
+
+    module ExecQuery
+      def exec_query(sql, name = nil, binds = [], *args, **kwargs)
+        start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+        result = super
+        if result.is_a?(ActiveRecord::Result)
+          end_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+          ActiveRecordQueryCounter.add_query(sql, name, binds, result.length, start_time, end_time)
+        end
+        result
+      end
+    end
+
+    module InternalExecQuery
+      def internal_exec_query(sql, name = nil, binds = [], *args, **kwargs)
+        start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+        result = super
+        if result.is_a?(ActiveRecord::Result)
+          end_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+          ActiveRecordQueryCounter.add_query(sql, name, binds, result.length, start_time, end_time)
+        end
+        result
       end
-      result
     end
   end
 end

data/lib/active_record_query_counter/thresholds.rb

@@ -6,6 +6,10 @@ module ActiveRecordQueryCounter
   class Thresholds
     attr_reader :query_time, :row_count, :transaction_time, :transaction_count
 
+    def initialize
+      clear
+    end
+
     def query_time=(value)
       @query_time = value&.to_f
     end

data/lib/active_record_query_counter/transaction_manager_extension.rb

@@ -3,6 +3,14 @@
 module ActiveRecordQueryCounter
   # Extension to ActiveRecord::ConnectionAdapters::TransactionManager to count transactions.
   module TransactionManagerExtension
+    class << self
+      def inject(transaction_manager_class)
+        unless transaction_manager_class.include?(self)
+          transaction_manager_class.prepend(self)
+        end
+      end
+    end
+
     def begin_transaction(*args, **kwargs)
       if open_transactions == 0
         @active_record_query_counter_transaction_start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)

data/lib/active_record_query_counter.rb

@@ -48,6 +48,20 @@ module ActiveRecordQueryCounter
       end
     end
 
+    # Disable query counting in a block. Any queries or transactions inside the block will not
+    # be counted.
+    #
+    # @return [Object] the return value of the block
+    def disable(&block)
+      counter = current_counter
+      begin
+        self.current_counter = nil
+        yield
+      ensure
+        self.current_counter = counter
+      end
+    end
+
     # Increment the query counters.
     #
     # @param row_count [Integer] the number of rows returned by the query
@@ -219,12 +233,8 @@ module ActiveRecordQueryCounter
     # @param connection_class [Class] the connection adapter class to extend
     # @return [void]
     def enable!(connection_class)
-      unless connection_class.include?(ConnectionAdapterExtension)
-        connection_class.prepend(ConnectionAdapterExtension)
-      end
-      unless ActiveRecord::ConnectionAdapters::TransactionManager.include?(TransactionManagerExtension)
-        ActiveRecord::ConnectionAdapters::TransactionManager.prepend(TransactionManagerExtension)
-      end
+      ConnectionAdapterExtension.inject(connection_class)
+      TransactionManagerExtension.inject(ActiveRecord::ConnectionAdapters::TransactionManager)
 
       @cache_subscription ||= ActiveSupport::Notifications.subscribe("sql.active_record") do |_name, _start_time, _end_time, _id, payload|
         if payload[:cached]

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_record_query_counter
 version: !ruby/object:Gem::Version
-  version: 2.1.0
+  version: 2.2.0
 platform: ruby
 authors:
 - Brian Durand
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-09-13 00:00:00.000000000 Z
+date: 2023-11-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -78,7 +78,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.10
+rubygems_version: 3.4.20
 signing_key:
 specification_version: 4
 summary: Count total number of ActiveRecord queries and row counts inside a block