JPA 主键何时被赋值？ | Baeldung中文网

1. 概述

本文将深入探讨 JPA 在何时为实体的主键（Primary Key）赋值。我们会先解析 JPA 规范中的定义，再通过实际代码示例，展示不同主键生成策略（@GeneratedValue）下的行为差异。

掌握这一点对理解实体生命周期、事务边界以及 ID 可见性非常关键，尤其在处理级联操作或依赖主键做后续逻辑时，踩坑很容易。

2. 问题背景

JPA 通过 EntityManager 管理实体的生命周期。当我们调用 persist() 方法时，实体从“瞬时状态（transient）”变为“托管状态（managed）”。

但关键问题是：主键到底什么时候被设置到实体对象上的？

JPA 规范明确指出：

一个新实体实例通过调用 persist 方法或级联 persist 操作，成为托管且持久化的。

因此，我们重点关注 EntityManager.persist() 调用前后，主键的赋值时机。

3. 主键生成策略与赋值时机

JPA 提供了四种主要的主键生成策略，它们在 主键赋值时机 上有显著差异：

✅ GenerationType.AUTO
✅ GenerationType.IDENTITY
✅ GenerationType.SEQUENCE
✅ GenerationType.TABLE

下面我们逐个分析。

3.1 GenerationType.AUTO

这是 @GeneratedValue 的默认策略。JPA 提供商会根据底层数据库自动选择最合适的策略（通常是 IDENTITY 或 SEQUENCE）。

@Entity
@Table(name = "app_admin")
public class Admin {

    @Id
    @GeneratedValue
    private Long id;

    @Column(name = "admin_name")
    private String name;

    // standard getters and setters
}

⚠️ 注意：由于行为依赖具体实现和数据库，不建议在生产环境中使用 AUTO，容易造成行为不一致，尤其是在迁移数据库时。

3.2 GenerationType.IDENTITY

该策略依赖数据库的 自增列（auto-increment）。主键由数据库在 INSERT 执行时生成。

✅ 赋值时机：事务提交时（或 insert 执行后），JPA 才能从数据库获取生成的 ID 并回填到实体中。

支持数据库：MySQL、PostgreSQL、SQL Server、DB2、Derby、Sybase。

示例实体：

@Entity
@Table(name = "app_user")
public class User {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @Column(name = "user_name")
    private String name;

    // standard getters and setters
}

测试验证：

@Test
public void givenIdentityStrategy_whenCommitTransction_thenReturnPrimaryKey() {
    User user = new User();
    user.setName("TestName");
        
    entityManager.getTransaction().begin();
    entityManager.persist(user);
    Assert.assertNull(user.getId()); // 此时 ID 仍为 null
    entityManager.getTransaction().commit();

    Long expectPrimaryKey = 1L;
    Assert.assertEquals(expectPrimaryKey, user.getId()); // 提交后才有值
}

⚠️ 踩坑点：在事务提交前无法获取 ID，若后续逻辑依赖 ID（如放入缓存、发消息），必须等到 flush 或 commit。

3.3 GenerationType.SEQUENCE

使用数据库的 序列（Sequence） 来生成主键。需要提前在数据库创建序列。

创建序列示例（如 PostgreSQL）：

CREATE SEQUENCE article_seq
  MINVALUE 1
  START WITH 50
  INCREMENT BY 50

✅ 赋值时机：调用 persist() 后，事务提交前。JPA 会立即从序列获取下一个值并设置到实体。

实体定义：

@Entity
@Table(name = "article")
public class Article {
    
    @Id
    @GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "article_gen")
    @SequenceGenerator(name="article_gen", sequenceName="article_seq")
    private Long id;

    @Column(name = "article_name")
    private String name;

    // standard getters and setters
}

测试验证：

@Test
public void givenSequenceStrategy_whenPersist_thenReturnPrimaryKey() {
    Article article = new Article();
    article.setName("Test Name");

    entityManager.getTransaction().begin();
    entityManager.persist(article);
    Long expectPrimaryKey = 51L; // 序列从50开始，下一个为51
    Assert.assertEquals(expectPrimaryKey, article.getId()); // persist 后立即有值

    entityManager.getTransaction().commit();
}

支持数据库：Oracle、PostgreSQL、DB2。

✅ 优势：ID 提前可知，适合需要在事务中使用主键的场景。

3.4 GenerationType.TABLE

通过一张独立的数据库表来模拟序列，实现跨数据库兼容。

需要先创建一个“主键生成器表”：

@Table(name = "id_gen")
@Entity
public class IdGenerator {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    
    @Column(name = "gen_name")
    private String gen_name;

    @Column(name = "gen_value")
    private Long gen_value;

    // standard getters and setters
}

初始化数据：

INSERT INTO id_gen (gen_name, gen_value) VALUES ('id_generator', 0);
INSERT INTO id_gen (gen_name, gen_value) VALUES ('task_gen', 10000);

✅ 赋值时机：调用 persist() 后，事务提交前。JPA 会查询并更新该表，获取下一个 ID。

使用示例：

@Entity
@Table(name = "task")
public class Task {
    
    @TableGenerator(
        name = "id_generator", 
        table = "id_gen", 
        pkColumnName = "gen_name", 
        valueColumnName = "gen_value",
        pkColumnValue = "task_gen", 
        initialValue = 10000, 
        allocationSize = 10
    )
    @Id
    @GeneratedValue(strategy = GenerationType.TABLE, generator = "id_generator")
    private Long id;

    @Column(name = "name")
    private String name;

    // standard getters and setters
}

测试验证：

@Test
public void givenTableStrategy_whenPersist_thenReturnPrimaryKey() {
    Task task = new Task();
    task.setName("Test Task");

    entityManager.getTransaction().begin();
    entityManager.persist(task);
    Long expectPrimaryKey = 10000L;
    Assert.assertEquals(expectPrimaryKey, task.getId()); // persist 后立即有值

    entityManager.getTransaction().commit();
}

⚠️ 缺点：性能较差，每次生成 ID 都要访问数据库表，且可能引发锁竞争。

✅ 优点：数据库无关，适合异构环境或无法使用序列的数据库。

4. 总结

策略	赋值时机	数据库支持	是否推荐
`AUTO`	实现依赖	所有	❌ 不推荐
`IDENTITY`	提交时（或 flush）	MySQL, SQL Server, PostgreSQL 等	✅ 常用
`SEQUENCE`	`persist()` 后立即	Oracle, PostgreSQL, DB2	✅ 高性能推荐
`TABLE`	`persist()` 后立即	所有	⚠️ 仅用于特殊场景

📌 核心结论：

如果你需要在 persist() 后立刻拿到 ID（比如用于日志、缓存 key），优先选 SEQUENCE 或 TABLE。
如果使用 MySQL，IDENTITY 是最简单粗暴的选择，但要接受“提交前无 ID”的限制。
避免使用 AUTO，显式指定策略更安全可控。

完整示例代码可在 GitHub 获取：https://github.com/baeldung/tutorials/tree/master/persistence-modules/java-jpa-2

Persistence

REST

Security

1. 概述

2. 问题背景

3. 主键生成策略与赋值时机

3.1 GenerationType.AUTO

3.2 GenerationType.IDENTITY

3.3 GenerationType.SEQUENCE

3.4 GenerationType.TABLE

4. 总结