如何高效更新/保存实体列表？Spring Data JPA与JPA/Hibernate方案

一、Spring Data JPA 实现方案 #

这是贴合Spring生态的规范实现，核心思路是先批量查询已存在实体，再合并数据后批量保存，避免N+1查询，保证效率。

步骤说明 #

• 给实体类标记自然ID字段，方便基于该字段查询
• 从待保存列表中提取所有自然ID，批量查询数据库中已存在的实体并转成Map，用于快速匹配
• 遍历待保存实体：已存在的则更新字段，不存在的直接保留为新增
• 最后批量保存处理后的实体集合

代码示例 #

实体类（以商品为例）

@Entity
public class Product {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id; // 数据库主键

    @NaturalId(mutable = true) // mutable=true表示自然ID可更新，按需调整
    private String sku; // 业务自然ID，比如商品SKU

    private String name;
    private BigDecimal price;

    // 构造器、getter、setter省略
}

Repository接口

public interface ProductRepository extends JpaRepository<Product, Long> {
    // 根据自然ID列表批量查询已存在实体
    List<Product> findBySkuIn(List<String> skuList);
}

业务层实现

@Service
@Transactional
public class ProductService {
    private final ProductRepository productRepository;

    public ProductService(ProductRepository productRepository) {
        this.productRepository = productRepository;
    }

    public void batchUpsert(List<Product> products) {
        // 提取所有待处理的自然ID
        List<String> skuList = products.stream()
                .map(Product::getSku)
                .collect(Collectors.toList());

        // 批量查询已存在实体，转成Map快速查找
        Map<String, Product> existingProducts = productRepository.findBySkuIn(skuList)
                .stream()
                .collect(Collectors.toMap(Product::getSku, p -> p));

        // 合并数据，生成待保存集合
        List<Product> toSave = new ArrayList<>();
        for (Product product : products) {
            Product existing = existingProducts.get(product.getSku());
            if (existing != null) {
                // 更新已存在实体的业务字段
                existing.setName(product.getName());
                existing.setPrice(product.getPrice());
                toSave.add(existing);
            } else {
                // 新增实体直接加入列表
                toSave.add(product);
            }
        }

        // 批量保存
        productRepository.saveAll(toSave);
    }
}

优化点 #

开启JDBC批量操作提升效率，在application.properties中添加：

spring.jpa.properties.hibernate.jdbc.batch_size=50
spring.jpa.properties.hibernate.order_inserts=true
spring.jpa.properties.hibernate.order_updates=true

二、原生JPA/Hibernate 实现方案 #

如果需要脱离Spring Data JPA封装，直接用JPA EntityManager或Hibernate Session实现，核心逻辑一致，只是手动管理实体状态。

基于EntityManager的实现 #

@Service
@Transactional
public class ProductService {
    private final EntityManager entityManager;

    public ProductService(EntityManager entityManager) {
        this.entityManager = entityManager;
    }

    public void batchUpsert(List<Product> products) {
        // 提取自然ID列表
        List<String> skuList = products.stream()
                .map(Product::getSku)
                .collect(Collectors.toList());

        // 批量查询已存在实体
        TypedQuery<Product> query = entityManager.createQuery(
                "SELECT p FROM Product p WHERE p.sku IN :skus", Product.class);
        query.setParameter("skus", skuList);
        Map<String, Product> existingProducts = query.getResultList()
                .stream()
                .collect(Collectors.toMap(Product::getSku, p -> p));

        // 处理实体更新/新增
        for (Product product : products) {
            Product existing = existingProducts.get(product.getSku());
            if (existing != null) {
                // 更新字段，托管状态下事务提交时自动同步到数据库
                existing.setName(product.getName());
                existing.setPrice(product.getPrice());
            } else {
                // 持久化新增实体
                entityManager.persist(product);
            }
        }
    }
}

基于Hibernate Session的实现（小数据量场景） #

利用Hibernate的自然ID加载能力，适合数据量不大的场景：

@Service
@Transactional
public class ProductService {
    private final SessionFactory sessionFactory;

    public ProductService(SessionFactory sessionFactory) {
        this.sessionFactory = sessionFactory;
    }

    public void batchUpsert(List<Product> products) {
        Session session = sessionFactory.getCurrentSession();
        for (Product product : products) {
            // 通过自然ID加载实体，不存在则返回null
            Product existing = session.byNaturalId(Product.class)
                    .using("sku", product.getSku())
                    .loadOptional()
                    .orElse(null);

            if (existing != null) {
                existing.setName(product.getName());
                existing.setPrice(product.getPrice());
            } else {
                session.persist(product);
            }
        }
    }
}

注意：该方式会遍历执行单条查询，大数据量场景下会产生N+1问题，建议用批量查询合并的方案。

核心注意事项 #

• 必须在事务中执行，保证更新和新增的原子性
• 数据库层面给自然ID字段加唯一约束，避免并发场景下的重复插入
• 根据业务需求控制更新字段，避免覆盖创建时间等无需修改的字段
• 大数据量场景务必使用批量查询和批量操作，避免性能损耗

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