Module: ActiveRecordQueryCounter::TransactionManagerExtension

Defined in:
lib/active_record_query_counter/transaction_manager_extension.rb

Overview

Extension to ActiveRecord::ConnectionAdapters::TransactionManager to count transactions.

Instance Method Summary collapse

Instance Method Details

#begin_transaction(*args, **kwargs) ⇒ Object



6
7
8
9
10
11
# File 'lib/active_record_query_counter/transaction_manager_extension.rb', line 6

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

#commit_transaction(*args) ⇒ Object



13
14
15
16
17
18
19
20
# File 'lib/active_record_query_counter/transaction_manager_extension.rb', line 13

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

#rollback_transaction(*args) ⇒ Object



22
23
24
25
26
27
28
29
# File 'lib/active_record_query_counter/transaction_manager_extension.rb', line 22

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