Rails 5.1中使用group_by分组后如何统计分组元素数量？

嘿，这个需求我太熟了！你现在用的User.all.group_by{ |a| a.created_at.year }是在Ruby内存层面做分组，其实Rails的ActiveRecord提供了更高效的原生分组统计方案——直接借助数据库的GROUP BY和COUNT功能完成，不用把所有数据都加载到内存里，性能提升特别明显，尤其是数据量大的时候。

最优实现：数据库层面分组统计 #

直接用ActiveRecord的group和count方法组合，针对created_at的年份做分组：

适配多数据库的通用写法 #

如果想兼容PostgreSQL、MySQL、SQLite等多数数据库，可以用EXTRACT函数：

User.group("EXTRACT(YEAR FROM created_at)").count

这个会返回一个哈希，键是年份（浮点数格式，比如2023.0），值是对应年份的用户数量。如果需要把键转成整数，可以再加一步处理：

User.group("EXTRACT(YEAR FROM created_at)").count.transform_keys(&:to_i)

针对特定数据库的更简洁写法 #

• MySQL/SQLite：直接用YEAR函数更直观，返回的哈希键就是整数年份，不用额外转换：

  User.group("YEAR(created_at)").count
  

• PostgreSQL：也可以用DATE_TRUNC精确截取年份（返回的是日期对象，比如Fri, 01 Jan 2021 00:00:00 UTC +00:00），如果只需要年份数字的话，配合EXTRACT就好：

  User.group("DATE_TRUNC('year', created_at)").count
  

和你当前写法的对比 #

你原来的写法是先把所有User记录加载到内存，再用Ruby的group_by分组，最后统计数量还需要额外处理：

User.all.group_by{ |user| user.created_at.year }.transform_values(&:count)

这种方式在数据量小的时候没问题，但当用户数量上万甚至更多时，会把所有记录都塞进内存，性能会明显下降。而数据库层面的group + count只需要数据库返回分组后的统计结果，内存占用小、速度快，才是更推荐的Rails原生方案。

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