从源码角度解析MyBatis-Plus批量插入功能的奥秘

2023-09-24 17:23:53

批量插入是MyBatis-Plus的一大特色功能，它可以大幅提升数据入库速度，对于大数据量的插入场景尤其适用。本文将从源码角度对MyBatis-Plus的批量插入功能进行解析，帮助读者深入了解其工作原理和实现细节。

MyBatis-Plus批量插入功能的实现原理

MyBatis-Plus的批量插入功能主要由接口类和ServiceImpl实现类两个部分组成。

接口类

接口类提供了一个批量插入方法，该方法的定义如下：

public interface IService<T> {

    int insertBatchSomeColumn(List<T> entityList);

}

ServiceImpl实现类

ServiceImpl实现类中对insertBatchSomeColumn方法进行了实现，其代码如下：

@Override
public int insertBatchSomeColumn(List<T> entityList) {
    if (entityList == null || entityList.isEmpty()) {
        throw new IllegalArgumentException("entityList cannot be null or empty");
    }
    Class<?> entityClass = entityList.get(0).getClass();
    TableInfo tableInfo = TableInfo.getTableInfo(entityClass);
    if (!tableInfo.isWithInsertBatchSomeColumn()) {
        throw new RuntimeException(" can't execute. because not with insert batch some column function.");
    }
    return sqlInjector.insertBatchSomeColumn(entityList);
}

在insertBatchSomeColumn方法中，首先对传入的entityList进行非空判断，如果entityList为空，则抛出异常。然后，获取entityList中第一个实体的类类型，并根据该类类型获取相应的TableInfo对象。

如果TableInfo对象的isWithInsertBatchSomeColumn属性为true，则说明该实体类支持批量插入，否则抛出异常。最后，调用sqlInjector的insertBatchSomeColumn方法执行批量插入操作。

MyBatis-Plus批量插入功能的使用方法

要使用MyBatis-Plus的批量插入功能，您需要遵循以下步骤：

在您的实体类中添加@TableName注解，该注解用于指定实体类对应的数据库表名。
在您的Service接口中添加insertBatchSomeColumn方法。
在您的ServiceImpl类中实现insertBatchSomeColumn方法，并调用sqlInjector的insertBatchSomeColumn方法执行批量插入操作。