如何避免JPA调用persist直接插入数据库及提升批量插入效率

你遇到的persist每次都触发INSERT的问题，和FlushMode设置无关，核心原因是你的Customer表主键用了auto_increment，对应JPA默认的@GeneratedValue(strategy = GenerationType.IDENTITY)主键生成策略：这种策略下，JPA必须执行INSERT语句才能拿到数据库生成的主键值，才能将实体标记为持久化状态，所以就算你设置了COMMIT的FlushMode，每次persist也会立即触发INSERT。

1. 开启基础批量配置 #

如果你用的是主流JPA实现Hibernate，首先需要在persistence.xml中添加批量相关配置：

<!-- 配置批处理大小，建议设置为50-100，可根据实际场景调整 -->
<property name="hibernate.jdbc.batch_size" value="50"/>
<!-- 开启批量插入时的参数重排序，确保同类型实体插入可以合并为批量请求 -->
<property name="hibernate.order_inserts" value="true"/>
<!-- MySQL驱动专属配置，开启后会把多条单条INSERT合并为批量INSERT语句，必须配置 -->
<property name="hibernate.connection.url" value="jdbc:mysql://localhost:3306/你的数据库名?rewriteBatchedStatements=true"/>

注意：上述配置对IDENTITY主键策略无效，因为IDENTITY主键需要每次INSERT获取返回的主键值，根本攒不住批量SQL。

2. 替换主键生成策略（性能提升最明显） #

如果允许修改表结构和实体配置，建议把主键生成策略替换为以下任意一种，即可完美适配批量插入：

• MySQL 8.0及以上版本可以用SEQUENCE序列策略
• 业务层自主生成主键（比如雪花算法、UUID）
• 用TABLE表策略模拟序列生成主键

3. 使用无状态会话兼容IDENTITY主键 #

如果不能修改主键策略，可以用Hibernate提供的无状态会话StatelessSession实现批量插入，它跳过了一级缓存和持久化上下文的主键校验逻辑，可以正常攒批量SQL：

// 从EntityManager拿到原生Hibernate会话工厂
StatelessSession statelessSession = ((Session) entityManager.getDelegate()).getSessionFactory().openStatelessSession();
Transaction tx = statelessSession.beginTransaction();
for (int i = 0; i < 10000; i++) {
    Customer customer = new Customer();
    customer.setBirth(new Date());
    customer.setCreatedTime(new Date());
    customer.setUpdateTime(new Date());
    statelessSession.insert(customer);
}
tx.commit();
statelessSession.close();

4. 分段提交避免内存溢出 #

如果插入数据量超过10万，建议每插入固定数量数据就手动flush、清空持久化上下文，避免一级缓存占用过多内存：

for (int i = 0; i < 10000; i++) {
    Customer customer = new Customer();
    // 实体属性赋值
    entityManager.persist(customer);
    // 每1000条提交一次
    if (i % 1000 == 0) {
        entityManager.flush();
        entityManager.clear();
    }
}

5. 极限性能方案：直接JDBC批量操作 #

如果追求最高插入性能，直接通过EntityManager拿到JDBC连接，用PreparedStatement做原生批量插入，性能比JPA批量插入高30%以上，适合超大数据量的离线导入场景。

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