mybatis-plus(insertBatchSomeColumn批量添加方式)

Posted on by

前言

大傢平時在做業務時肯定會遇到會向表中批量添加數據的方法，那麼這種方法mybatis-plus給我們提供瞭嗎？

首先baseMapper中肯定沒有提供，如下：隻是添加單個實體的

但是IService貌似給我們提供瞭一個批量添加的方法：saveBatch(Collection<T> entityList)

那我們就拿這個方法來測試一下

    @Test
    public void testInsertMore(){
        //批量添加
        //INSERT INTO user ( id, name, age ) VALUES ( ?, ?, ? )
        List<User> list = new ArrayList<>();
        for (int i = 1; i <= 10; i++) {
            User user = new User();
            user.setName("ybc"+i);
            user.setAge(20+i);
            list.add(user);
        }
        boolean b = userService.saveBatch(list);
        System.out.println(b);
    }

插入成功後：以下是打印出的sql語句 

 可以發現：雖然saveBatch(Collection<T> entityList)這個方法我們代碼中隻是一行代碼，但是底層其實還是單條插入的。

但是這樣批量插入的速度有時其實是很慢的，那麼有沒有真正的批量插入方法呢？

其實mybatis-plus給我們預留瞭一個真正批量插入的擴展插件InsertBatchSomeColumn 

搭建工程

1）創建springboot項目

引入如下相關依賴

        <!--mybatis-plus啟動器-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.5.1</version>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-extension</artifactId>
            <version>3.5.1</version>
        </dependency>

2）編寫sql註入器

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
 
import java.util.List;
 
public class EasySqlInjector extends DefaultSqlInjector {
 
 
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        // 註意：此SQL註入器繼承瞭DefaultSqlInjector(默認註入器)，調用瞭DefaultSqlInjector的getMethodList方法，保留瞭mybatis-plus的自帶方法
        List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
        methodList.add(new InsertBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));
        return methodList;
    }
 
}

3）註入插件

@Configuration
public class MyBatisPlusConfig {
 
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor(){
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        //添加分頁插件
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
        //添加樂觀鎖插件
        interceptor.addInnerInterceptor(new OptimisticLockerInnerInterceptor());
        return interceptor;
    }
 
    @Bean
    public EasySqlInjector easySqlInjector() {
        return new EasySqlInjector();
    }
 
}

4）編寫自己的mapper繼承BaseMapper

public interface EasyBaseMapper<T> extends BaseMapper<T> {
    /**
     * 批量插入 僅適用於mysql
     *
     * @param entityList 實體列表
     * @return 影響行數
     */
    Integer insertBatchSomeColumn(Collection<T> entityList);
}

5）實體類的mapper繼承自己編寫的mapper

@Repository
public interface UserMapper<T> extends EasyBaseMapper<User> {
 
 
}

6）主啟動類

@SpringBootApplication
@MapperScan("com.atguigu.mybatisplus.mapper")
public class MybatisplusApplication {
 
    public static void main(String[] args) {
        SpringApplication.run(MybatisplusApplication.class, args);
    }
 
}

7）測試類測試

@SpringBootTest
public class MyBatisPlusServiceTest {
 
 
    @Autowired
    private UserMapper userMapper;
 
 
    @Test
    public void testInsertMore(){
        //批量添加
        //INSERT INTO user ( id, name, age ) VALUES ( ?, ?, ? )
        List<User> list = new ArrayList<>();
        for (int i = 1; i <= 10; i++) {
            User user = new User();
            user.setName("ybc"+i);
            user.setAge(20+i);
            list.add(user);
        }
        userMapper.insertBatchSomeColumn(list);
 
    }
 
}

測試結果：

==>  Preparing: INSERT INTO t_user (user_name,age,email,sex,is_deleted) VALUES (?,?,?,?,?) , (?,?,?,?,?) , (?,?,?,?,?) , (?,?,?,?,?) , (?,?,?,?,?) , (?,?,?,?,?) , (?,?,?,?,?) , (?,?,?,?,?) , (?,?,?,?,?) , (?,?,?,?,?)
==> Parameters: ybc1(String), 21(Integer), null, null, null, ybc2(String), 22(Integer), null, null, null, ybc3(String), 23(Integer), null, null, null, ybc4(String), 24(Integer), null, null, null, ybc5(String), 25(Integer), null, null, null, ybc6(String), 26(Integer), null, null, null, ybc7(String), 27(Integer), null, null, null, ybc8(String), 28(Integer), null, null, null, ybc9(String), 29(Integer), null, null, null, ybc10(String), 30(Integer), null, null, null
<==    Updates: 10
Closing non transactional SqlSession [org.apache.ibatis.session.defaults.DefaultSqlSession@496a31da]

可以看到隻執行瞭一條sql語句。

采坑點：希望自己註意自己的目錄結構（圖左）

我之前采用的是右邊的目錄結構導致一直報錯：

Caused by: org.springframework.beans.factory.UnsatisfiedDependencyException: Error creating bean with name 'userServiceImpl': Unsatisfied dependency expressed through field 'baseMapper'; nested exception is org.springframework.beans.factory.NoUniqueBeanDefinitionException: No qualifying bean of type 'com.atguigu.mybatisplus.mapper.UserMapper<com.atguigu.mybatisplus.pojo.User>' available: expected single matching bean but found 2: easyBaseMapper,userMapper

 一直不知道原因出在什麼地方，後來考慮到可能是mapper文件掃描時出現瞭問題，於是進行瞭修改，就OK瞭。

總結

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。

推薦閱讀：