在ActiveRecord中按多组ID查询并保留分组结构

你的思路完全没问题！扁平化ID后一次性查询再映射回去，是最高效的做法（只触发一次数据库查询，避免N+1问题），具体实现可以这样：

首先，处理原ID数组，提取所有需要查询的有效ID，然后批量获取用户并转成哈希以便快速查找：

# 假设你的原ID数组是这个
id_groups = [[10], [], [10, 1, 3], []]

# 扁平化数组并过滤空值，得到所有需要查询的用户ID
target_ids = id_groups.flatten.compact

# 一次查询所有用户，转成以ID为键的哈希，方便O(1)查找
user_lookup = User.where(id: target_ids).index_by(&:id)

接下来，遍历原分组数组，把每个ID替换成对应的User对象就可以了，空数组会自动保持原样：

# 生成保留原结构的User数组
user_groups = id_groups.map do |group|
  group.map { |user_id| user_lookup[user_id] }
end

这样user_groups就是你想要的结果：[[#<User id=10>], [], [#<User id=10>, #<User id=1>, #<User id=3>], []]

额外补充几个小细节：

• 用flatten.compact是为了处理可能存在的nil值（比如原数组里如果有[nil]这种情况），避免向数据库查询无效的NULL ID。
• index_by(&:id)是Rails提供的便捷方法，能帮我们快速把ActiveRecord集合转成以主键为键的哈希，比手动循环构建哈希更简洁高效。
• 整个过程只执行一次SELECT * FROM users WHERE id IN (...)查询，完全符合Rails的性能最佳实践。

内容的提问来源于stack exchange，提问作者korotindev