前言

MyBatis-Plus 是基于 MyBatis 進(jìn)行封裝的一套優(yōu)秀的持久層框架，它提供了豐富的便捷操作方法和強(qiáng)大的代碼生成器，大大簡(jiǎn)化了 MyBatis 的使用。在 MyBatis-Plus 中，我們可以使用 insertBatchSomeColumn 方法來(lái)實(shí)現(xiàn)批量新增指定字段的操作。

mybatis-plus的 IService接口 默認(rèn)提供 saveBatch批量插入，也是唯一一個(gè)默認(rèn)批量插入，在數(shù)據(jù)量不是很大的情況下可以直接使用，但這種是一條一條執(zhí)行的效率上會(huì)有一定的瓶頸，在這里先看下saveBatch的執(zhí)行情況

可以看到sql語(yǔ)句是一條一條執(zhí)行的,插入多少條數(shù)據(jù)就相當(dāng)于執(zhí)行了多少次的插入sql, 點(diǎn)進(jìn)saveBatch方法，看看內(nèi)部是怎么實(shí)現(xiàn)的

 注意看，方法有一個(gè)事務(wù)注解，說(shuō)明插入整批數(shù)據(jù)會(huì)作為一個(gè)事務(wù)進(jìn)行,

默認(rèn)1000條一次，怪不得執(zhí)行時(shí)每隔1000條會(huì)處于一個(gè)準(zhǔn)備執(zhí)行狀態(tài)，等待幾秒后才會(huì)往下執(zhí)行（這里等待的時(shí)間就是1000個(gè)單條的insert語(yǔ)句執(zhí)行的時(shí)間）

再往下點(diǎn)是一個(gè)saveBatch接口，參數(shù)分別是插入的對(duì)象集合、插入批次數(shù)量也就是默認(rèn)的1000

這里也有一個(gè)事務(wù)的注解，這是因?yàn)?code>saveBatch是一個(gè)重載方法，插入的時(shí)候也可以指定插入批次數(shù)量調(diào)用

繼續(xù)往下進(jìn)入executeBatch

這里就是真正執(zhí)行的方法了，idxLimit會(huì)對(duì)比DEFAULT_BATCH_SIZE和集合長(zhǎng)度兩個(gè)數(shù)中的最小數(shù)，作為批量大小，也就是說(shuō)當(dāng)集合長(zhǎng)度不夠1000，那么執(zhí)行的時(shí)候批量大小就是集合的長(zhǎng)度，就執(zhí)行一次。

for循環(huán)中的consumer：對(duì)應(yīng)的類型是一個(gè)函數(shù)式接口，代表一個(gè)接受兩個(gè)輸入?yún)?shù)且不返回任何內(nèi)容的操作符。意思是給定兩個(gè)參數(shù)sqlSession、循環(huán)中當(dāng)前element對(duì)象，執(zhí)行一次傳遞過(guò)來(lái)的consumer匿名函數(shù)。也就是上邊源碼里的插入匿名函數(shù)。

當(dāng)i == indLimit時(shí)：執(zhí)行一次預(yù)插入，并重新計(jì)算idxLimit的值

if中idxLimit計(jì)算規(guī)則：當(dāng)前idxLimit加batchSize(默認(rèn)1000) 和 集合長(zhǎng)度 取最小值，計(jì)算出來(lái)的結(jié)果肯定不會(huì)超過(guò)集合的長(zhǎng)度，最后的批次時(shí)idxLimit等于集合的長(zhǎng)度，將這個(gè)值作為下一次執(zhí)行預(yù)插入的時(shí)間點(diǎn)。

sqlSession.flushStatements()：當(dāng)有處于事務(wù)中的時(shí)候，起到一種預(yù)插入的作用，執(zhí)行了這行代碼之后,要插入的數(shù)據(jù)會(huì)鎖定數(shù)據(jù)庫(kù)的一行記錄,并把數(shù)據(jù)庫(kù)默認(rèn)返回的主鍵賦值給插入的對(duì)象,這樣就可以把該對(duì)象的主鍵賦值給其他需要的對(duì)象中去了，這里不是事務(wù)提交啊。

