ruby - 使用 find_by_sql 复制包含和连接行为(单个查询建立关联)

Question

我们有一个相当简单的查询，可以与 AR 查询结构完美配合:

user = User.includes({:network => :product}).joins({:network => :product}).where(:id => user_id).first

这对数据库执行单个查询，填充急切加载的集合(即:我可以调用 user.product.network.name 而无需调用对数据库的额外查询)。

我们想向用户表中不存在的 User 对象添加一些“伪”列，但我们希望作为此初始查询的一部分进行填充。 SQL 看起来像这样:

SELECT 
    "users".*,
    "networks".*,
    "products".*, 
    (select count("spaces".id) from spaces "spaces" where "spaces".owner_id = "users".id) as spaces_count
FROM
    users "users" 
JOIN
    networks "networks" 
    on "networks".id = "users".network_id 
JOIN products "products" 
    on "products".id="networks".product_id 
WHERE "users".id=?

添加带有(未优化的)内部查询的“spaces_count”列是现阶段唯一的“pseduo”列，但 View 将添加更多。

在 User 类(一个 ActiveRecord 模型)中有:

attr_accessible :spaces_count

为了执行 SQL 查询，我们使用以下内容:

user = find_by_sql(sql)
ActiveRecord::Associations::Preloader.new([user], {:network => :product}).run

其中 sql 是上面包含的查询。

结果是一个 User 模型，按预期填充，包括 spaces_count “伪”列，但是作为调用 run 的一部分在关联预加载器上，它又两次访问数据库:

1.9.3p125 :465 > ActiveRecord::Associations::Preloader.new([user], {:network => :product}).run
  Network Load (0.5ms)  SELECT "networks".* FROM "networks" WHERE "networks"."id" IN (1)
  Product Load (0.3ms)  SELECT "products".* FROM "products" WHERE "products"."id" IN (1)
=> [{:network=>:product}] 

有没有一种方法可以使用 ActiveRecord::Associations::Preloader 急切地获取关联，而不需要额外访问数据库，按照 includes 和 joins 是吗？

我们使用的是 Rails 3.2、Ruby 1.9.3-p125 和 PG 9.1。

Answer 1

我不知道是否有办法完全按照您的要求进行操作，我知道您不必这样做 - 只需使用更多这些 AR“查询结构”即可。

首先，您可以指定一个自定义选择:

users = User.includes({:network => :project}).select('users.*, (select count("spaces".id) from spaces "spaces" where "spaces".owner_id = "users".id) as spaces_count').joins({:network => :product}).where(:id => user_id).first

您可以按任何顺序添加这些内容 - 我个人的习惯是“包含”先出现，然后按照它在查询中出现的顺序排列其他所有内容，但我就是这样。这仍然会获得您需要的预加载，并且您可以通过在返回的对象上调用它们的名称来获取“伪列”的值:

users.first.spaces_count

注意PostgreSQL 适配器似乎将这些值作为字符串返回，无论它们在数据库中的实际类型是什么，因此您可能需要在之后转换它们:

users.first.spaces_count.to_i

最后一点，虽然我希望这个解决方案对你有用，但我也对你建议的方法的答案感兴趣——通过 AR::Associations::Preloader 自定义预先加载听起来既酷又有用。