ruby-on-rails - ActiveRecord::Relation#explain 总是完整地运行查询

Question

我尝试在 Rails 3 和 4 中使用 explain 方法来估计在一个特别昂贵的查询中返回的行数。它连接 3 个表，可能导致 1000 万行表的表扫描，结合 count() 聚合特别慢(数据库是 Postgres 9.3)。

我的问题是这样的。如果我使用内置的 explain() 方法，查询总是在返回结果之前在幕后完整运行。这可能需要 2 分钟多的时间。在其他情况下，我要分析的查询可能需要数小时才能运行(例如报告)。

我有一个稍微丑陋的解决方案，我执行一个 to_sql，在前面添加“解释”，然后执行查询。这适用于 Rails 3，但需要对 Rails 4 进行一些修改。

所以我想我的问题是这样的。有没有办法让内置的 AR explain() 方法做我想做的事，有没有其他优雅的方法来做到这一点，或者这是 AR::explain() 中的一个错误，需要在某些时候记录和修复点？

Answer 1

我是这样操作的。在 Rails 3 和 4 中，我都为 ActiveRecord::Relation 编写了初始化程序。

首先，在 Rails 3 中:

class ActiveRecord::Relation
  HUGE_COUNT = 20000

  def count(column_name = nil, options = {})
    exact, has_conditions = false, false
    h = (column_name.class == Hash ? column_name : options)
    exact = h[:exact]
    has_conditions = h[:conditions]
    has_distinct = (column_name.class == String) && (column_name =~ /\bdistinct\b/i)
    h = h.except(:exact) # Remove it because super won't understand it
    column_name.class == Hash ? column_name = h : options = h
    if exact || has_conditions || has_distinct
      super
    else
      est = estimated_count
      est > HUGE_COUNT ? est : super
    end
  end

  def estimated_count
    node = connection.execute("EXPLAIN #{self.to_sql}").first
    match = node['QUERY PLAN'].match(/rows=\d+\b/)
    match ? match[0].split('=').last.to_i : 0
  end

结束

Rails 4 相同，除了:

  def estimated_count
    node = {}
    connection.unprepared_statement do
      node = connection.execute("EXPLAIN #{self.to_sql}").first
    end
    match = node['QUERY PLAN'].match(/rows=\d+\b/)
    match ? match[0].split('=').last.to_i : 0
  end

HUGE_COUNT 很低，因为到目前为止我发现这通常非常准确，在 1% 或 2% 以内。这对我的需要很好，但显然这是相当危险的......