ActiveRecord子查询内部连接

时间:2015-01-16 22:44:52

标签: ruby activerecord postgis arel rgeo

我正在尝试将“原始”PostGIS SQL查询转换为Rails ActiveRecord查询。我的目标是将两个连续的ActiveRecord查询(每个需要大约1毫秒)转换为单个ActiveRecord查询(~1ms)。使用下面的SQL和ActiveRecord::Base.connection.execute,我能够验证时间的减少。

因此,我的直接请求是帮助我将此查询转换为ActiveRecord查询(以及执行它的最佳方式)。

SELECT COUNT(*)
FROM "users"
INNER JOIN (
  SELECT "centroid"
  FROM "zip_caches"
  WHERE "zip_caches"."postalcode" = '<postalcode>'
) AS "sub" ON ST_Intersects("users"."vendor_coverage", "sub"."centroid")
WHERE "users"."active" = 1;

注意,值<postalcode>是此查询中唯一的可变数据。显然,这里有两个模型UserZipCacheUserZipCache没有直接关系。

当前的两步ActiveRecord查询如下所示。

zip = ZipCache.select(:centroid).where(postalcode: '<postalcode>').limit(1).first
User.where{st_intersects(vendor_coverage, zip.centroid)}.count

2 个答案:

答案 0 :(得分:14)

Disclamer:我从未使用过PostGIS

在您的最终请求中,您似乎错过了WHERE "users"."active" = 1;部分。

这是我要做的:

首先在用户上添加active范围(用于可重用性)

scope :active, -> { User.where(active: 1) }

然后,对于实际查询,您可以在不执行子查询的情况下使用子查询,并在User模型的连接中使用它,例如:

subquery = ZipCache.select(:centroid).where(postalcode: '<postalcode>')
User.active
    .joins("INNER JOIN (#{subquery.to_sql}) sub ON ST_Intersects(users.vendor_coverage, sub.centroid)")
    .count

这允许最小的原始SQL,同时只保留一个查询。

在任何情况下,通过将logger级别设置为debug来检查控制台/日志中的实际sql请求。

答案 1 :(得分:2)

令人惊奇的工具scuttle.io非常适合转换这些类型的查询:

User.select(Arel.star.count).where(User.arel_table[:active].eq(1)).joins(
  User.arel_table.join(ZipCach.arel_table).on(
    Arel::Nodes::NamedFunction.new(
      'ST_Intersects', [
        User.arel_table[:vendor_coverage], Sub.arel_table[:centroid]
      ]
    )
  ).join_sources
)