JPA中如何用自定义查询替代SequenceGenerator生成实体ID？

没问题！你完全可以用自定义查询（比如从视图取值）来替代默认的序列nextval语句生成ID，下面给你分享几种实用的方案，适配不同的场景：

方案1：自定义标识符生成器（基于Hibernate） #

如果你的项目用的是Hibernate作为JPA实现，最规范的方式是自定义一个标识符生成器，让它执行你的视图查询来获取ID。

步骤1：实现自定义生成器类 #

创建一个类实现Hibernate的IdentifierGenerator接口，在generate方法里执行你的视图查询：

import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.id.IdentifierGenerator;
import java.io.Serializable;

public class ViewBasedIdGenerator implements IdentifierGenerator {

    @Override
    public Serializable generate(SharedSessionContractImplementor session, Object object) {
        // 执行自定义视图查询获取ID
        String viewQuery = "select value from NAME_OF_VIEW";
        return (Long) session.createNativeQuery(viewQuery).uniqueResult();
    }
}

步骤2：修改实体类注解 #

把原来的@SequenceGenerator和@GeneratedValue替换成自定义生成器的配置：

/** 
 * User hierarchy JPA entity. 
 */
@Entity
@Table(name="USER_HIERARCHY")
@Data
@AllArgsConstructor
@RequiredArgsConstructor
@Builder
@EqualsAndHashCode(of={"id"},callSuper=false)
@ToString(of={"id"})
public class UserHierarchy extends AuditableEntity{
    /**
     * User hierarchy id.
     *
     * @param id New value for this User hierarchy's id.
     * @return The current value of this User hierarchy's id.
     */
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY, generator = "viewBasedIdGenerator")
    @GenericGenerator(
        name = "viewBasedIdGenerator",
        strategy = "com.yourcompany.yourpackage.ViewBasedIdGenerator" // 替换成你的类路径
    )
    private Long id;

    // 其他字段保持不变...
}

方案2：手动获取ID再持久化（简单直接） #

如果不想写自定义生成器，也可以在持久化实体前手动执行查询获取ID，再赋值给实体的id字段：

// 1. 先从视图查询获取ID
Long newHierarchyId = entityManager.createNativeQuery("select value from NAME_OF_VIEW")
                                   .getSingleResult();

// 2. 构建实体并设置ID
UserHierarchy hierarchy = UserHierarchy.builder()
                                       .id(newHierarchyId)
                                       .user(targetUser)
                                       .team(targetTeam)
                                       // 其他字段赋值...
                                       .build();

// 3. 持久化实体
entityManager.persist(hierarchy);

⚠️ 注意：这种方式要去掉实体上的@GeneratedValue注解，避免JPA自动生成ID；同时要确保视图查询的ID是唯一的，避免并发场景下的主键冲突。

方案3：数据库触发器（数据库层面处理） #

如果你的数据库支持触发器（比如PostgreSQL、MySQL），可以把ID生成逻辑放到数据库层面，JPA代码几乎不用改动：

示例（PostgreSQL触发器） #

-- 创建触发器函数：从视图取ID赋值给新插入的记录
CREATE OR REPLACE FUNCTION set_user_hierarchy_id()
RETURNS TRIGGER AS $$
BEGIN
  SELECT value INTO NEW.id FROM NAME_OF_VIEW;
  RETURN NEW;
END;
$$ LANGUAGE plpgsql;

-- 给USER_HIERARCHY表绑定插入触发器
CREATE TRIGGER trigger_set_user_hierarchy_id
BEFORE INSERT ON USER_HIERARCHY
FOR EACH ROW EXECUTE FUNCTION set_user_hierarchy_id();

这时候实体类的id字段可以保留@GeneratedValue(strategy = GenerationType.IDENTITY)，JPA会自动识别数据库生成的ID；或者也可以去掉生成注解，让触发器完全接管ID赋值。

内容的提问来源于stack exchange，提问作者Carlos Martínez Romero