如何在Mongoid中使用$in？RSpec关联查询结果异常求助

核心问题排查 #

你遇到的查询返回全部25条数据，大概率是两个常见错误导致的：

1. Mongoid查询对象未正确赋值 #

Mongoid的查询方法是不可变的，每次调用.in或.where都会返回一个新的查询对象，而非修改原对象。如果你的代码是这样写的：

query = Ticket.all
query.in(origin_id: origins) # 这里只是生成新查询，但没赋值给query
results = query.to_a # 仍然是Ticket.all的结果

那必然会返回所有数据。

2. 测试数据初始化错误（let的惰性加载坑） #

RSpec的let是惰性求值的——只有当你显式调用first_origin/second_origin时，对应的Origin对象才会被创建。如果你的create_list直接写在let定义之外（而非before块中并触发变量调用），那create_list(:ticket, 10, origin: first_origin)中的first_origin会是nil，导致前20条Ticket的origin_id也为nil，和后面5条无差。此时用origin_id.in(origins)查询时，如果origins数组因变量未初始化而包含nil，就会匹配所有origin_id为nil的25条数据。

修复步骤 #

第一步：修正测试数据初始化 #

把数据创建逻辑放到before块中，并确保Origin对象被实例化：

# 方式1：保留let，显式触发变量
let(:first_origin) { create(:origin, owner: "123@tenant") }
let(:second_origin) { create(:origin, owner: "123@tenant") }

before do
  # 调用let变量，触发Origin对象创建
  first_origin
  second_origin
  create_list(:ticket, 10, origin: first_origin)
  create_list(:ticket, 10, origin: second_origin)
  create_list(:ticket, 5)
end

# 方式2：直接在before中创建，避免惰性问题
before do
  first_origin = create(:origin, owner: "123@tenant")
  second_origin = create(:origin, owner: "123@tenant")
  create_list(:ticket, 10, origin: first_origin)
  create_list(:ticket, 10, origin: second_origin)
  create_list(:ticket, 5)
end

第二步：修正查询语句 #

确保链式调用的查询结果被正确赋值：

# 写法1：一步链式调用
results = Ticket.in(origin_id: [first_origin.id, second_origin.id]).to_a

# 写法2：分步赋值
query = Ticket.all
query = query.in(origin_id: [first_origin.id, second_origin.id])
results = query.to_a

更优雅的关联查询方式 #

不需要手动处理ID数组，直接利用Mongoid的关联查询：

# 直接传入Origin对象数组
results = Ticket.where(origin: [first_origin, second_origin]).to_a

# 或者先筛选租户的Origin，再关联查询
tenant_origins = Origin.where(owner: "123@tenant")
results = Ticket.where(origin: tenant_origins).to_a

验证数据正确性 #

可以在测试中加入调试代码，确认Ticket的origin_id是否符合预期：

puts Ticket.pluck(:origin_id).uniq
# 预期输出：两个BSON::ObjectId值 + nil

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