一次查询 ActiveRecord 的记录和关系计算

Question

TL；博士？请参阅编辑 2

我有一个小的 Rails 应用程序，其中包含人们可以玩的几种不同类型的游戏：它基于体育运动，因此他们可以每周挑选每场比赛的获胜者（模型 PickEm，属性 correct boolean nil 表示未完成的比赛），并预测特定球队比赛的结果（模型Guess，属性score 为整数，nil 表示未完成的比赛）。每个Userhas_manyPickEms 和Guesses。而且我正在尝试显示排名（correct/total - 总数都不是nil，score/total 可能）。

我发现我可以收集用户及其相关记录，但在尝试显示排名时，我发现每个用户都在触发另一个查询 - 随着用户群的增加，速度缓慢且不可持续。那是因为@user.pick_em_score 是pick_ems.where(correct: true).size 而@user.guess_Score 是guesses.where.not(score: nil).sum(:score)。所以我打电话给user.pick_em_score，它会运行那个查询。我觉得应该有一种方法可以一次获取每个 User 以及这些特定计数，而不是缓冲一大堆不必要的额外内容。

我需要什么：

• User记录
• User.pick_em_score（计算correct记录）
• User.pick_ems 数NOT NULL
• User.guesses_score（由guesses.sum(:score)计算）
• User.guesses 数NOT NULL

我在 Rails 的 ActiveRecord 助手中找到的大部分内容，尤其是与计算相关的内容，都是仅用于检索计算的。看起来我可能需要直接深入研究select() 等。但我无法让它工作。有人能指出我正确的方向吗？

编辑

澄清一下：我知道我可以将此信息写入User 模型，但这过于严格：下个赛季，我需要在User 中添加一个新列以供 那一年的结果等等。此外，这是第三级回调更新相关模型——Match 模型在保存时已经更新了相关的PickEms 和Guesses。 我正在寻找能够处理此信息的最简单的 ActiveRecord 查询或查询，如标题所示。理想情况下，一个查询可以返回上述信息，但如果需要几个，没关系。

我以前使用 PHP 直接在 MySQL 中工作，但这些技能已经生锈了（我想，在原始 MySQL 中，我会有几个子选择语句来帮助提取这些计数），我也喜欢 能够使用 Rails 的 ActiveRecord 助手等，并尽可能避免构建原始 SQL。

第二次编辑：

我似乎将其归结为 开始 工作的一次调用，但我正在编写大量 SQL。 它也很脆弱，IMO，并试图运行它失败了。看起来我只是将 Rails 中的百万单数 SELECT 查询直接推送到 SQL 中，但这可能仍然是一个进步。

User.unscoped.select('users.*',
  '(SELECT COUNT(*) FROM pick_ems WHERE pick_ems.user_id = users.id AND pick_ems.correct) AS correct_pick_ems',
  '(SELECT COUNT(*) FROM pick_ems WHERE pick_ems.user_id = users.id AND pick_ems.correct IS NOT NULL) AS total_pick_ems',
  '(SELECT SUM(guesses.score) FROM guesses WHERE guesses.user_id = users.id AND guesses.score IS NOT NULL) AS guesses_score',
  '(SELECT COUNT(*) FROM guesses WHERE guesses.user_id = users.id AND guesses.score IS NOT NULL) AS guesses_count' )

问题似乎是：有没有办法使用 Rails 而不是原始 SQL 将我们看到的 users.id 与这些子查询联系起来？或者只是……一般来说是一种更好的构建方式？

此外，我正在为WHERE 运行另一组SELECTs，这将取决于total_pick_ems 和guesses_count 是> 0，但由于我不能使用那些别名列，我必须再拨打SELECT 一次。

Answer 1

欢迎来到 AR。它真的只适用于简单的 CRUD 之类的查询。一旦您真的想愤怒地查询您的数据，它就没有能力执行您想要的查询，而无需求助于批发 SQL 字符串，因此通常会放弃链接的能力。

这正是我转向 Sequel 的原因，因为它确实具有使用更完整的 SQL 功能集组合查询的功能，包括连接条件、窗口函数、递归公用表表达式和高级预加载。与 AR 和 Arel 相比，作者反应迅速，文档也非常出色。

我不希望您会喜欢这个答案，但总有一天，您会开始将目光投向带有导轨的自以为是的组件之外，我不得不说这些组件几乎不是同类产品中最好的。 Sequel 还使我的应用程序速度比我使用 AR 能够获得的速度快很多倍，这不仅是开发人员的快乐，还意味着可以运行的服务器更少。是的，这将是一个学习曲线，但 IMO 最好学习有你支持的工具。

Answer 2

连接可能会起作用。如下所示

User.unscoped.joins(:guesses).joins(:pick_ems).
where("guesses.score IS NOT NULL").
select("users.*, 
sum(guesses.score) as guesses_score,
count(guesses.id) as guesses_count,
count(case when pick_ems.correct = True then 1 else null end) 
as correct_pick_ems,
count(case when pick_ems.correct != null then 1 else null end)
as total_pick_ems,
").
group("users.id")

如果您一次需要为有限数量的用户提供此信息，请使用类方法进行查询或预先加载 (User.includes(:guesses, :pick_ems))，例如

def correct_pick_ems
   pick_ems.count(&:correct)
end

会起作用的。

但是，如果您大部分时间都需要所有用户的这些信息，那么用户表中的缓存计数器会更理想。

Answer 3

您需要的是某种自定义（智能）counter_cache，仅在特定条件下计数（例如，正确为真）

您可以使用conditional after_save 和after_destroy 触发器来构建您自己的自定义counter_cache，如下所示：

class PickEm
  belongs_to :user

  after_save :increment_finished_counter_cache, if: Proc.new { |pick_em| pick_em.correct }
  after_destroy :decrement_finished_counter_cache, if: Proc.new { |pick_em| pick_em.correct }

  private

  def increment_finished_counter_cache
    self.user.update_column(:finished_games_counter, self.user.finished_games_counter + 1) #update_column should not trigger any validations or callbacks
  end

  def decrement_finished_counter_cache
    self.user.update_column(:finished_games_counter, self.user.finished_games_counter - 1) #update_column should not trigger any validations or callbacks
  end
end

注意事项：

• 代码未经测试（仅展示想法）

• 有些人说最好避免将自定义计数器命名为 rails 命名它们 (foo_counter_cache)

Answer 4

您应该对其进行基准测试，但我的直觉是，将所有这些数据添加到单个 SELECT 不会比将其分解为单独的 SELECTs 快得多（我实际上有过这样的情况后者更快）。通过分解它，您还可以坚持更多的 ActiveRecord 和更少的原始 SQL，例如：

user_ids_to_pick_em_score = User.joins(:pick_ems).where(pick_ems: {correct: true}).group(:user_id).count
user_ids_to_pick_ems_count = User.joins(:pick_ems).where.not(pick_ems: {correct: nil}).group(:user_id).count
user_ids_to_guesses_score = Hash[User.select("users.id, SUM(guesses.score) AS total_score").joins(:guesses).group(:user_id).map{|u| [u.id, u.total_score]}]
user_ids_to_guesses_count = User.joins(:guesses).where.not(guesses: {score: nil}).group(:user_id).count

编辑：要显示它们，您可以这样做：

<%- User.select(:id, :name).find_each do |u| -%>
  Name: <%= u.name %>
  Picks Correct: <%= user_ids_to_pick_em_score[u.id] %>/<%= user_ids_to_pick_ems_count[u.id] %>
  Total Score: <%= user_ids_to_guesses_score[u.id] %>/<%= user_ids_to_guesses_count[u.id] %>
<%- end -%>