基于现有主键最大值自动生成新主键的JPA @GeneratedValue策略咨询

我完全懂你的痛点——在已经有大量数据的多环境（DEV、TEST等）表中引入JPA，既不想手动写DB查询去拿主键最大值，又没法硬编码initialValue，还绝对不能删现有数据，现在Hibernate一上来就从ID=0开始插，这肯定会冲突，确实头疼。下面给你几个实用的解决方案：

方案1：用数据库原生自增策略（最推荐） #

如果你的数据库支持自增列（比如MySQL的AUTO_INCREMENT、PostgreSQL的SERIAL/BIGSERIAL、SQL Server的IDENTITY），直接用GenerationType.IDENTITY就搞定了。这个策略会完全交给数据库管理主键自增，它会自动识别表中已有的主键最大值，新插入的记录直接从最大值+1开始生成，完全不用你操心多环境的差异问题。

修改你的示例代码如下：

@Entity
@Table(name="DUMMY")
public class Dummy {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Column(name="primary_key_dummy")
    private Integer pk;
}

这个方案的优势是零额外维护，完全依赖数据库原生能力，规避了Hibernate自己生成ID导致的冲突问题，各个环境的数据库会自己处理主键序列，完美适配你的场景。

方案2：利用数据库序列（适合支持序列的数据库） #

如果你的数据库用的是序列（比如Oracle、PostgreSQL的序列），可以先给每个环境的序列初始化到对应表的主键最大值，然后用GenerationType.SEQUENCE配合@SequenceGenerator。

比如在Oracle中，先执行SQL获取当前主键最大值：

SELECT MAX(primary_key_dummy) FROM DUMMY;

假设得到最大值是1000，就把序列的起始值改成1001：

ALTER SEQUENCE dummy_seq START WITH 1001;

然后在代码中配置：

@Entity
@Table(name="DUMMY")
public class Dummy {
    @Id
    @GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "dummy_seq")
    @SequenceGenerator(name = "dummy_seq", sequenceName = "dummy_seq", allocationSize = 1)
    @Column(name="primary_key_dummy")
    private Integer pk;
}

这个方案只需要在每个环境执行一次序列初始化脚本，之后Hibernate就会从序列的当前值开始生成主键，不用再管后续的自增逻辑。

方案3：自定义主键生成器（灵活但略复杂） #

如果上述两种方案都不适用，你可以自定义一个主键生成器，让它在第一次生成ID时自动查询表的主键最大值，之后在此基础上自增。不过要注意并发场景下的主键冲突问题，最好配合数据库的唯一约束。

先实现Hibernate的IdentifierGenerator接口：

public class CustomIdGenerator implements IdentifierGenerator {
    @Override
    public Serializable generate(SharedSessionContractImplementor session, Object object) throws HibernateException {
        // 获取实体对应的表名和主键列名
        Table tableAnnotation = object.getClass().getAnnotation(Table.class);
        String tableName = tableAnnotation.name();
        
        Field idField = Arrays.stream(object.getClass().getDeclaredFields())
                .filter(f -> f.isAnnotationPresent(Id.class))
                .findFirst()
                .orElseThrow(() -> new HibernateException("找不到主键字段"));
        String idColumnName = idField.getAnnotation(Column.class).name();
        
        // 查询当前主键最大值
        String sql = String.format("SELECT MAX(%s) FROM %s", idColumnName, tableName);
        Integer maxId = (Integer) session.createNativeQuery(sql).uniqueResult();
        
        // 返回新的主键值
        return maxId != null ? maxId + 1 : 1;
    }
}

然后在实体类中使用这个自定义生成器：

@Entity
@Table(name="DUMMY")
public class Dummy {
    @Id
    @GenericGenerator(name = "custom_id_generator", strategy = "com.yourpackage.CustomIdGenerator")
    @GeneratedValue(generator = "custom_id_generator")
    @Column(name="primary_key_dummy")
    private Integer pk;
}

这个方案适合特殊场景，但如果系统并发量较高，建议在查询最大值时加锁，避免多个线程同时生成相同的ID。

优先推荐方案1，因为它最省心，完全不需要额外的配置或脚本，直接利用数据库的原生能力就能解决你的问题。

备注：内容来源于stack exchange，提问作者ckaus