• 粗略的實(shí)驗(yàn)
• 最后

最近在壓測(cè)一批接口，發(fā)現(xiàn)接口處理速度慢的有點(diǎn)超出預(yù)期，感覺很奇怪，后面定位發(fā)現(xiàn)是數(shù)據(jù)庫(kù)批量保存這塊很慢。

這個(gè)項(xiàng)目用的是 mybatis-plus，批量保存直接用的是 mybatis-plus 提供的 saveBatch。

我點(diǎn)進(jìn)去看了下源碼，感覺有點(diǎn)不太對(duì)勁：

我繼續(xù)追蹤了下，從這個(gè)代碼來(lái)看，確實(shí)是 for 循環(huán)一條一條執(zhí)行了 sqlSession.insert，下面的 consumer 執(zhí)行的就是上面的 sqlSession.insert:

然后累計(jì)一定數(shù)量后，一批 flush。

從這點(diǎn)來(lái)看，這個(gè) saveBach 的性能肯定比直接一條一條 insert 快。

我直接進(jìn)行一個(gè)粗略的實(shí)驗(yàn)，簡(jiǎn)單創(chuàng)建了一張表來(lái)對(duì)比一波！

粗略的實(shí)驗(yàn)

1000條數(shù)據(jù)，一條一條插入

@Test
voidMybatisPlusSaveOne(){
SqlSessionsqlSession=sqlSessionFactory.openSession();
try{
StopWatchstopWatch=newStopWatch();
stopWatch.start("mybatisplussaveone");
for(inti=0;i1000;i++){
OpenTestopenTest=newOpenTest();
openTest.setA("a"+i);
openTest.setB("b"+i);
openTest.setC("c"+i);
openTest.setD("d"+i);
openTest.setE("e"+i);
openTest.setF("f"+i);
openTest.setG("g"+i);
openTest.setH("h"+i);
openTest.setI("i"+i);
openTest.setJ("j"+i);
openTest.setK("k"+i);
//一條一條插入
openTestService.save(openTest);
}
sqlSession.commit();
stopWatch.stop();
log.info("mybatis plus save one："+stopWatch.getTotalTimeMillis());
}finally{
sqlSession.close();
}
}

可以看到，執(zhí)行一批 1000 條數(shù)的批量保存，耗費(fèi)的時(shí)間是 121011 毫秒。

1000條數(shù)據(jù)用 mybatis-plus 自帶的 saveBatch 插入

@Test
voidMybatisPlusSaveBatch(){
SqlSessionsqlSession=sqlSessionFactory.openSession();
try{
ListopenTestList=newArrayList<>();
for(inti=0;i1000;i++){
OpenTestopenTest=newOpenTest();
openTest.setA("a"+i);
openTest.setB("b"+i);
openTest.setC("c"+i);
openTest.setD("d"+i);
openTest.setE("e"+i);
openTest.setF("f"+i);
openTest.setG("g"+i);
openTest.setH("h"+i);
openTest.setI("i"+i);
openTest.setJ("j"+i);
openTest.setK("k"+i);
openTestList.add(openTest);
}
StopWatchstopWatch=newStopWatch();
stopWatch.start("mybatisplussavebatch");
//批量插入
openTestService.saveBatch(openTestList);
sqlSession.commit();
stopWatch.stop();
log.info("mybatis plus save batch："+stopWatch.getTotalTimeMillis());
}finally{
sqlSession.close();
}
}

耗費(fèi)的時(shí)間是 59927 毫秒，比一條一條插入快了一倍，從這點(diǎn)來(lái)看，效率還是可以的。

然后常見的還有一種利用拼接 sql 方式來(lái)實(shí)現(xiàn)批量插入，我們也來(lái)對(duì)比試試看性能如何。

1000條數(shù)據(jù)用手動(dòng)拼接 sql 方式插入

搞個(gè)手動(dòng)拼接：

來(lái)跑跑下性能如何：

@Test
voidMapperSaveBatch(){
SqlSessionsqlSession=sqlSessionFactory.openSession();
try{
ListopenTestList=newArrayList<>();
for(inti=0;i1000;i++){
OpenTestopenTest=newOpenTest();
openTest.setA("a"+i);
openTest.setB("b"+i);
openTest.setC("c"+i);
openTest.setD("d"+i);
openTest.setE("e"+i);
openTest.setF("f"+i);
openTest.setG("g"+i);
openTest.setH("h"+i);
openTest.setI("i"+i);
openTest.setJ("j"+i);
openTest.setK("k"+i);
openTestList.add(openTest);
}
StopWatchstopWatch=newStopWatch();
stopWatch.start("mappersavebatch");
//手動(dòng)拼接批量插入
openTestMapper.saveBatch(openTestList);
sqlSession.commit();
stopWatch.stop();
log.info("mapper save batch："+stopWatch.getTotalTimeMillis());
}finally{
sqlSession.close();
}
}

耗時(shí)只有 2275 毫秒，性能比 mybatis-plus 自帶的 saveBatch 好了 26 倍！

這時(shí)，我又突然回想起以前直接用 JDBC 批量保存的接口，那都到這份上了，順帶也跑跑看！

