def self.included(base)
base.class_eval do
def after_callback_transaction_committed?
committed = (Thread.current[:after_callback_committed] || {})[unique_transaction_key]
return !committed.nil? && committed
end
def after_callback_mark_committed committed
(Thread.current[:after_callback_committed] ||= {})[unique_transaction_key] = committed
end
def after_callback_cleanup_committed
(Thread.current[:after_callback_committed] ||= {})[unique_transaction_key] = nil
end
def release_savepoint_with_callback
increment_transaction_pointer
after_callback_mark_committed false
begin
trigger_before_commit_callbacks
trigger_before_commit_on_create_callbacks
trigger_before_commit_on_save_callbacks
trigger_before_commit_on_update_callbacks
trigger_before_commit_on_destroy_callbacks
release_savepoint_without_callback
after_callback_mark_committed true
trigger_after_commit_callbacks
trigger_after_commit_on_create_callbacks
trigger_after_commit_on_save_callbacks
trigger_after_commit_on_update_callbacks
trigger_after_commit_on_destroy_callbacks
AfterCommit.cleanup(self)
after_callback_cleanup_committed
ensure
decrement_transaction_pointer
end
end
alias_method_chain :release_savepoint, :callback
def rollback_to_savepoint_with_callback
increment_transaction_pointer
begin
unless after_callback_transaction_committed?
trigger_before_rollback_callbacks
rollback_to_savepoint_without_callback
trigger_after_rollback_callbacks
end
ensure
AfterCommit.cleanup(self)
after_callback_cleanup_committed
decrement_transaction_pointer
end
end
alias_method_chain :rollback_to_savepoint, :callback
end
end