module ActiveRecord::Railties::JobRuntime

def instrument(operation, payload = {}, &block)
  if operation == :perform && block
    super(operation, payload) do
      db_runtime_before_perform = ActiveRecord::RuntimeRegistry.sql_runtime
      result = block.call
      payload[:db_runtime] = ActiveRecord::RuntimeRegistry.sql_runtime - db_runtime_before_perform
      result
    end
  else
    super
  end
end