Mongoid返回文档计数错误问题排查

问题本质 #

你遇到的核心问题是：result.value[:records]是Mongoid Criteria对象（未执行的查询链），而非内存中的数组：

• 调用to_a.count时，会先执行查询把文档加载成数组，再统计数组长度，得到的是分页后的实际结果数20。
• 调用count时，Mongoid会忽略之前的skip/limit分页参数，直接对原始查询条件（state: "created"）执行全量计数，因此返回总文档数25。

Mongoid的count方法底层对应MongoDB的countDocuments，默认不会考虑查询链中的分页限制，这是和ActiveRecord的核心差异点。

解决办法 #

1. 测试断言替换为size方法 #

Mongoid的Criteria#size会识别查询链中的skip/limit参数：

• 如果文档已加载到内存，直接返回数组长度；
• 如果未加载，会执行带分页限制的计数查询（db.collection.find(...).limit(20).count()）。

修改测试代码：

it "returns paginated documents" do
  expect(result.value[:records].to_a.count).to eq(20)
  expect(result.value[:records].size).to eq(20) # 替换count为size
end

2. 优化Pagy适配代码（可选） #

如果希望业务代码中调用count也能返回分页后的数量，可以修改Pagy扩展，让分页后的集合直接返回加载后的数组：

module PagyExtension
  module Mongoid
    def pagy_get_vars(collection, vars)
      vars[:count] = collection.count
      super
    end

    # 重写pagy方法，返回加载后的数组
    def pagy(collection, vars={})
      pagy = Pagy.new(pagy_get_vars(collection, vars))
      [pagy, collection.skip(pagy.offset).limit(pagy.items).to_a]
    end
  end
end

Pagy::Backend.prepend PagyExtension::Mongoid

这样返回的records就是数组，调用count会直接返回数组长度，但注意这会提前加载所有分页文档到内存，适合小数据量场景。

补充说明 #

Mongoid官方文档明确指出：count用于统计符合条件的总文档数，size用于统计当前查询链（含分页）的结果数，length等价于to_a.length。根据场景选择合适的方法即可。

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