Spring Boot中能否重写JpaRepository的saveAll方法以使用自定义UPSERT查询？

当然可行！这是优化批量数据写入性能的常见操作，尤其是当你需要同时处理新增和更新数据时。我来给你梳理两种常用的实现方案：

方案一：自定义Repository实现类重写saveAll #

这种方式灵活性最高，能完全掌控SQL逻辑，适合复杂的UPSERT场景。

步骤1：定义实体类 #

假设你的实体类是User，带有@Id主键：

@Entity
@Table(name = "users")
public class User {
    @Id
    private Long id;
    private String username;
    private String email;
    
    // 省略getter、setter和构造方法
}

步骤2：扩展JpaRepository接口 #

声明要重写的saveAll方法：

public interface UserRepository extends JpaRepository<User, Long> {
    @Override
    <S extends User> List<S> saveAll(Iterable<S> entities);
}

步骤3：编写Repository实现类 #

继承SimpleJpaRepository，实现自定义的saveAll逻辑：

@Repository
public class UserRepositoryImpl extends SimpleJpaRepository<User, Long> implements UserRepository {

    private final EntityManager entityManager;

    // 构造方法必须继承父类的参数
    public UserRepositoryImpl(JpaEntityInformation<User, Long> entityInfo, EntityManager entityManager) {
        super(entityInfo, entityManager);
        this.entityManager = entityManager;
    }

    @Override
    @Transactional // 必须添加事务注解保证原子性
    public <S extends User> List<S> saveAll(Iterable<S> entities) {
        List<S> entityList = StreamSupport.stream(entities.spliterator(), false)
                .collect(Collectors.toList());

        if (entityList.isEmpty()) {
            return Collections.emptyList();
        }

        // 构建批量UPSERT SQL（这里以PostgreSQL为例）
        String tableName = User.class.getAnnotation(Table.class).name();
        StringBuilder sql = new StringBuilder();
        sql.append("INSERT INTO ").append(tableName)
                .append(" (id, username, email) VALUES ");

        // 生成占位符和参数列表
        List<String> placeholders = new ArrayList<>();
        List<Object> params = new ArrayList<>();
        for (S user : entityList) {
            placeholders.add("(?, ?, ?)");
            params.add(user.getId());
            params.add(user.getUsername());
            params.add(user.getEmail());
        }
        sql.append(String.join(", ", placeholders));

        // 添加UPSERT冲突处理逻辑
        sql.append(" ON CONFLICT (id) DO UPDATE SET ")
                .append("username = EXCLUDED.username, email = EXCLUDED.email");

        // 执行原生SQL
        Query query = entityManager.createNativeQuery(sql.toString());
        for (int i = 0; i < params.size(); i++) {
            query.setParameter(i + 1, params.get(i));
        }
        query.executeUpdate();

        return entityList;
    }
}

方案二：用@Query注解结合默认方法实现 #

这种方式更简洁，适合字段较少、逻辑简单的场景：

public interface UserRepository extends JpaRepository<User, Long> {

    @Modifying
    @Transactional
    @Query(value = "INSERT INTO users (id, username, email) VALUES (:ids, :usernames, :emails) " +
            "ON CONFLICT (id) DO UPDATE SET username = EXCLUDED.username, email = EXCLUDED.email",
            nativeQuery = true)
    void batchUpsert(@Param("ids") List<Long> ids, 
                     @Param("usernames") List<String> usernames, 
                     @Param("emails") List<String> emails);

    // 重写saveAll，调用自定义的批量UPSERT方法
    @Override
    default <S extends User> List<S> saveAll(Iterable<S> entities) {
        List<S> entityList = StreamSupport.stream(entities.spliterator(), false)
                .collect(Collectors.toList());
        
        if (!entityList.isEmpty()) {
            List<Long> ids = entityList.stream().map(User::getId).toList();
            List<String> usernames = entityList.stream().map(User::getUsername).toList();
            List<String> emails = entityList.stream().map(User::getEmail).toList();
            
            batchUpsert(ids, usernames, emails);
        }
        return entityList;
    }
}

关键注意事项 #

• 数据库兼容性：不同数据库的UPSERT语法差异很大：
  • MySQL使用ON DUPLICATE KEY UPDATE
  • Oracle使用MERGE INTO
  • SQL Server使用MERGE
• 事务管理：所有修改操作必须添加@Transactional注解，避免数据不一致
• 字段灵活性：如果实体字段较多，建议用反射自动获取字段名，避免硬编码SQL
• 空值处理：记得判断输入的Iterable是否为空，避免执行无效SQL

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