Ruby on Rails按年月层级分组用户数据的实现方法

嘿，这个需求其实很好实现，咱们可以用几种方式来搞定，既可以分步来让逻辑更清晰，也能一步到位写得更简洁，甚至还能优化成数据库层面的分组来提升性能：

方法一：分步构建层级哈希（直观易读） #

先把所有用户按年份分组，得到一个以年份为键、对应年份用户数组为值的哈希；接着对每个年份下的用户数组，再按月份做二次分组，就能得到你想要的层级结构了：

# 第一步：按年份分组
year_groups = User.all.group_by { |user| user.created_at.year }

# 第二步：对每个年份的用户数组按月分组，转换为层级结构
hierarchical_groups = year_groups.transform_values do |users|
  users.group_by { |user| user.created_at.month }
end

这样得到的hierarchical_groups就是{ 年份 => { 月份 => [用户记录数组] } }的结构，逻辑非常直白，新手也能一眼看懂。

方法二：链式调用一步到位（简洁高效） #

如果喜欢精简代码，可以把上面两步合并成一行链式调用，效果完全一样：

hierarchical_groups = User.all.group_by { |u| u.created_at.year }.transform_values { |users| users.group_by { |u| u.created_at.month } }

方法三：数据库层面分组（大数据量优化） #

如果你的User表数据量很大，把所有用户对象加载到内存里做分组会很消耗资源，这时候推荐直接在数据库层面完成分组，只拉取必要的信息再组装成结构：

# 从数据库按年、月分组，获取每个年月对应的用户ID集合
monthly_user_ids = User.group("EXTRACT(YEAR FROM created_at)", "EXTRACT(MONTH FROM created_at)")
                       .pluck("EXTRACT(YEAR FROM created_at)", "EXTRACT(MONTH FROM created_at)", "ARRAY_AGG(id)")

# 把查询结果组装成层级哈希
hierarchical_groups = monthly_user_ids.each_with_object({}) do |(year, month, ids), hash|
  year = year.to_i
  month = month.to_i
  hash[year] ||= {}
  hash[year][month] = User.where(id: ids)
end

这种方式能大幅减少内存占用，数据量越大，性能优势越明显。

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