class ActiveRecord::FutureResult::EventBuffer

def flush
  events, @events = @events, []
  events.each do |event|
    event.payload[:lock_wait] = @future_result.lock_wait
    ActiveSupport::Notifications.publish_event(event)
  end
end

def initialize(future_result, instrumenter)
  @future_result = future_result
  @instrumenter = instrumenter
  @events = []
end

def instrument(name, payload = {}, &block)
  event = @instrumenter.new_event(name, payload)
  @events << event
  event.record(&block)
end