ActiveRecord 限制从 company_ids 集合返回的用户

Question

我有一个 company_id 数组
在User模型中，我想find_by company_ids，但是对于每个company_id，返回的用户列表不能超过5个用户。我该怎么做。

我试过这样做

User.where(company_id: [1, 2, 3]).group('users.company_id').having('count(company_id) <= 5')

但这只会返回公司用户不超过5人的集合。而且，这不会返回所有记录。

我想要做的是，对于company_ids = [1, 2, 3]，该集合应该能够为 company_id 1、company_id 2、company_id 3 返回最多 5 个用户，而不是为所有这些 company_id 加载所有用户。

Answer 1

由于您的 MySQL 版本，我认为您可以进一步使用 UNION ALL。

您可以创建一个查询，通过company_id 选择用户，然后应用LIMIT：

(select * from users where company_id = 1 LIMIT 2)
UNION ALL
(select * from users where company_id = 2 LIMIT 2)
UNION ALL
(select * from users where company_id = 3 LIMIT 2)

由于 Rails 没有对 UNION 查询的内置支持，您可以创建自己的语句：

sql = [1, 2, 3].map { |id| "(SELECT * FROM users WHERE company_id = #{id} LIMIT 2)" }
               .join(" UNION ALL ")
User.find_by_sql(sql)

---

使用 MySQL8+ 更新

SELECT id, name, company_id
FROM (
  SELECT id, name, company_id, ROW_NUMBER() OVER w AS company_user
  FROM users
  WINDOW w AS (PARTITION BY company_id)
) users
WHERE company_user <= 2;

ActiveRecord 方式是：

User
  .select(:id, :name, :company_id)
  .from(
    User.select(:id, :name, :company_id, 'ROW_NUMBER() OVER w AS company_user')
        .from('users WINDOW w AS (PARTITION BY company_id)')
  )
  .where('company_user <= 2')