牡丹江0453免费信息网站华企网络

优化MyBatis-Plus批量插入策略

一、用Mybatis-plus中的saveBatch方法

MyBatis-Plus 的 Service 层提供了 saveBatch 方法，可以方便地实现批量插入操作。但查阅源码发现，saveBatch的底层还是逐条插入的，这就会导致当数据量大的时候程序运行效率变慢

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.stereotype.Service;
import java.util.List;// 假设实体类为 User
class User {private Long id;private String name;// 构造函数、getter 和 setter 方法public User() {}public User(Long id, String name) {this.id = id;this.name = name;}public Long getId() {return id;}public void setId(Long id) {this.id = id;}public String getName() {return name;}public void setName(String name) {this.name = name;}
}// 假设 UserMapper 继承自 BaseMapper
interface UserMapper extends com.baomidou.mybatisplus.core.mapper.BaseMapper<User> {}// UserService 继承自 ServiceImpl
@Service
class UserService extends ServiceImpl<UserMapper, User> {public boolean saveUserBatch(List<User> userList) {return this.saveBatch(userList);}
}// 调用示例
public class Main {public static void main(String[] args) {UserService userService = new UserService();java.util.List<User> userList = new java.util.ArrayList<>();userList.add(new User(1L, "Alice"));userList.add(new User(2L, "Bob"));boolean result = userService.saveUserBatch(userList);if (result) {System.out.println("批量插入成功");} else {System.out.println("批量插入失败");}}
}

二、InsertBatchSomeColumn插件

它继承了Abstractmethod类,在 MyBatis-Plus 中，InsertBatchSomeColumn 是一种用于批量插入部分列数据的插入策略，它可以帮助我们在批量插入数据时只插入部分需要的列，而不是插入所有列。

1.使用前配置

首先，确保你的项目中已经添加了 MyBatis-Plus 和 MySQL 驱动的依赖。如果你使用的是 Maven 项目，可以在 pom.xml 中添加以下依赖：

<dependencies><!-- MyBatis-Plus --><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-boot-starter</artifactId><version>3.5.3.1</version></dependency><!-- MySQL 驱动 --><dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId><version>8.0.32</version></dependency>
</dependencies>

2.代码示例

1.配置类 MybatisPlusConfig

• 创建一个自定义的 DefaultSqlInjector，并在 getMethodList 方法中添加 InsertBatchSomeColumn 方法，将其注入到 MyBatis-Plus 中。

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;import java.util.List;@Configuration
public class MybatisPlusConfig {@Beanpublic DefaultSqlInjector sqlInjector() {return new DefaultSqlInjector() {@Overridepublic List<AbstractMethod> getMethodList(Class<?> mapperClass) {List<AbstractMethod> methodList = super.getMethodList(mapperClass);methodList.add(new InsertBatchSomeColumn());return methodList;}};}
}    

2).实体类 User

使用 @TableName 注解指定该实体类对应的数据库表名。
为每个属性提供了对应的 getter 和 setter 方法，这里使用了 Lombok 的 @Data 注解来自动生成这些方法

ximport com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;@Data
@TableName("user")
public class User {private Long id;private String name;@TableField("age")private Integer age;private String email;
}    

3).Mapper 接口 UserMapper

继承自 BaseMapper，继承了 MyBatis-Plus 提供的基本 CRUD 方法。
使用 @InsertProvider 注解指定插入方法的 SQL 提供者为 InsertBatchSomeColumn，并调用其 sql 方法生成 SQL 语句。
定义了 insertBatchSomeColumn 方法，用于批量插入部分列数据。

import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
import com.baomidou.mybatisplus.extension.plugins.inner.SqlParserInner;
import org.apache.ibatis.annotations.InsertProvider;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.builder.annotation.ProviderMethodResolver;import java.util.List;public interface UserMapper extends BaseMapper<User>, ProviderMethodResolver {@InsertProvider(type = InsertBatchSomeColumn.class, method = "sql")int insertBatchSomeColumn(@Param("list") List<User> list);
}    

4).测试类 InsertBatchTest

实现 CommandLineRunner 接口，在 run 方法中进行测试。
创建多个 User 对象并添加到列表中。
调用 UserMapper 的 insertBatchSomeColumn 方法进行批量插入，并打印插入的行数。

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;import java.util.ArrayList;
import java.util.List;@SpringBootApplication
public class InsertBatchTest implements CommandLineRunner {@Autowiredprivate UserMapper userMapper;public static void main(String[] args) {SpringApplication.run(InsertBatchTest.class, args);}@Overridepublic void run(String... args) throws Exception {List<User> userList = new ArrayList<>();User user1 = new User();user1.setName("Alice");user1.setAge(20);userList.add(user1);User user2 = new User();user2.setName("Bob");user2.setAge(22);userList.add(user2);int rows = userMapper.insertBatchSomeColumn(userList);System.out.println("插入行数: " + rows);}
}    

    user2.setName("Bob");user2.setAge(22);userList.add(user2);int rows = userMapper.insertBatchSomeColumn(userList);System.out.println("插入行数: " + rows);
}

}