Rails：在使用 group by 子句检索记录时更新多条记录单列值

Question

我正在使用 group by 子句从表中检索一些数据，之后我想为初始查询检索到的所有记录更新单个列值。

所以我需要确保从初始查询到我更新记录之前没有添加新行。

解释：假设模型 Test 有 5 列：Id A B C Status

所以我首先查询如下：

Test.select('count(*), A, B').where(status:'new', C: 'value').group('A, B')

在我处理完代码中的组后，我想将初始查询中 used 的所有记录的状态更改为 done

问题是，如果我只是这样做：

Test.where(status: 'new', C: 'value')

为了现在使用update_all，自第一个回答此条件的查询以来，有可能将新记录添加到表中，我会将它们的状态设置为“完成”，尽管它们并没有真正被处理，因为在初始查询它们不存在。

所以基本上我需要一种方法来锁定从初始查询到更新的插入表，或者以某种方式获取在查询组中“考虑”的 id 列表。

我该怎么做？

Answer 1

我建议在第一个查询中也从 select 语句中选择 id。

Test.select('count(*), A, B, Id').where(status:'new', C: 'value').group('A, B').

从上述查询中选择 ID。然后使用另一个查询来更新它们。

Test.where("Id IN ?", ids ).update_all()

根据您的要求进行必要的更改。

Answer 2

group_by 和 update_all (Rails 7+)

Starting from Rails 7, there is an ability to use group_by and having with update_all.

所以现在，可以编写如下查询：

Post.joins(:comments).group("posts.id").having("count(comments.id) >= 2").update_all

生成以下 SQL (query may differ depending on DBMS)：

UPDATE "posts" SET "flagged" = ? WHERE "posts"."id" IN (
  SELECT "posts"."id" FROM "posts" INNER JOIN "comments" ON "comments"."post_id" = "posts"."id" 
  GROUP BY posts.id HAVING (count(comments.id) >= 2)
)  [["flagged", "t"]]

它在单个事务中选择和更新记录。

来源：

• Pull Request。
• Rails 7 adds the ability to delete and update queries using GROUP_BY and HAVINGSaeloun。