在Ruby on Rails+MongoDB中合并两个查询集

你遇到的问题很典型——当需要跨不同集合查询并实现数据库级分页时，直接用数组拼接会丢失Mongoid Criteria的优势，无法在数据库层面做分页优化。下面给你两种可行的解决方案，根据你的项目情况选择：

方案一：使用MongoDB聚合框架的$unionWith（无需修改现有模型） #

MongoDB 4.4及以上版本支持$unionWith操作符，可以直接在数据库层面合并两个集合的查询结果，然后添加分页阶段。这是最适合你现有结构的方案，不需要改动模型和数据。

实现代码： #

首先构建聚合管道，合并两个集合的匹配结果，再添加排序和分页逻辑：

client_id = "58358494767f430fc4154fe8"
page = 1 # 当前页码
per_page = 10 # 每页条数

pipeline = [
  # 匹配CoachTrainingSession中符合条件的记录，并添加来源标识
  { "$match" => { "client_ids" => client_id } },
  { "$addFields" => { "source" => "coach" } },
  # 合并OrganizationTrainingSession的匹配结果，同样添加来源标识
  { "$unionWith" => {
      "coll" => "organization_training_sessions", # 注意集合名是模型的复数形式
      "pipeline" => [
        { "$match" => { "client_ids" => client_id } },
        { "$addFields" => { "source" => "organization" } }
      ]
    }
  },
  # 按时间倒序排序（分页前必须排序，否则结果顺序不稳定）
  { "$sort" => { "date_time" => -1 } },
  # 分页：跳过前(page-1)*per_page条，取per_page条
  { "$skip" => (page - 1) * per_page },
  { "$limit" => per_page }
]

# 从任意一个模型的collection执行聚合
raw_results = CoachTrainingSession.collection.aggregate(pipeline)

# 将原始文档转换为对应的模型实例
combined_sessions = raw_results.map do |doc|
  case doc["source"]
  when "coach"
    CoachTrainingSession.new(doc.except("source"))
  when "organization"
    OrganizationTrainingSession.new(doc.except("source"))
  end
end

关键说明： #

• $unionWith会跨集合合并查询结果，返回的是MongoDB原始文档，所以需要手动转换成Mongoid模型实例，添加source字段是为了区分记录来自哪个模型。
• 确保你的MongoDB版本≥4.4，否则这个操作符不可用。
• 你已经给两个模型的client_ids字段加了索引，这会让$match阶段的查询效率很高，不用担心性能问题。

方案二：使用单表继承（STI）（适合可重构模型的场景） #

如果你的项目还处于早期，或者可以接受数据迁移，那么用Mongoid的单表继承（STI）是更优雅的方案。将两个模型抽象为同一个父类的子类，所有数据存储在同一个集合中，这样查询和分页就和普通模型一样简单。

重构后的模型代码： #

# app/models/training_session.rb
class TrainingSession
  include Mongoid::Document
  include Mongoid::Timestamps
  
  # 公共字段
  field :session_id
  field :date_time, type: DateTime
  field :completed, type: Boolean, default: false
  field :cancelled, type: Boolean, default: false
  field :client_ids, type: Array
  field :group_name
  
  # 公共索引
  index({ client_id: 1 })
  index({ session_id: 1 })
  index({ date_time: 1 })
  index({ client_ids: 1 })
  index({ group_name: 1 })
end

# app/models/coach_training_session.rb
class CoachTrainingSession < TrainingSession
  belongs_to :coach
  has_many :booked_sessions
end

# app/models/organization_training_session.rb
class OrganizationTrainingSession < TrainingSession
  belongs_to :organization
  field :coach_ids, type: Array
  field :recurring_sessions_id
  
  # 子类专属索引
  index({ completed: 1 })
  index({ cancelled: 1 })
  index({ coach_ids: 1 })
end

查询和分页代码： #

现在你可以直接用父类查询所有训练会话，并且使用Mongoid自带的分页方法（比如page和per，需要kaminari或will_paginate gem支持）：

client_id = "58358494767f430fc4154fe8"
combined_sessions = TrainingSession.where(client_ids: client_id)
                                   .order(date_time: :desc)
                                   .page(1)
                                   .per(10)

优缺点： #

• 优点：查询逻辑极简，完全利用Mongoid的Criteria特性，分页、排序都很方便，不需要额外处理文档转换。
• 缺点：需要迁移现有数据到新的集合，并且两个模型的所有字段会存在同一个集合中，可能会有一些冗余字段，但Mongoid的STI会自动处理_type字段来区分子类。

总结 #

• 如果你不能改动现有数据结构，优先选方案一的$unionWith聚合方法。
• 如果你可以重构模型，方案二的STI会让代码更简洁易维护。

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