Module: AfterCommit::ConnectionAdapters

Defined in:
lib/after_commit/connection_adapters.rb

Class Method Summary collapse

Class Method Details

.included(base) ⇒ Object



3
4
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
# File 'lib/after_commit/connection_adapters.rb', line 3

def self.included(base)
  base.class_eval do

    if method_defined?(:transaction)
      def transaction_with_callback(*args, &block)
        # @disable_rollback is set to false at the start of the
        # outermost call to #transaction.  After committing, it is
        # set to true to prevent exceptions causing a spurious
        # rollback.
        outermost_call = @disable_rollback.nil?
        @disable_rollback = false if outermost_call
        transaction_without_callback(*args, &block)
      ensure
        @disable_rollback = nil if outermost_call
      end
      alias_method_chain :transaction, :callback

    elsif method_defined?(:begin_db_transaction)
      def begin_db_transaction_with_callback(*args, &block)
        # @disable_rollback is set to false at the start of the
        # outermost call to #transaction.  After committing, it is
        # set to true to prevent exceptions causing a spurious
        # rollback.
        outermost_call = @disable_rollback.nil?
        @disable_rollback = false if outermost_call
        begin_db_transaction_without_callback(*args, &block)
      ensure
        @disable_rollback = nil if outermost_call
      end
      alias_method_chain :begin_db_transaction, :callback
    end

    # The commit_db_transaction method gets called when the outermost
    # transaction finishes and everything inside commits. We want to
    # override it so that after this happens, any records that were saved
    # or destroyed within this transaction now get their after_commit
    # callback fired.
    def commit_db_transaction_with_callback
      increment_transaction_pointer
      result    = nil
      begin
        trigger_before_commit_callbacks
        trigger_before_commit_on_create_callbacks
        trigger_before_commit_on_update_callbacks
        trigger_before_commit_on_save_callbacks
        trigger_before_commit_on_destroy_callbacks

        result = commit_db_transaction_without_callback
        @disable_rollback = true

        trigger_after_commit_callbacks
        trigger_after_commit_on_create_callbacks
        trigger_after_commit_on_update_callbacks
        trigger_after_commit_on_save_callbacks
        trigger_after_commit_on_destroy_callbacks
        result
      rescue
        # Need to decrement the transaction pointer before calling
        # rollback... to ensure it is not incremented twice
        unless @disable_rollback
          decrement_transaction_pointer
          @already_decremented = true
        end
        
        # We still want to raise the exception.
        raise
      ensure
        AfterCommit.cleanup(self)
        decrement_transaction_pointer unless @already_decremented
      end
    end
    alias_method_chain :commit_db_transaction, :callback

    # In the event the transaction fails and rolls back, nothing inside
    # should recieve the after_commit callback, but do fire the after_rollback
    # callback for each record that failed to be committed.
    def rollback_db_transaction_with_callback
      return if @disable_rollback
      increment_transaction_pointer
      begin
        result = nil
        trigger_before_rollback_callbacks
        result = rollback_db_transaction_without_callback
        trigger_after_rollback_callbacks
        result
      ensure
        AfterCommit.cleanup(self)
        decrement_transaction_pointer
      end
      decrement_transaction_pointer
    end
    alias_method_chain :rollback_db_transaction, :callback
    
    def unique_transaction_key
      [object_id, transaction_pointer]
    end
    
    def old_transaction_key
      [object_id, transaction_pointer - 1]
    end
    
    protected
    
    def trigger_before_commit_callbacks
      AfterCommit.records(self).each do |record|
        record.send :callback, :before_commit
      end 
    end

    def trigger_before_commit_on_create_callbacks
      AfterCommit.created_records(self).each do |record|
        record.send :callback, :before_commit_on_create
      end 
    end
  
    def trigger_before_commit_on_update_callbacks
      AfterCommit.updated_records(self).each do |record|
        record.send :callback, :before_commit_on_update
      end 
    end
  
    def trigger_before_commit_on_save_callbacks
      AfterCommit.saved_records(self).each do |record|
        record.send :callback, :before_commit_on_save
      end
    end
    
    def trigger_before_commit_on_destroy_callbacks
      AfterCommit.destroyed_records(self).each do |record|
        record.send :callback, :before_commit_on_destroy
      end 
    end

    def trigger_before_rollback_callbacks
      AfterCommit.records(self).each do |record|
        record.send :callback, :before_rollback
      end 
    end

    def trigger_after_commit_callbacks
      # Trigger the after_commit callback for each of the committed
      # records.
      AfterCommit.records(self).each do |record|
        record.send :callback, :after_commit
      end
    end
        
    def trigger_after_commit_on_create_callbacks
      # Trigger the after_commit_on_create callback for each of the committed
      # records.
      AfterCommit.created_records(self).each do |record|
        record.send :callback, :after_commit_on_create
      end
    end
  
    def trigger_after_commit_on_update_callbacks
      # Trigger the after_commit_on_update callback for each of the committed
      # records.
      AfterCommit.updated_records(self).each do |record|
        record.send :callback, :after_commit_on_update
      end
    end
  
    def trigger_after_commit_on_save_callbacks
      # Trigger the after_commit_on_save callback for each of the committed
      # records.
      AfterCommit.saved_records(self).each do |record|
        record.send :callback, :after_commit_on_save
      end
    end
    
    def trigger_after_commit_on_destroy_callbacks
      # Trigger the after_commit_on_destroy callback for each of the committed
      # records.
      AfterCommit.destroyed_records(self).each do |record|
        record.send :callback, :after_commit_on_destroy
      end
    end

    def trigger_after_rollback_callbacks
      # Trigger the after_rollback callback for each of the committed
      # records.
      AfterCommit.records(self).each do |record|
        record.send :callback, :after_rollback
      end 
    end
    
    def transaction_pointer
      Thread.current[:after_commit_pointer] ||= 0
    end
    
    def increment_transaction_pointer
      Thread.current[:after_commit_pointer] ||= 0
      Thread.current[:after_commit_pointer] += 1
    end
    
    def decrement_transaction_pointer
      Thread.current[:after_commit_pointer] ||= 0
      Thread.current[:after_commit_pointer] -= 1
    end

  end 
end