快捷導(dǎo)航

MyBatis saveBatch 性能調(diào)優(yōu)的實(shí)現(xiàn)

更新時間：2023年07月21日 10:09:37 作者：Nate

本文主要介紹了MyBatis saveBatch 性能調(diào)優(yōu)的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

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

@Test
void MybatisPlusSaveOne() {
    SqlSession sqlSession = sqlSessionFactory.openSession();
    try {
        StopWatch stopWatch = new StopWatch();
        stopWatch.start("mybatis plus save one");
        for (int i = 0; i < 1000; i++) {
            OpenTest openTest = new OpenTest();
            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)的時間是 121011 毫秒。

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

@Test
void MybatisPlusSaveBatch() {
    SqlSession sqlSession = sqlSessionFactory.openSession();
    try {
        List<OpenTest> openTestList = new ArrayList<>();
        for (int i = 0; i < 1000; i++) {
            OpenTest openTest = new OpenTest();
            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);
        }
        StopWatch stopWatch = new StopWatch();
        stopWatch.start("mybatis plus save batch");
        //批量插入
        openTestService.saveBatch(openTestList);
        sqlSession.commit();
        stopWatch.stop();
        log.info("mybatis plus save batch：" + stopWatch.getTotalTimeMillis());
    } finally {
        sqlSession.close();
    }
}

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

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

3、1000 條數(shù)據(jù)用手動拼接 SQL 方式插入，搞個手動拼接：

來跑跑下性能如何：

@Test
void MapperSaveBatch() {
    SqlSession sqlSession = sqlSessionFactory.openSession();
    try {
        List<OpenTest> openTestList = new ArrayList<>();
        for (int i = 0; i < 1000; i++) {
            OpenTest openTest = new OpenTest();
            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);
        }
        StopWatch stopWatch = new StopWatch();
        stopWatch.start("mapper save batch");
        //手動拼接批量插入
        openTestMapper.saveBatch(openTestList);
        sqlSession.commit();
        stopWatch.stop();
        log.info("mapper save batch：" + stopWatch.getTotalTimeMillis());
    } finally {
        sqlSession.close();
    }
}

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

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

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

@Test
void JDBCSaveBatch() throws SQLException {
    SqlSession sqlSession = sqlSessionFactory.openSession();
    Connection connection = sqlSession.getConnection();
    connection.setAutoCommit(false);
    String sql = "insert into open_test(a,b,c,d,e,f,g,h,i,j,k) values(?,?,?,?,?,?,?,?,?,?,?)";
    PreparedStatement statement = connection.prepareStatement(sql);
    try {
        for (int i = 0; i < 1000; 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();
        }
        StopWatch stopWatch = new StopWatch();
        stopWatch.start("JDBC save batch");
        statement.executeBatch();
        connection.commit();
        stopWatch.stop();
        log.info("JDBC save batch：" + stopWatch.getTotalTimeMillis());
    } finally {
        statement.close();
        sqlSession.close();
    }
}

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

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

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

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

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

同時我也上網(wǎng)查了下 rewriteBatchedStatements 參數(shù)，好家伙，好像有用！

直接將 jdbcurl 加上了這個參數(shù)：

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

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

然后我繼續(xù) debug ，來探探 rewriteBatchedStatements 究竟是怎么 rewrite 的！如果這個參數(shù)是 true，則會執(zhí)行下面的方法且直接返回：

看下 executeBatchedInserts 究竟干了什么：

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

果然！SQL 語句被 rewrite了：

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

那為什么默認(rèn)不給這個參數(shù)設(shè)置為 true 呢？主要有以下兩點(diǎn)：

如果批量語句中的某些語句失敗，則默認(rèn)重寫會導(dǎo)致所有語句都失敗。

批量語句的某些語句參數(shù)不一樣，則默認(rèn)重寫會使得查詢緩存未命中。

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

最后

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

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

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

到此這篇關(guān)于MyBatis saveBatch 性能調(diào)優(yōu)的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)MyBatis saveBatch 性能調(diào)優(yōu)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: