在 Java 持久化领域，Hibernate 一直是 ORM 框架的标杆。随着 Hibernate 6 的正式发布，其底层架构与 API 迎来了多项重要改进，其中自定义 ID 生成器的增强尤为引人注目。新版本允许开发者实现一种更智能的 ID 生成策略：如果实体已存在业务定义的 ID，则直接使用；否则优雅地降级为数据库自增或序列生成。这一特性解决了长期以来开发者需要手动编写复杂逻辑的痛点，为数据迁移、分布式 ID 兼容以及遗留系统整合提供了更简洁的解决方案。

背景：ID 生成策略的演进

传统 Hibernate 中，ID 生成器（如 IDENTITY、SEQUENCE、TABLE）大多为“无状态”模式，即一旦配置了自增策略，所有新实体的 ID 都必须由数据库或生成器统一管理。这导致一个常见困境：当系统需要从外部导入已有 ID 的数据（例如数据迁移、多数据源合并）时，开发者不得不手动关闭 ID 生成器，或者通过 @GeneratedValue 与 @GenericGenerator 的组合编写复杂的自定义生成器，且通常无法在运行时动态决定是“使用已有 ID”还是“生成新 ID”。

Hibernate 6 的新方案：智能回退生成器

Hibernate 6 在 org.hibernate.id 包下引入了对“回退模式”（fallback）的原生支持。核心思路是定义一个生成器，该生成器在执行 generate() 方法时首先检查实体对象中是否已经设置了 ID 属性（即 id 字段非空且有效）。如果已有值，则直接返回该值，跳过自动生成；如果 ID 为 null 或默认值（如 0），则调用底层的自增或序列生成逻辑。

这一机制通过新的 IdentifierGenerator 接口以及内置的 CustomIdGenerator 支持来实现。开发者只需继承 AbstractIdGenerator 或实现 IdentifierGenerator，并在 generate() 方法中通过 session.getEntityPersister() 或直接读取实体属性来判断，即可完成逻辑。

代码示例（示意）

public class FallbackIdGenerator implements IdentifierGenerator {
    @Override
    public Object generate(SharedSessionContractImplementor session, Object object) {
        // 尝试获取当前实体的 ID 属性值（假设字段名为 id）
        if (object instanceof BaseEntity) {
            Serializable existingId = ((BaseEntity) object).getId();
            if (existingId != null && !existingId.equals(0)) {
                return existingId; // 使用已有 ID
            }
        }
        // 回退：调用默认的自增策略（例如 IDENTITY）
        return IdentifierGeneratorHelper.get().generate(session, object);
    }
}

在实体类中，通过 @GenericGenerator 引用该自定义生成器：

@Entity
@GenericGenerator(name = "fallback", type = FallbackIdGenerator.class)
@Id @GeneratedValue(generator = "fallback")
private Long id;

三大典型应用场景

  1. 数据迁移与批量导入：当需要从旧系统导入数十万条记录，且每条记录附带有业务含义的 ID（如订单号、员工工号）时，可直接在实体中设置 ID，Hibernate 6 会自动保留；而未设置的记录则正常自增。避免了“先导入再更新 ID”的两步操作。

  2. 分布式 ID 与业务 ID 共存：微服务架构中，部分实体的 ID 由分布式 ID 生成器（如雪花算法）提供，另一部分仍依赖数据库自增。通过自定义回退生成器，可以在同一个实体类中灵活处理两种模式，无需切换生成策略。

  3. 多租户系统兼容：不同租户可能使用不同 ID 生成方式（有的手工分配，有的自动生成）。将自定义生成器与租户上下文结合，可在运行时动态决定是否使用已有 ID，提升系统扩展性。

注意事项与最佳实践

  • AbstractIdGenerator 在 Hibernate 6 中已被标记为弃用（deprecated），建议直接实现 IdentifierGenerator 或使用 CompositeIdGenerator 组合。
  • 回退逻辑需要谨慎处理 ID 的合法性。例如，如果业务 ID 允许为 0 或特定字符串，需要明确定义“未设置”的判定条件，避免误判。
  • 对于使用 @Id 且类型为基本类型（如 long）的实体，建议使用包装类型 Long，并将默认值设为 null，以便区分“未设置”与“合法 0”。
  • 该生成器在批处理（batch）场景下仍需注意性能，因为每次生成都需要检查实体状态，但相比手动配置 Agent 或拦截器，效率已显著提升。

总结：更灵活，更贴近业务

Hibernate 6 的自定义 ID 生成器增强，本质上是对“单一生成策略”的打破。通过“先检后用”的回退机制，开发者得以用极简的代码实现业务逻辑与持久化生成的解耦。这一变化不仅降低了遗留系统迁移的复杂度，也为现代云原生应用中的 ID 管理提供了新的范式。对于正在升级到 Hibernate 6 或面临 ID 兼容挑战的团队而言，这无疑是一项值得立即关注的生产力提升。