MyBatis-Plus实现高效批量插入的深度解析与实践

引言

在数据库操作中，批量插入是提升性能的重要手段。MyBatis-Plus作为MyBatis的增强工具，提供了saveBatch方法来实现批量操作。然而，默认实现存在一些局限性，本文将深入探讨如何通过自定义扩展实现真正高效的批量插入。

默认批量插入的问题分析

MyBatis-Plus默认的saveBatch方法实际上是使用JDBC的批处理模式，本质上还是执行多条INSERT语句。这种方式虽然比单条插入效率高，但与MySQL支持的INSERT INTO ... VALUES (...), (...), ...这种真正的批量插入语句相比，性能仍有差距。

主要问题体现在：

1. 默认实现无法利用MySQL的多值插入语法
2. 对于有默认值的字段处理不够灵活
3. 无法动态根据实体属性值决定插入列

解决方案设计思路

方案一：SqlInjector+Interceptor组合方案

通过自定义SqlInjector添加新的批量插入方法，再配合Interceptor实现SQL重写：

1. 自定义SqlInjector：继承DefaultSqlInjector，添加新的批量插入方法
2. Interceptor实现：拦截执行过程，分析实体属性值动态生成SQL
3. 动态列处理：根据实体实际赋值的属性决定插入列

关键点在于通过实体属性值反向推导出需要插入的列，而不是使用所有表字段。

方案二：动态SQL分组方案

另一种实现思路是通过分析实体属性值的null情况，对批量数据进行分组：

1. 属性分析：检查每个实体的各字段值是否为null或空
2. 分组处理：将具有相同null/非null字段模式的实体分为一组
3. 动态SQL生成：为每组生成对应的INSERT语句

这种方式可以更精确地控制插入列，但实现复杂度较高。

技术实现细节

自定义SqlInjector实现

创建自定义的MysqlInsertBatchInjector：

public class MysqlInsertBatchInjector extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        methodList.add(new MysqlInsertBatch());
        return methodList;
    }
}

批量插入方法定义

定义新的批量插入方法：

public class MysqlInsertBatch extends AbstractMethod {
    @Override
    public MappedStatement injectMappedStatement(...) {
        // 创建MappedStatement但不生成完整SQL
        // 实际SQL由拦截器生成
    }
}

拦截器实现

关键拦截器实现逻辑：

public class MysqlInsertBatchInterceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        // 1. 获取参数和实体列表
        // 2. 分析第一个实体的非空属性
        // 3. 构建INSERT INTO table (col1,col2) VALUES (?,?),(?,?)形式SQL
        // 4. 重写SQL并执行
    }
}

性能优化建议

1. 批量大小控制：合理设置每批次插入的数据量，建议500-1000条/批
2. 事务管理：确保批量操作在同一个事务中执行
3. 连接池配置：适当增大连接池大小应对批量操作
4. 索引优化：临时禁用非必要索引提升插入速度

注意事项

1. 字段一致性：同一批次的所有实体必须具有相同的非空字段
2. 默认值处理：数据库默认值字段可以不设置
3. 主键生成：注意自增ID的处理方式
4. 异常处理：做好批量失败的回滚和重试机制

总结

通过自定义扩展MyBatis-Plus的批量插入功能，可以显著提升大数据量插入的性能。本文介绍的两种方案各有优缺点，开发者可以根据实际需求选择适合的方案。对于需要极致性能的场景，推荐使用SqlInjector+Interceptor的组合方案；对于字段变化较多的场景，则可以考虑动态SQL分组方案。

在实际应用中，还需要结合具体的业务场景和数据特点进行调整，才能达到最佳的批量插入效果。