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

2025-05-14 02:39:08作者：贡沫苏Truman

引言

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

默认批量插入的问题分析

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

主要问题体现在：

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

解决方案设计思路

方案一：SqlInjector+Interceptor组合方案

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

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

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

方案二：动态SQL分组方案

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

属性分析：检查每个实体的各字段值是否为null或空
分组处理：将具有相同null/非null字段模式的实体分为一组
动态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并执行
    }
}