使用Mongoid执行聚合后继续查询的实现方案

这个问题我之前也碰到过，确实挺头疼的——既要保留聚合后的排序，又想继续用ActiveRecord的链式查询。给你几个可行的方案，看哪种适配你的场景：

方案1：将聚合逻辑完全迁移到数据库层面（优先推荐） #

如果你的复杂聚合操作能通过SQL语句实现（比如分组统计、自定义排序规则等），那直接用ActiveRecord的查询方法构建ActiveRecord::Relation，而不是先转成数组。这样返回的结果依然是可链式调用的关系对象，既能保留排序，又能继续执行where等查询。

举个例子，假设你的聚合是按category分组，统计每个分组的记录数，然后按记录数倒序排序：

# 构建带聚合和排序的Relation
identities = Identity.select("identities.*, COUNT(*) as record_count")
                     .group(:category)
                     .order("record_count DESC")

# 现在可以直接链式调用where，排序依然保留
filtered_identities = identities.where(status: :active)

这种方式性能最优，因为所有逻辑都在数据库层面完成，避免了Ruby层面的数据处理开销。

方案2：先筛选再按原排序重新排序 #

如果你的聚合逻辑必须在Ruby层面完成（比如涉及复杂的业务计算，没法用SQL表达），那可以先提取聚合后的排序ID列表，再基于这个ID列表做筛选，最后用数据库的排序函数还原原顺序。

MySQL 版本 #

# 先得到聚合后的数组和对应的排序ID列表
aggregated_identities = your_complex_aggregation_method # 你的聚合逻辑，返回数组
sorted_ids = aggregated_identities.map(&:id)

# 筛选并保留原排序
filtered_identities = Identity.where(something_else: true, id: sorted_ids)
                              .order("FIELD(id, #{sorted_ids.join(',')})")

PostgreSQL 版本 #

PostgreSQL没有FIELD函数，需要用array_position来实现：

filtered_identities = Identity.where(something_else: true, id: sorted_ids)
                              .order("array_position(ARRAY[#{sorted_ids.join(',')}], id)")

这种方法的核心是利用数据库的排序函数，把原数组的ID顺序作为排序依据，确保筛选后的结果和聚合后的顺序一致。

方案3：用子查询包装聚合逻辑 #

如果你的聚合逻辑可以写成子查询，那可以用from方法将子查询作为数据源，在外部继续构建查询。这样整个流程依然是ActiveRecord::Relation，支持链式调用。

比如：

# 先定义聚合排序的子查询（返回Relation，不是数组）
subquery = Identity.select("*")
                   .where(your_aggregation_conditions)
                   .order(your_sorting_logic)

# 基于子查询构建新的查询，保留排序并添加筛选条件
filtered_identities = Identity.from(subquery, :identities)
                              .where(something_else: true)

这种方法兼顾了聚合逻辑的灵活性和链式查询的便利性，适合那些可以拆分为子查询的复杂场景。

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