1000條數(shù)據(jù)用 JDBC executeBatch 插入

@Test
voidJDBCSaveBatch()throwsSQLException{
SqlSessionsqlSession=sqlSessionFactory.openSession();
Connectionconnection=sqlSession.getConnection();
connection.setAutoCommit(false);

Stringsql="insertintoopen_test(a,b,c,d,e,f,g,h,i,j,k)values(?,?,?,?,?,?,?,?,?,?,?)";
PreparedStatementstatement=connection.prepareStatement(sql);
try{
for(inti=0;i1000;i++){
statement.setString(1,"a"+i);
statement.setString(2,"b"+i);
statement.setString(3,"c"+i);
statement.setString(4,"d"+i);
statement.setString(5,"e"+i);
statement.setString(6,"f"+i);
statement.setString(7,"g"+i);
statement.setString(8,"h"+i);
statement.setString(9,"i"+i);
statement.setString(10,"j"+i);
statement.setString(11,"k"+i);
statement.addBatch();
}
StopWatchstopWatch=newStopWatch();
stopWatch.start("JDBCsavebatch");
statement.executeBatch();
connection.commit();
stopWatch.stop();
log.info("JDBC save batch："+stopWatch.getTotalTimeMillis());
}finally{
statement.close();
sqlSession.close();
}
}

耗時(shí)是 55663 毫秒，所以 JDBC executeBatch 的性能跟 mybatis-plus 的 saveBatch 一樣（底層一樣）。

綜上所述，拼接 sql 的方式實(shí)現(xiàn)批量保存效率最佳。

但是我又不太甘心，總感覺應(yīng)該有什么別的法子，然后我就繼續(xù)跟著 mybatis-plus 的源碼 debug 了一下，跟到了 mysql 的驅(qū)動(dòng)，突然發(fā)現(xiàn)有個(gè) if 里面的條件有點(diǎn)顯眼：

就是這個(gè)叫 rewriteBatchedStatements 的玩意，從名字來(lái)看是要重寫批操作的 Statement，前面batchHasPlainStatements 已經(jīng)是 false，取反肯定是 true，所以只要這參數(shù)是 true 就會(huì)進(jìn)行一波操作。

我看了下默認(rèn)是 false。

同時(shí)我也上網(wǎng)查了下 rewriteBatchedStatements 參數(shù)，好家伙，好像有用！我直接將 jdbcurl 加上了這個(gè)參數(shù)：

然后繼續(xù)跑了下 mybatis-plus 自帶的 saveBatch，果然性能大大提高，跟拼接 SQL 差不多！

順帶我也跑了下 JDBC 的 executeBatch ，果然也提高了。

然后我繼續(xù) debug ，來(lái)探探 rewriteBatchedStatements 究竟是怎么 rewrite 的！

如果這個(gè)參數(shù)是 true，則會(huì)執(zhí)行下面的方法且直接返回：

看下 executeBatchedInserts 究竟干了什么：

看到上面我圈出來(lái)的代碼沒，好像已經(jīng)有點(diǎn)感覺了，繼續(xù)往下 debug。

果然！sql 語(yǔ)句被 rewrite了：

對(duì)插入而言，所謂的 rewrite 其實(shí)就是將一批插入拼接成 insert into xxx values (a),(b),(c)...這樣一條語(yǔ)句的形式然后執(zhí)行，這樣一來(lái)跟拼接 sql 的效果是一樣的。

那為什么默認(rèn)不給這個(gè)參數(shù)設(shè)置為 true 呢？

原來(lái)是這樣的：

1. 如果批量語(yǔ)句中的某些語(yǔ)句失敗，則默認(rèn)重寫會(huì)導(dǎo)致所有語(yǔ)句都失敗。
2. 批量語(yǔ)句的某些語(yǔ)句參數(shù)不一樣，則默認(rèn)重寫會(huì)使得查詢緩存未命中。

看起來(lái)影響不大，所以我給我的項(xiàng)目設(shè)置上了這個(gè)參數(shù)！

基于 Spring Boot + MyBatis Plus + Vue & Element 實(shí)現(xiàn)的后臺(tái)管理系統(tǒng) + 用戶小程序，支持 RBAC 動(dòng)態(tài)權(quán)限、多租戶、數(shù)據(jù)權(quán)限、工作流、三方登錄、支付、短信、商城等功能

• 項(xiàng)目地址：https://github.com/YunaiV/ruoyi-vue-pro
• 視頻教程：https://doc.iocoder.cn/video/

最后

稍微總結(jié)下我粗略的對(duì)比（雖然粗略，但實(shí)驗(yàn)結(jié)果符合原理層面的理解），如果你想更準(zhǔn)確地實(shí)驗(yàn)，可以使用JMH，并且測(cè)試更多組數(shù)（如 5000，10000等）的情況。

所以如果有使用 jdbc 的 Batch 性能方面的需求，要將 rewriteBatchedStatements 設(shè)置為 true，這樣能提高很多性能。

然后如果喜歡手動(dòng)拼接 sql 要注意一次拼接的數(shù)量，分批處理。