MyBatis-Plus批量保存或更新性能优化实践

背景介绍

MyBatis-Plus作为MyBatis的增强工具，在日常开发中提供了诸多便利功能。其中saveOrUpdateBatch方法是一个常用的批量操作方法，它能够智能判断数据是新增还是更新。然而在实际使用中，该方法在处理大批量数据时可能会遇到性能瓶颈。

问题分析

在MyBatis-Plus 3.5.4.1版本中，saveOrUpdateBatch方法的实现存在以下性能问题：

1. 单条校验效率低：方法内部对每条记录都单独执行查询操作来判断是否需要更新，当处理大批量数据时，会产生大量数据库查询请求。

2. 全量数据传输浪费：即使只需要判断记录是否存在，方法也会查询并返回完整的实体数据，造成了不必要的数据传输和处理开销。

优化方案

针对上述问题，我们可以采用以下优化策略：

批量查询优化

将单条记录的查询改为批量查询，一次性获取所有需要判断的记录ID。这样可以显著减少数据库交互次数，提升整体性能。

选择性字段查询

通过MyBatis-Plus的Wrapper功能，我们可以只查询必要的字段（如主键ID），而不是获取完整的实体数据。这减少了数据传输量，提高了查询效率。

具体实现

优化后的实现思路如下：

1. 首先收集批量数据中的所有主键ID
2. 使用Wrapper构建只查询主键字段的批量查询
3. 执行批量查询获取已存在的ID集合
4. 根据查询结果决定每条记录是执行插入还是更新操作

关键代码示例：

// 构建只查询主键的Wrapper
Map<String,Object> params = new HashMap<>();
params.put(Constants.WRAPPER, Wrappers.query()
    .select(keyProperty)
    .in(keyProperty, idList));

// 执行批量查询
List<主键类型> existIds = sqlSession.selectList(
    getSqlStatement(SqlMethod.SELECT_OBJS), 
    params);

性能对比

优化前后的性能差异主要体现在：

1. 数据库交互次数：从N次（N为记录数）降低到1次
2. 数据传输量：从完整实体数据减少到仅主键字段
3. 处理时间：整体处理时间大幅缩短，尤其在处理万级以上数据时效果显著

注意事项

1. 批量操作时应合理设置批次大小，避免单次操作数据量过大
2. 在高并发场景下，需考虑乐观锁等机制保证数据一致性
3. 对于特别大的数据集，建议采用分批次处理的方式

总结

通过对MyBatis-Plus saveOrUpdateBatch方法的优化，我们有效解决了批量操作时的性能瓶颈问题。这种优化思路不仅适用于此特定场景，也可以推广到其他需要批量判断数据是否存在的业务场景中。在实际开发中，我们应该根据具体业务需求和数据规模，选择最适合的优化方案。