Spring Data R2DBC中saveAll()如何实现单SQL批量插入多条数据

解决方案 #

可以实现单SQL批量插入，以下是两种常用可行方案：

方案1：启用Spring Data R2DBC原生批量支持（推荐，无需手动改业务代码）

该方案适配Spring Data R2DBC 3.0及以上版本，框架原生提供了批量处理开关，仅需要修改配置即可：

1. 在application配置文件（以yml为例）添加如下配置：

spring:
  data:
    r2dbc:
      repositories:
        # 开启Repository层批量处理能力
        enable-batch-processing: true
  r2dbc:
    # MySQL示例：追加rewriteBatchedStatements=true参数，驱动会自动合并批量插入语句
    url: r2dbc:mysql://localhost:3306/你的库名?rewriteBatchedStatements=true
    username: 用户名
    password: 密码

不同数据库对应驱动的批量重写参数不同：PostgreSQL可在连接URL追加reWriteBatchedInserts=true即可生效。

配置完成后直接调用原有ReactiveCrudRepository的saveAll方法即可，框架会自动将多条单条插入语句合并为你需要的多VALUES批量插入语句。如果需要返回Mono类型，直接调用saveAll(entities).collectList()即可得到Mono<List<S>>类型的返回结果。

方案2：使用DatabaseClient手动实现批量插入（兼容性更强，适配所有版本）

如果你的Spring Data R2DBC版本低于3.0，可以直接通过DatabaseClient手动构造批量插入逻辑，示例代码如下：

import org.springframework.r2dbc.core.DatabaseClient;
import reactor.core.publisher.Mono;
import java.util.List;
import java.util.Map;

// 假设你的实体类为TestEntity
public Mono<Integer> batchInsertTestEntities(List<TestEntity> entityList) {
    String insertSql = "INSERT INTO test(id, value) VALUES (:id, :value)";

    // 构造绑定参数列表
    List<Map<String, Object>> params = entityList.stream()
            .map(entity -> Map.of(
                    "id", entity.getId(),
                    "value", entity.getValue()
            ))
            .toList();

    return databaseClient.sql(insertSql)
            .bindList(params)
            .fetch()
            .rowsUpdated();
}

如果需要动态拼接不确定长度的VALUES列表，可以自行拼接SQL字符串，注意做好参数绑定避免SQL注入风险。

注意事项 #

• 批量插入的批次大小建议控制在500-1000条/批，避免单次SQL过长超过数据库最大允许的数据包大小。
• 如果需要返回插入后的实体数据，可在SQL末尾添加对应数据库的返回子句，比如PostgreSQL加RETURNING id, value，配合fetch().all()接收返回结果。

内容的提问来源于stack exchange，提问作者Adasiek