module ActiveRecord::RuntimeRegistry

def async_sql_runtime
  ActiveSupport::IsolatedExecutionState[:active_record_async_sql_runtime] ||= 0.0
end

def async_sql_runtime=(runtime)
  ActiveSupport::IsolatedExecutionState[:active_record_async_sql_runtime] = runtime
end

def cached_queries_count
  ActiveSupport::IsolatedExecutionState[:active_record_cached_queries_count] ||= 0
end

def cached_queries_count=(count)
  ActiveSupport::IsolatedExecutionState[:active_record_cached_queries_count] = count
end

def queries_count
  ActiveSupport::IsolatedExecutionState[:active_record_queries_count] ||= 0
end

def queries_count=(count)
  ActiveSupport::IsolatedExecutionState[:active_record_queries_count] = count
end

def reset
  reset_runtimes
  reset_queries_count
  reset_cached_queries_count
end

def reset_cached_queries_count
  qc = cached_queries_count
  self.cached_queries_count = 0
  qc
end

def reset_queries_count
  qc = queries_count
  self.queries_count = 0
  qc
end

def reset_runtimes
  rt, self.sql_runtime = sql_runtime, 0.0
  self.async_sql_runtime = 0.0
  rt
end

def sql_runtime
  ActiveSupport::IsolatedExecutionState[:active_record_sql_runtime] ||= 0.0
end

def sql_runtime=(runtime)
  ActiveSupport::IsolatedExecutionState[:active_record_sql_runtime] = runtime
end