Spring Batch使用MongoItemWriter时如何实现$addToSet的upsert逻辑？

方案1：自定义MongoItemWriter子类实现批量$addToSet操作（最推荐） #

无需完全重写MongoItemWriter的全部逻辑，仅需重写其protected修饰的doWrite方法，直接调用Spring Data MongoDB的BulkOperations实现批量upsert，全程保留Spring Batch原生的chunk批量提交、事务管理、失败回滚等能力：

• 核心实现示例：

public class AddToSetMongoItemWriter<T> extends MongoItemWriter<T> {
    private MongoOperations mongoOperations;
    private Class<T> domainClass;
    // 自定义查询条件构造函数，例如根据业务主键匹配目标文档
    private Function<T, Query> queryProvider;
    // 自定义更新规则构造函数，封装$addToSet逻辑
    private Function<T, Update> updateProvider;

    @Override
    protected void doWrite(Chunk<? extends T> chunk) throws Exception {
        // 初始化批量操作对象
        BulkOperations bulkOps = mongoOperations.bulkOps(BulkOperations.BulkMode.UNORDERED, domainClass);
        for (T item : chunk.getItems()) {
            Query query = queryProvider.apply(item);
            Update update = updateProvider.apply(item);
            bulkOps.upsert(query, update);
        }
        // 一次性提交批量操作
        bulkOps.execute();
    }

    // 省略参数初始化、getter/setter逻辑
}

• 使用时仅需传入对应业务规则即可，例如给数组字段userTags添加元素的更新构造逻辑为：Update update = new Update().addToSet("userTags", item.getCurrentTag());
• 优势：
  • 无额外查询开销，$addToSet的去重判断由Mongo服务端原子执行，不需要客户端提前查询每条数据
  • 批量操作仅产生1次Mongo交互，性能和原生MongoItemWriter一致
  • 完全复用Spring Batch的批量、事务配置能力，不需要调整原有作业的全局配置

方案2：反射复用原有私有逻辑（兼容性较差，不推荐） #

如果必须复用原有MongoItemWriter的周边处理逻辑，可以通过反射修改saveOrUpdate相关私有方法的访问权限，注入自定义的更新规则。但该方案对Spring Batch版本敏感度高，版本升级可能直接失效，稳定性远低于自定义doWrite的方案。

当前Spring Batch的MongoItemWriter确实是面向简单写入/更新场景设计，对Mongo丰富的更新操作符支持不足，自定义BulkOperations的实现是目前生产环境最通用的落地方式。

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