Module: ActiveRecord::Turntable::ActiveRecordExt::Transactions

Defined in:
lib/active_record/turntable/active_record_ext/transactions.rb

Instance Method Summary collapse

Instance Method Details

#add_to_transaction ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/active_record/turntable/active_record_ext/transactions.rb', line 32

def add_to_transaction
  return super unless self.class.turntable_enabled?

  if has_transactional_callbacks?
    self.turntable_shard.connection.add_transaction_record(self)
  else
    sync_with_transaction_state
    set_transaction_state(self.turntable_shard.connection.transaction_state)
  end
  remember_transaction_record_state
end

#with_transaction_returning_status ⇒ Object

Note:

Override to start transaction on current shard



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/active_record/turntable/active_record_ext/transactions.rb', line 5

def with_transaction_returning_status
  klass = self.class
  return super unless klass.turntable_enabled?

  status = nil
  if self.new_record? && self.turntable_shard_key.to_s == klass.primary_key &&
      self.id.nil? && klass.prefetch_primary_key?
    self.id = klass.next_sequence_value
  end
  self.class.connection.shards_transaction([self.turntable_shard]) do
    add_to_transaction
    begin
      status = yield
    rescue ActiveRecord::Rollback
      clear_transaction_record_state
      status = nil
    end

    raise ActiveRecord::Rollback unless status
  end
  status
ensure
  if @transaction_state && @transaction_state.committed?
    clear_transaction_record_state
  end
end