如何用Spring Data MongoTemplate原子实现MongoDB实体Upsert并添加嵌套数组元素

要实现你需要的原子Upsert逻辑（不存在则插入新文档，存在则往已有文档的items数组追加元素），直接用MongoDB原生更新操作配合Spring Data的MongoTemplate即可——聚合操作更适用于数据查询与转换，并非原子性文档更新的正确选择。

核心实现思路 #

利用MongoDB的updateOne方法并开启upsert选项，该操作是原子性的：MongoDB会在单个数据库操作中完成「检查目标文档是否存在→不存在则插入新文档/存在则更新数组」的逻辑，完全避免并发场景下的中间状态问题。

具体代码示例 #

假设你的实体类定义如下（简化版）：

public class MyDocument {
    private Integer id;
    private List<Item> items;
    // 构造器、Getter、Setter省略
}

public class Item {
    private String name;
    // 构造器、Getter、Setter省略
}

对应的更新代码实现：

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 org.springframework.data.mongodb.core.query.UpdateOptions;

// 注入MongoTemplate
@Autowired
private MongoTemplate mongoTemplate;

public void upsertTargetDocument(Integer targetId, Item newItem) {
    // 1. 构造查询条件：匹配id为targetId的文档
    Query query = Query.query(Criteria.where("id").is(targetId));
    
    // 2. 构造更新操作：往items数组中追加新元素
    Update update = new Update().push("items", newItem);
    
    // 3. 开启upsert选项：不存在目标文档则自动插入新文档
    UpdateOptions options = UpdateOptions.options().upsert(true);
    
    // 执行原子更新操作
    mongoTemplate.updateOne(query, update, options, MyDocument.class);
}

关键细节补充 #

• 原子性保障：updateOne配合upsert:true是MongoDB原生支持的原子操作，由数据库层面强一致性保证，不会出现多线程竞态问题。
• 批量追加元素：如果需要一次性追加多个Item，可使用push的each方法：

  List<Item> newItems = Arrays.asList(new Item("name_1"), new Item("name_2"));
  Update update = new Update().push("items").each(newItems);
  

• 避免重复元素：如果需要确保items数组中无重复元素，可将push替换为addToSet（需保证Item的字段能被MongoDB正确识别为重复项）：

  Update update = new Update().addToSet("items", newItem);
  

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