【性能】mysql使用rewriteBatchStatements

type

Post

status

Published

date

Feb 22, 2024

slug

tech-mysql

summary

rewriteBatchStatements 是 MySQL 的 JDBC 驱动程序（Connector/J）中的一个参数，用于控制批量插入的行为。当设置为 true 时，它可以改写批量插入语句，以提高性能和效率。以下是对 rewriteBatchStatements 的简要介绍： 1. 目的： ◦ rewriteBatchStatements 参数的主要目的是优化批量插入操作的性能。 ◦ 通常情况下，批量插入操作会将多个单条插入语句合并为一个批量插入语句，以减少与数据库的通信次数，提高插入效率。 2. 工作原理： ◦ 当将 rewriteBatchStatements 参数设置为 true 时，MySQL 的 JDBC 驱动程序会尝试将多个单条插入语句合并为一个批量插入语句。 ◦ 这种合并操作可以减少网络开销和数据库操作次数，从而提高批量插入操作的效率。 3. 注意事项： ◦ 使用 rewriteBatchStatements 参数可能会带来一些潜在的风险和限制，因此在设置之前需要仔细考虑。 ◦ 有些情况下，数据库的配置或数据表的结构可能不适合使用该参数，可能会导致性能下降或错误。 4. 使用示例： ◦ 在 JDBC 连接字符串中设置 rewriteBatchStatements=true 来启用该参数，例如： <TEXT>jdbc:mysql://localhost:3306/mydatabase?rewriteBatchStatements=true 总的来说，rewriteBatchStatements 参数是用于优化 MySQL 数据库中批量插入操作的一个选项，可以在合适的情况下提高插入操作的性能。但在使用时需要注意潜在的风险，并根据实际情况进行调整和测试。

参数简介

rewriteBatchStatements 是 MySQL 的 JDBC 驱动程序（Connector/J）中的一个参数，用于控制批量插入的行为。当设置为 true 时，它可以改写批量插入语句，以提高性能和效率。以下是对 rewriteBatchStatements 的简要介绍：

目的：

rewriteBatchStatements 参数的主要目的是优化批量插入操作的性能。

通常情况下，批量插入操作会将多个单条插入语句合并为一个批量插入语句，以减少与数据库的通信次数，提高插入效率。

工作原理：

当将 rewriteBatchStatements 参数设置为 true 时，MySQL 的 JDBC 驱动程序会尝试将多个单条插入语句合并为一个批量插入语句。

这种合并操作可以减少网络开销和数据库操作次数，从而提高批量插入操作的效率。

注意事项：

使用 rewriteBatchStatements 参数可能会带来一些潜在的风险和限制，因此在设置之前需要仔细考虑。

有些情况下，数据库的配置或数据表的结构可能不适合使用该参数，可能会导致性能下降或错误。

使用示例：

在 JDBC 连接字符串中设置 rewriteBatchStatements=true 来启用该参数，例如：


<TEXT>
jdbc:mysql://localhost:3306/mydatabase?rewriteBatchStatements=true

总的来说，rewriteBatchStatements 参数是用于优化 MySQL 数据库中批量插入操作的一个选项，可以在合适的情况下提高插入操作的性能。但在使用时需要注意潜在的风险，并根据实际情况进行调整和测试。

转载文章

最近在压测一批接口，发现少数接口处理速度慢的有点超出预期，感觉很奇怪，后面定位发现是数据库批量保存这块很慢。

这个项目用的是 mybatis-plus，批量保存直接用的是 mybatis-plus 提供的 saveBatch。

点进去看了下源码，感觉有点不太对劲：

这里继续追踪了下，从这个代码来看，确实是 for 循环一条一条执行了 sqlSession.insert，下面的 consumer 执行的就是上面的 sqlSession.insert：

然后累计一定数量后，一批 flush。

从这点来看，这个 saveBach 的性能肯定比直接一条一条 insert 快。

我们直接进行一个粗略的实验，简单创建了一张表来对比一波！

粗略的实验

1000条数据，一条一条插入


    @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();
        }
    }

可以看到，执行一批 1000 条数的批量保存，耗费的时间是 121011 毫秒。

1000条数据用 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();
        }
    }

耗费的时间是 59927 毫秒，比一条一条插入快了一倍，从这点来看，效率还是可以的。

然后常见的还有一种利用拼接 sql 方式来实现批量插入，我们也来对比试试看性能如何。

1000条数据用手动拼接 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 批量保存的接口，那都到这份上了，顺带也跑跑看！

1000条数据用 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 的方式实现批量保存效率最佳。

但是我又不太甘心，总感觉应该有什么别的法子，然后我就继续跟着 mybatis-plus 的源码 debug 了一下，跟到了 mysql 的驱动，突然发现有个 if 里面的条件有点显眼：

就是这个叫 rewriteBatchedStatements 的玩意，从名字来看是要重写批操作的 Statement，前面batchHasPlainStatements 已经是 false，取反肯定是 true，所以只要这参数是 true 就会进行一波操作。

我看了下默认是 false。

同时我们也上网稍微查了下 rewriteBatchedStatements 参数，有用！

这里就将 jdbcurl 加上了这个参数：

然后继续跑了下 mybatis-plus 自带的 saveBatch，果然性能大大提高，跟拼接 SQL 差不多！

顺带我也跑了下 JDBC 的 executeBatch ，果然也提高了。

然后我继续 debug ，来探探 rewriteBatchedStatements 究竟是怎么 rewrite 的！

如果这个参数是 true，则会执行下面的方法且直接返回：

看下 executeBatchedInserts 究竟干了什么：

看到上面我圈出来的代码没，好像已经有点感觉了，继续往下 debug。

果然！sql 语句被 rewrite了：

对插入而言，所谓的 rewrite 其实就是将一批插入拼接成 insert into xxx values (a),(b),(c)...这样一条语句的形式然后执行，这样一来跟拼接 sql 的效果是一样的。

那为什么默认不给这个参数设置为 true 呢？

我简单问了下 ChatGPT：

如果批量语句中的某些语句失败，则默认重写会导致所有语句都失败。

批量语句的某些语句参数不一样，则默认重写会使得查询缓存未命中。

最后

稍微总结下我粗略的对比（虽然粗略，但实验结果符合原理层面的理解），如果你想更准确地实验，可以使用JMH，并且测试更多组数（如 5000，10000等）的情况。

批量保存方式	数据量（条）	耗时（ms）
单条循环插入	1000	121011
mybatis-plus saveBatch	1000	59927
mybatis-plus saveBatch(添加rewtire参数)	1000	2589
手动拼接sql	1000	2275
jdbc executeBatch	1000	55663
jdbc executeBatch(添加rewtire参数)	1000	324

所以如果有使用 jdbc 的 Batch 性能方面的需求，要将 rewriteBatchedStatements 设置为 true，这样能提高很多性能。

然后如果喜欢手动拼接 sql 要注意一次拼接的数量，分批处理。

好了，以上就是今天这篇文章的内容分享，希望能对大家有所帮助，我们下篇见。

转载：

从50s优化到0.3s，昨晚班没白加！

The following article is from yes的练级攻略 Author 是Yes呀

https://mp.weixin.qq.com/s/VTRWbX2preK6aR_wUs8I9Q