JPA实现Postgres触发器生成主键表插入并返回主键方案咨询

直接使用继承JpaRepository的默认saveAndFlush等方法无法直接满足需求，但只需要对实体类主键字段做少量配置，不需要额外重写基础CRUD方法，即可适配触发器生成主键的逻辑，同时自动回写生成的主键值；如果不想依赖JPA实现厂商的专属特性，也可以通过自定义原生SQL方法实现。

你的场景下主键完全由数据库端BEFORE INSERT触发器生成，JPA不需要在应用层生成主键值，只需要做2个核心配置修改主键字段即可：

@Entity
@Validated
@Table(name = "my_table", schema="common")
public class MyModel {

    @Id
    // 配置1：insert语句不包含主键字段，完全交给触发器生成，避免传入null值干扰触发器判断
    @Column(name = "col_id", insertable = false)
    // 配置2：告知JPA该字段值在插入时由数据库生成，插入完成后自动回读值到实体属性
    @org.hibernate.annotations.Generated(org.hibernate.annotations.GenerationTime.INSERT)
    private String id;

    @Column(name = "second_col")
    private String secCol;
    @Column(name = "third_col")
    private String thirdCol;

    // 构造方法、getter/setter保留原有逻辑即可
    public MyModel() {}

    public MyModel(String secCol, String thirdCol) {
        // 不需要给id赋值，保持null即可
        this.secCol = secCol;
        this.thirdCol = thirdCol;
    }

    // 必须给id字段加setter，方便JPA回写生成的主键值
    public String getId() {return id;}
    public void setId(String id) {this.id = id;}
    // 其余字段getter/setter按常规逻辑编写即可
}

配置完成后直接调用默认的myRepo.saveAndFlush(myModel)即可：

• JPA生成的INSERT语句只会包含second_col、third_col两个字段，触发器可以正常读取字段值计算主键
• 插入完成后JPA会自动将触发器生成的col_id值回写到传入的myModel对象的id属性，直接调用myModel.getId()就能拿到生成的主键，不需要额外查询。

注意：不要给id字段配置GenerationType.IDENTITY/GenerationType.SEQUENCE等常规主键生成策略，这类配置会让JPA按自增/序列逻辑生成主键，干扰触发器执行。

如果你不想使用Hibernate专属的@Generated注解，可以直接在Repository中自定义原生SQL方法，利用PostgreSQL的INSERT ... RETURNING语法直接返回生成的主键，逻辑完全可控：

public interface MyRepo extends JpaRepository <MyModel, String> {
    @Modifying
    @Query(
        value = "INSERT INTO common.my_table(second_col, third_col) " +
                "VALUES (:#{#model.secCol}, :#{#model.thirdCol}) " +
                "RETURNING col_id",
        nativeQuery = true
    )
    String insertAndReturnGeneratedId(@Param("model") MyModel model);
}

调用该方法时会直接返回触发器生成的主键值，手动set回实体对象即可，不需要依赖任何JPA厂商的专属特性，也不会出现JPA自动生成SQL不符合预期的问题。

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