Contents

# frozen_string_literal: true

module ActiveRecordQueryCounter
  # Extension to ActiveRecord::ConnectionAdapters::TransactionManager to count transactions.
  module TransactionManagerExtension
    def begin_transaction(*args, **kwargs)
      if open_transactions == 0
        @active_record_query_counter_transaction_start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
      end
      super
    end

    def commit_transaction(*args)
      if @active_record_query_counter_transaction_start_time && open_transactions == 1
        end_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
        ActiveRecordQueryCounter.add_transaction(@active_record_query_counter_transaction_start_time, end_time)
        @active_record_query_counter_transaction_start_time = nil
      end
      super
    end

    def rollback_transaction(*args)
      if @active_record_query_counter_transaction_start_time && open_transactions == 1
        end_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
        ActiveRecordQueryCounter.add_transaction(@active_record_query_counter_transaction_start_time, end_time)
        @active_record_query_counter_transaction_start_time = nil
      end
      super
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
active_record_query_counter-2.1.0	lib/active_record_query_counter/transaction_manager_extension.rb
active_record_query_counter-2.0.0	lib/active_record_query_counter/transaction_manager_extension.rb