Spring MongoDB如何实现计数低于阈值时返回结果的原子操作

下面提供两种可直接落地的实现，都不存在竞态问题：

方案1：单次查询限制条数（推荐，性能最优） #

直接在查询时限制最多返回1001条数据，通过返回结果的长度判断是否超出阈值。MongoDB单次查询基于一致性快照执行，执行过程中不会插入其他写入操作的结果，完全规避两次请求之间的竞态。
代码实现如下：

// Java 版本
List<Document> results = mongoTemplate.find(
    myQuery.limit(1001), 
    Document.class
);
if (results.size() > 1000) {
    throw new ResultSetTooLargeException();
}
return results;

// Kotlin 版本
val results = mongoTemplate.find(
    myQuery.limit(1001), 
    Document::class.java
)
if (results.size > 1000) {
    throw ResultSetTooLargeException()
}
return results

该方案仅需一次数据库请求，性能远高于计数+查询的两次请求模式，99%的场景下都适用。

方案2：聚合操作同时返回计数和结果（适合需要拿到真实总计数的场景） #

如果你的业务逻辑需要拿到符合条件的文档总数量（而非仅判断是否超阈值），可以使用MongoDB的$facet聚合操作，一次请求同时拿到计数和结果集，天然原子：

val aggregation = newAggregation(
    match(myQuery.queryObject),
    facet(
        count().`as`("totalCount")
    ).`as`("countInfo"),
    facet(
        limit(1000)
    ).`as`("results")
)
val result = mongoTemplate.aggregate(aggregation, Document::class.java, AggregationResult::class.java).uniqueMappedResult
if (result.countInfo[0].totalCount > 1000) {
    throw ResultSetTooLargeException()
}
return result.results

不推荐的方案 #

• 分布式锁/Redis锁：可以解决竞态问题，但会引入额外依赖，性能损耗大，仅在有特殊读写控制需求时考虑。
• MongoDB事务：4.0以上版本支持多操作事务，但性能差，部署要求高（需要副本集/分片集群），没必要为这个场景使用。

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