按条件查询文档：超N取第N条、不足N取最后一条的单查询实现

需求说明 #

按指定条件查询数据时，若符合条件的总记录数大于N，则取第N条记录；若总记录数小于N，则直接取符合条件的最后一条记录。 #

现有实现思路及问题分析 #

预设变量：

N = 100
query = {} # 自定义查询条件

• 方案1

Model.where(query).limit(N).to_a.last

问题：执行效率极低，会将前N条记录全部加载到内存后再取最后一条，数据量较大时会产生严重的性能损耗。

• 方案2

Model.where(query).limit(N).last

问题：无法生效，ActiveRecord 中的last方法会自动生成limit(1)查询逻辑，直接覆盖之前设置的limit(N)。

• 方案3

Model.where(query).skip(N-1).first

问题：仅在符合条件的总记录数≥N时能返回正确结果，总记录数小于N时会返回空值。

• 方案4

m = Model.where(query).skip(N-1).first
m = Model.where(query).last if !m

问题：可正常返回结果，但需要执行两次数据库查询，性能低于单次查询方案。 #

最优单次查询实现 #

适用ActiveRecord（Rails）场景 #

核心逻辑：先按业务排序规则取前N条记录，再将这部分结果反向排序取第一条，单次查询即可满足需求：

# 可根据业务需要调整排序字段和排序规则
subquery = Model.where(query).order(id: :asc).limit(N).select(:id)
result = Model.where(id: subquery).order(id: :desc).first

逻辑说明：

1. 若符合条件的总记录数≥N，第一步取到的是前N条记录，反向排序后的第一条就是原排序规则下的第N条
2. 若符合条件的总记录数<N，第一步取到的是全部符合条件的记录，反向排序后的第一条就是原排序规则下的最后一条
3. 全程仅执行一次数据库查询，无多余内存开销，性能最优。

适用MongoDB场景 #

使用聚合查询实现，逻辑完全一致：

db.collection.aggregate([
  { $match: query },
  { $sort: { _id: 1 } }, // 按业务要求调整排序规则
  { $limit: N },
  { $sort: { _id: -1 } },
  { $limit: 1 }
])

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