class ActiveRecord::ConnectionAdapters::Transaction::Callback

def after_commit
  @callback.call if @event == :after_commit
end

def after_rollback
  @callback.call if @event == :after_rollback
end

def before_commit
  @callback.call if @event == :before_commit
end

def initialize(event, callback)
  @event = event
  @callback = callback
end