最后方法執(zhí)行完后@Transactional注解會(huì)默認(rèn)提交事務(wù)，如果調(diào)用的方法上還有@Transactional注解，默認(rèn)的事務(wù)傳播類型是Propagation.REQUIRED，不會(huì)新開(kāi)啟事務(wù)，如果沒(méi)有@Transactional注解才會(huì)新開(kāi)起事務(wù)

下面我們使用真正的批量新增方法insertBatchSomeColumn,看看兩者的區(qū)別

1. 什么是批量新增指定字段

批量新增指定字段是指在一次 SQL 語(yǔ)句中執(zhí)行多條 INSERT 語(yǔ)句，但是只插入指定的字段。批量新增指定字段可以提高數(shù)據(jù)操作效率，減少數(shù)據(jù)庫(kù)與應(yīng)用程序之間的網(wǎng)絡(luò)傳輸次數(shù)，減輕數(shù)據(jù)庫(kù)服務(wù)器的壓力，提高系統(tǒng)的并發(fā)性能。

2. MyBatis-Plus 批量新增指定字段的方法的使用

mybatis-plus提供了InsertBatchSomeColumn批量insert方法。通過(guò)SQL 自動(dòng)注入器接口 ISqlInjector注入通用方法 SQL 語(yǔ)句 然后繼承 BaseMapper 添加自定義方法，全局配置 sqlInjector 注入 MP 會(huì)自動(dòng)將類所有方法注入到 mybatis 容器中。我們需要通過(guò)這種方式注入下。

/**
 * 自定義Sql注入
 *
 */
public class EasySqlInjector extends DefaultSqlInjector {
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        // 注意：此SQL注入器繼承了DefaultSqlInjector(默認(rèn)注入器)，調(diào)用了DefaultSqlInjector的getMethodList方法，保留了mybatis-plus的自帶方法
        List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
        methodList.add(new InsertBatchSomeColumn(i -> i.getFieldFill() != FieldFill.UPDATE));
        return methodList;
    }
}

 自定義EasyBaseMapper

/**
 * @author 武天
 * @date 2023/3/22 18:19
 */
public interface EasyBaseMapper<T> extends BaseMapper<T> {
    /**
     * 批量插入 僅適用于mysql
     *
     * @param entityList 實(shí)體列表
     * @return 影響行數(shù)
     */
    Integer insertBatchSomeColumn(Collection<T> entityList);
}

 mapper繼承的這里改為繼承剛書(shū)寫(xiě)的自定義mapper

/**
 * 
 * 
 * @author wutian
 * @email ${email}
 * @date 2022-10-03 22:12:06
 */
@Mapper
public interface BreakDownCluesDao extends EasyBaseMapper<ExcelBreakdownClues> {
}

調(diào)用測(cè)試

可以看到明顯的區(qū)別,sql只執(zhí)行了一條,后面都是入?yún)⒍禾?hào)相隔的形式,耗時(shí)也有所優(yōu)化,這是插入數(shù)量不是很多的情況,如果是上萬(wàn)條或者上千萬(wàn)條數(shù)據(jù),這種形式的優(yōu)越性就體現(xiàn)出來(lái)了 

3. 批量新增指定字段的注意事項(xiàng)

在使用 MyBatis-Plus 進(jìn)行批量新增指定字段時(shí)，需要注意以下幾點(diǎn)：

• 每次新增的數(shù)據(jù)量不要過(guò)大，建議每批次新增的數(shù)據(jù)量控制在 1000 條以內(nèi)。
• 要新增的指定字段不能為 null，需要手動(dòng)設(shè)置默認(rèn)值。
• 如果要新增的指定字段在實(shí)體類中有對(duì)應(yīng)的字段值，會(huì)被忽略。

總結(jié)

到此這篇關(guān)于mybatis-plus用insertBatchSomeColumn方法批量新增指定字段的文章就介紹到這了,更多相關(guān)mybatis-plus批量新增insertBatchSomeColumn內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論