Spring Boot中删除MongoDB记录后如何更新优先级列表？

针对你删除configurations集合文档后需要填补priority空缺的需求，有两种更高效的方案，比全量查询再循环更新的方式更简洁且性能更好：

方案一：单条删除时仅更新后续优先级（最优） #

不需要全量查询所有文档，只需要在删除目标文档后，将所有优先级大于被删除文档优先级的记录的priority减1，直接填补空缺。这种方式仅操作必要的文档，性能最高，适合单条删除的场景。

代码实现 #

@Transactional // 开启事务保证原子性，需MongoDB 4.0+及配置事务管理器
public void deleteAndAdjustPriority(String configId) {
    // 1. 查询要删除的文档，获取其priority
    Configuration toDelete = mongoTemplate.findById(configId, Configuration.class);
    if (toDelete == null) {
        return;
    }
    int deletedPriority = toDelete.getPriority();
    
    // 2. 删除目标文档
    mongoTemplate.remove(toDelete);
    
    // 3. 更新所有优先级大于deletedPriority的文档，优先级减1
    Query updateQuery = new Query(Criteria.where("priority").gt(deletedPriority));
    Update update = new Update().inc("priority", -1);
    mongoTemplate.updateMulti(updateQuery, update, Configuration.class);
}

方案二：全量重新整理优先级（适合批量删除/优先级混乱场景） #

如果存在批量删除或手动修改导致priority整体混乱的情况，可以用MongoDB的聚合管道给所有文档重新分配连续的优先级，再批量更新。

代码实现 #

public void resetAllPriorities() {
    // 1. 聚合查询：按原priority升序排序，生成新的连续优先级
    Aggregation aggregation = Aggregation.newAggregation(
        Aggregation.sort(Sort.Direction.ASC, "priority"),
        Aggregation.setWindowFields()
            .partitionBy("") // 全局分区，对所有文档生效
            .sortBy(Sort.Direction.ASC, "priority")
            .output("newPriority").documentNumber() // 生成从1开始的连续序号
    );
    
    List<Document> aggregatedResults = mongoTemplate.aggregate(
        aggregation, "configurations", Document.class
    ).getMappedResults();
    
    // 2. 批量更新所有文档的priority
    List<UpdateQuery> updateQueries = new ArrayList<>();
    for (Document doc : aggregatedResults) {
        String docId = doc.getObjectId("_id").toHexString();
        int newPriority = doc.getInteger("newPriority");
        
        UpdateQuery updateQuery = UpdateQuery.query(Criteria.where("_id").is(docId))
            .apply(Update.update("priority", newPriority));
        updateQueries.add(updateQuery);
    }
    
    if (!updateQueries.isEmpty()) {
        mongoTemplate.bulkUpdate(updateQueries, Configuration.class);
    }
}

原方案的问题对比 #

你的原方案需要全量查询所有文档，循环修改后再批量更新，当集合数据量大时，会占用较多内存和数据库带宽；而上面的方案一仅操作需要调整的部分文档，性能提升明显，且通过事务保证了删除和更新的原子性，避免数据不一致。

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