如何使用MongoTemplate移除MongoDB集合中数组的对应索引元素？

需求说明 #

批量更新Collection1集合中所有满足array1包含元素"32004"的文档：

• 移除array1中的"32004"元素
• 同步移除array2中与该元素索引位置相同的元素

实现思路 #

借助MongoDB的更新管道（Update Pipeline）结合MongoTemplate的updateMulti方法完成批量操作，核心步骤：

1. 匹配出所有array1包含目标元素的文档
2. 在更新管道中计算目标元素在array1中的索引
3. 分别对array1和array2进行过滤，移除对应位置的元素

具体代码实现 #

import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import static org.springframework.data.mongodb.core.aggregation.Aggregation.*;

// 注入MongoTemplate实例
@Autowired
private MongoTemplate mongoTemplate;

public void syncRemoveArrayElements() {
    // 1. 构建查询条件：筛选array1包含"32004"的文档
    Query query = Query.query(Criteria.where("array1").in("32004"));

    // 2. 构建更新管道
    Update update = Update.fromAggregation(
        newAggregation(
            // 计算目标元素在array1中的索引，存入临时字段
            addFields()
                .addField("targetIndex")
                .withValue(indexOfArray("$array1", "32004"))
                .build(),
            // 更新array1：过滤掉目标元素
            addFields()
                .addField("array1")
                .withValue(filter("$array1", "$$this != '32004'"))
                .build(),
            // 更新array2：移除对应索引的元素
            addFields()
                .addField("array2")
                .withValue(map(
                    range(0, size("$array2")),
                    "$$idx",
                    cond(eq("$$idx", "$targetIndex"), "$$REMOVE", elementAt("$array2", "$$idx"))
                ))
                .build(),
            // 清理临时字段
            project().andExclude("targetIndex")
        )
    );

    // 3. 执行批量更新
    mongoTemplate.updateMulti(query, update, "Collection1");
}

代码说明 #

• 查询条件：通过Criteria.where("array1").in("32004")快速定位符合要求的文档
• 索引计算：用$indexOfArray获取"32004"在array1中的位置，作为后续操作的依据
• array1更新：直接用$filter排除目标元素，逻辑简洁高效
• array2更新：
  1. 用$range生成array2的完整索引序列
  2. 用$map遍历索引，匹配到目标索引时返回$$REMOVE移除对应元素，否则保留原元素
• 临时字段清理：最后通过$project移除临时生成的targetIndex，避免冗余字段残留

注意事项 #

• 要求MongoDB版本≥3.4（该版本开始支持更新管道功能）
• 上述代码仅移除array1中第一个匹配的"32004"元素；若需移除所有匹配项，可调整为基于元素值的双向过滤逻辑

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