mybatis-plus(insertBatchSomeColumn批次新增方式)

2023-03-31 06:00:54

前言

大家平時在做業務時肯定會遇到會向表中批次新增資料的方法，那麼這種方法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了。