1. 程式人生 > >給你的MyBatis-Plus裝上批量插入的翅膀

給你的MyBatis-Plus裝上批量插入的翅膀

努力和選擇,哪個更重要?關注微信公眾號【天開易想】這是一位懂網際網路研發和架構的戶外、籃球老鐵

前言

各位好,我是易哥(thinkYi).

大家有用過MyBatis-Plus(簡稱MP)的都知道它是一個MyBatis的增強工具,旨在MyBatis的基礎上只做增強不做改變,為簡化開發、提高效率而生.

特點

  • 無侵入:只做增強不做改變,引入它不會對現有工程產生影響,如絲般順滑
  • 損耗小:啟動即會自動注入基本 CURD,效能基本無損耗,直接面向物件操作
  • 強大的 CRUD 操作:內建通用 Mapper、通用 Service,僅僅通過少量配置即可實現單表大部分 CRUD 操作,更有強大的條件構造器,滿足各類使用需求
  • 支援 Lambda 形式呼叫:通過 Lambda 表示式,方便的編寫各類查詢條件,無需再擔心欄位寫錯
  • 支援主鍵自動生成:支援多達 4 種主鍵策略(內含分散式唯一 ID 生成器 - Sequence),可自由配置,完美解決主鍵問題
  • 支援 ActiveRecord 模式:支援 ActiveRecord 形式呼叫,實體類只需繼承 Model 類即可進行強大的 CRUD 操作
  • 支援自定義全域性通用操作:支援全域性通用方法注入( Write once, use anywhere )
  • 內建程式碼生成器:採用程式碼或者 Maven 外掛可快速生成 Mapper 、 Model 、 Service 、 Controller 層程式碼,支援模板引擎,更有超多自定義配置等您來使用
  • 內建分頁外掛:基於 MyBatis 物理分頁,開發者無需關心具體操作,配置好外掛之後,寫分頁等同於普通 List 查詢
  • 分頁外掛支援多種資料庫:支援 MySQL、MariaDB、Oracle、DB2、H2、HSQL、SQLite、Postgre、SQLServer 等多種資料庫
  • 內建效能分析外掛:可輸出 Sql 語句以及其執行時間,建議開發測試時啟用該功能,能快速揪出慢查詢
  • 內建全域性攔截外掛:提供全表 delete 、 update 操作智慧分析阻斷,也可自定義攔截規則,預防誤操作

正文

在實際專案開發中,我們常常有把資料批量儲存到資料庫的需求,大家或多或少的用mybatis-plus實現過吧?

元件依賴

首先我們要通過Maven引入mybatis-plus 開源元件,在pom.xml檔案加入下面的程式碼:

<!--mybatis-->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>3.4.0</version>
</dependency>
<!--mybatis plus extension,包含了mybatis plus core-->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-extension</artifactId>
    <version>3.4.0</version>
</dependency>

通過檢視原始碼發現API介面提供的批量插入的介面:

![](https://img2020.cnblogs.com/blog/958080/202009/958080-20200924100841439-698831042.png)

啟動服務後,用Postman除錯,後臺列印如下:

![](https://img2020.cnblogs.com/blog/958080/202009/958080-20200924100931046-736319120.png)

從圖上可以看出這個所謂的批量插入介面,其實就是一個for迴圈插入,Oh,My God!簡直就是噩夢一般.

![](https://img2020.cnblogs.com/blog/958080/202009/958080-20200924100945507-192888525.jpg)

難不成要手工實現,這樣,

INSERT INTO test (a, b, c) VALUES
<foreach collection="list" item="item" separator=",">
    (#{item.a}, #{item.b}, #{item.c})
</foreach>

我們閱讀mybatis-plus的原始碼,在com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn包中其實是有批量插入程式碼實現的,這裡我就不貼原始碼了,大家自行跟蹤一下.我們來手工擴充套件一下:

擴充套件程式碼

Talk is cheap, show me the code.先展示程式碼.再帶大家慢慢解釋為什麼這樣實現:

![](https://img2020.cnblogs.com/blog/958080/202009/958080-20200924101008467-402005313.png)

在MybatisPlusConfig檔案中注入該Bean,程式碼如下:

@Configuration
public class MybatisPlusConfig {

    /**
     * 分頁外掛
     *
     * @return PaginationInterceptor
     */
    @Bean
    public PaginationInterceptor paginationInterceptor() {
        return new PaginationInterceptor();
    }

    @Bean
    public EasySqlInjector easySqlInjector() {
        return new EasySqlInjector();
    }
}

還要擴充套件一下自帶的BaseMapper,程式碼如下:

import com.baomidou.mybatisplus.core.mapper.BaseMapper;

import java.util.Collection;

/**
 * 擴充套件通用 Mapper,支援資料批量插入
 *
 * @author 天開易想
 */
public interface EasyBaseMapper<T> extends BaseMapper<T> {

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

我們即可在業務類中,實現下面的引用了,程式碼如下:

/**
 * 定義業務mapper介面,繼承剛剛擴充套件的EasyBaseMapper
 *
 * @author 天開易想
 */
@Mapper
public interface TestMapper extends EasyBaseMapper<Test> {
}

/**
 * 業務實現類介面,即可引用
 *
 * @author 天開易想
 */
@Service
public class TestServiceImpl extends ServiceImpl<TestMapper, Test> implements TestService {

    @Override
    public Integer testBatch(Collection<Test> testList) {
        return baseMapper.insertBatchSomeColumn(testList);
    }

因為在BaseMapper中是不能拿來直接引用的,為什麼不能直接引用,據說是只支援MySql資料庫,所以作者沒有內建的原因吧!

到這裡,我們的批量插入功能就大功告成了,各位程式猿大佬,有沒有啥疑問的,我們來互相交流!

歡迎關注我的微信公眾號
![](https://img2020.cnblogs.com/blog/958080/202009/958080-20200923114611959-1094485321.