# The best kind of instrumentation is in the actual db provider, however we don't want to double instrument

class Mysql2::Result
  alias_method :each_without_profiling, :each
  def each(*args, &blk)
    return each_without_profiling(*args, &blk) unless @miniprofiler_sql_id

    start        = Time.now
    result       = each_without_profiling(*args,&blk)
    elapsed_time = SqlPatches.elapsed_time(start)

    @miniprofiler_sql_id.report_reader_duration(elapsed_time)
    result
  end
end

class Mysql2::Client
  alias_method :query_without_profiling, :query
  def query(*args,&blk)
    return query_without_profiling(*args,&blk) unless SqlPatches.should_measure?

    result, record = SqlPatches.record_sql( args[0] ) do
      query_without_profiling(*args,&blk)
    end
    result.instance_variable_set("@miniprofiler_sql_id", record) if result
    result
  end
end

SqlPatches.patched = true
