江苏龙网

  1. 主页 > 生活百科 > >

MyBatis 批量插入数据的 3 种方法！( 二 )

小知识

② 业务逻辑层实现import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;import com.example.demo.mapper.UserMapper;import com.example.demo.model.User;import com.example.demo.service.UserService;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.stereotype.Service;import java.util.List;@Servicepublic class UserServiceImpl extends ServiceImpl<UserMapper,User>implements UserService {@Autowiredprivate UserMapper userMapper;public boolean saveBatchCustom(List<User> list){return userMapper.saveBatchCustom(list);}}③ 数据持久层实现import com.baomidou.mybatisplus.core.mapper.BaseMapper;import com.example.demo.model.User;import org.Apache.ibatis.annotations.Mapper;import java.util.List;@Mapperpublic interface UserMapper extends BaseMapper<User>{boolean saveBatchCustom(List<User> list);}经过以上代码实现，我们就可以使用 MP 来实现数据的批量插入功能了，但本篇除了具体的实现代码之外，我们还要知道每种方法的执行效率，所以接下来我们来编写 MP 的测试代码 。
MP 性能测试import com.example.demo.model.User;import com.example.demo.service.impl.UserServiceImpl;import org.junit.jupiter.api.Test;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.boot.test.context.SpringBootTest;import java.util.ArrayList;import java.util.List;@SpringBootTestclass UserControllerTest {// 最大循环次数private static final int MAXCOUNT = 100000;@Autowiredprivate UserServiceImpl userService;/*** MP 批量插入*/@Testvoid saveBatch() {long stime = System.currentTimeMillis(); // 统计开始时间List<User> list = new ArrayList<>();for (int i = 0; i < MAXCOUNT; i++) {User user = new User();user.setName("test:" + i);user.setPassword("123456");list.add(user);}// MP 批量插入userService.saveBatch(list);long etime = System.currentTimeMillis(); // 统计结束时间System.out.println("执行时间：" + (etime - stime));}}以上程序的执行总共花费了 6088 毫秒，如下图所示：

MyBatis 批量插入数据的 3 种方法！

文章插图
 
从上述结果可知，使用 MP 的批量插入功能（插入数据 10W 条），它的性能比循环单次插入的性能提升了 14.5 倍 。
MP 源码分析从 MP 和循环单次插入的执行时间我们可以看出，使用 MP 并不是像有些朋友认为的那样，还是循环单次执行的，为了更清楚的说明此问题，我们查看了 MP 的源码 。
MP 的核心实现代码是 saveBatch 方法，此方法的源码如下：
MyBatis 批量插入数据的 3 种方法！

文章插图
 
我们继续跟进 saveBatch 的重载方法：
MyBatis 批量插入数据的 3 种方法！

文章插图
 
从上述源码可以看出，MP 是将要执行的数据分成 N 份，每份 1000 条，每满 1000 条就会执行一次批量插入，所以它的性能要比循环单次插入的性能高很多 。
那为什么要分批执行，而不是一次执行？别着急，当我们看了第 3 种实现方法之后我们就明白了 。
3.原生批量插入原生批量插入方法是依靠 MyBatis 中的 foreach 标签，将数据拼接成一条原生的 insert 语句一次性执行的，核心实现代码如下 。
① 业务逻辑层扩展在 UserServiceImpl 添加 saveBatchByNative 方法，实现代码如下：
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;import com.example.demo.mapper.UserMapper;import com.example.demo.model.User;import com.example.demo.service.UserService;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.stereotype.Service;import java.util.List;@Servicepublic class UserServiceImpl extends ServiceImpl<UserMapper, User>implements UserService {@Autowiredprivate UserMapper userMapper;public boolean saveBatchByNative(List<User> list) {return userMapper.saveBatchByNative(list);}}② 数据持久层扩展在 UserMapper 添加 saveBatchByNative 方法，实现代码如下：
import com.baomidou.mybatisplus.core.mapper.BaseMapper;import com.example.demo.model.User;import org.apache.ibatis.annotations.Mapper;import java.util.List;@Mapperpublic interface UserMapper extends BaseMapper<User> {boolean saveBatchByNative(List<User> list);}③ 添加 UserMapper.xml创建 UserMapper.xml 文件，使用 foreach 标签拼接 SQL，具体实现代码如下：
<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd"><mapper namespace="com.example.demo.mapper.UserMapper"><insert id="saveBatchByNative">INSERT INTO `USER`(`NAME`,`PASSWORD`) VALUES<foreach collection="list" separator="," item="item">(#{item.name},#{item.password})</foreach></insert></mapper>


推荐阅读


上一篇：JavaScript 框架大战已结束，赢家只有一个

下一篇：linux性能诊断神器pstat