Module: ActiveRecordQueryCounter::TransactionManagerExtension

Defined in:
lib/active_record_query_counter/transaction_manager_extension.rb

Overview

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

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.inject(transaction_manager_class) ⇒ Object 



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

def inject(transaction_manager_class)
  unless transaction_manager_class.include?(self)
    transaction_manager_class.prepend(self)
  end
end

Instance Method Details

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



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

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 



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

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 



30
31
32
33
34
35
36
37
38
 
# File 'lib/active_record_query_counter/transaction_manager_extension.rb', line 30

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)
    ActiveRecordQueryCounter.increment_rollbacks
    @active_record_query_counter_transaction_start_time = nil
  end
  super
end