Hibernate 6.5基于@IdGeneratorType实现动态ID生成遇问题

可以通过@IdGeneratorType实现你想要的动态策略选择，但你的实现大概率遗漏了方言判断逻辑和生成时机的正确映射，导致Hibernate无法正确生成或获取ID。以下是具体的实现要点和修正方案：

核心要点说明 #

1. @IdGeneratorType的正确用法
   官方Javadoc的描述是准确的：IdentifierGenerator确实不适合和@IdGeneratorType配合，后者需要关联实现Generator接口的类——而你同时实现BeforeExecutionGenerator和OnExecutionGenerator是完全可行的，这两个都是Generator的子接口，分别对应"预生成ID（比如序列）"和"数据库生成后获取ID（比如identity）"的场景。

2. 动态策略的关键：方言判断时机
   必须在生成器初始化阶段（initialize方法）获取当前数据库方言，以此决定使用identity还是序列策略：

   • 若方言支持identity（如MySQL），则让数据库生成ID，Hibernate在插入后获取；
   • 若方言支持序列（如PostgreSQL），则提前从序列生成ID再插入。

修正后的完整实现 #

1. 自定义生成器注解 #

@IdGeneratorType(CustomIdGeneratorImpl.class)
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.FIELD, ElementType.METHOD})
public @interface CustomIdGenerator {
    // 允许自定义序列名，默认使用通用序列
    String sequenceName() default "global_entity_seq";
}

2. 生成器核心实现 #

public class CustomIdGeneratorImpl implements BeforeExecutionGenerator, OnExecutionGenerator {
    private boolean useIdentityStrategy;
    private SequenceGenerator sequenceGenerator;

    @Override
    public void initialize(IdentifierGeneratorDefinition definition, EntityPersister persister,
                          SessionFactoryImplementor sessionFactory) throws MappingException {
        // 从SessionFactory获取当前数据库方言
        Dialect dialect = sessionFactory.getJdbcServices().getDialect();
        useIdentityStrategy = dialect.supportsIdentityColumns();

        // 若不使用identity，初始化序列生成器
        if (!useIdentityStrategy) {
            CustomIdGenerator annotation = definition.getAnnotation(CustomIdGenerator.class);
            sequenceGenerator = new SequenceGenerator();
            sequenceGenerator.setSequenceName(annotation.sequenceName());
            sequenceGenerator.initialize(definition, persister, sessionFactory);
        }
    }

    // 根据策略返回生成时机：identity用AFTER（插入后获取），序列用BEFORE（插入前生成）
    @Override
    public GenerationTiming getGenerationTiming() {
        return useIdentityStrategy ? GenerationTiming.AFTER : GenerationTiming.BEFORE;
    }

    // 预生成ID逻辑（序列策略）
    @Override
    public Object generate(SharedSessionContractImplementor session, Object entity, Object currentValue, EventType eventType) {
        if (useIdentityStrategy) {
            // identity策略下返回null，让数据库生成ID
            return null;
        } else {
            // 调用序列生成器生成ID
            return sequenceGenerator.generate(session, entity, currentValue, eventType);
        }
    }

    // identity策略下，声明需要从数据库获取的ID字段
    @Override
    public String[] getGeneratedIdentifierPropertyNames() {
        return useIdentityStrategy ? new String[]{"id"} : new String[0];
    }

    // 获取数据库生成的ID（identity策略专用）
    @Override
    public Object generate(SharedSessionContractImplementor session, Object entity, Object currentValue, EventType eventType, IdentifierValue generatedValue) {
        return generatedValue.getValue();
    }
}

3. 实体类使用方式 #

@Entity
public class User {
    @Id
    @CustomIdGenerator(sequenceName = "user_seq")
    private Long id;

    // 其他字段和方法
}

原实现失效原因分析 #

你之前的代码可能没有做到以下几点：

• 在initialize方法中正确获取方言并切换策略；
• 针对不同策略返回正确的GenerationTiming，导致Hibernate不知道该提前生成还是事后获取ID；
• 为identity策略实现getGeneratedIdentifierPropertyNames方法，Hibernate无法识别需要从数据库读取生成的ID。

以上修正后，生成器会在应用启动时根据方言自动适配，插入数据时不会再出现ID为null的问题。

内容的提问来源于stack exchange，提问作者Vojtěch Sýkora