Hibernate自定义SnowFlake ID生成器：实例化时获取ID值方法

情况1：想在实体实例化后立刻获取ID #

Hibernate无法通过注解自动实现这一点——因为你是手动new出的实体对象，此时它还不属于Hibernate的持久化上下文管理范畴，Hibernate不会主动介入为其生成ID。

你提到的手动调用ID生成器设置ID是最直接的方案，或者可以修改实体的构造方法，在实例化时自动生成ID：

@Entity
@Table(name = "shipment")
public class ShipmentEntity {

    @Id
    @GenericGenerator(name = "shipment", strategy = "com.example.domain.id.SnowFlakeIdGenerator")
    @GeneratedValue(generator = "shipment")
    @Column(name = "id")
    private Long id;
    
    public ShipmentEntity() {
        // 实例化时自动生成ID
        this.id = new SnowFlakeIdGenerator().generate(null, this);
    }
    ...
}

注意：如果你的SnowFlakeIdGenerator需要依赖外部配置（比如workerId），构造方法中直接实例化的方式可能需要调整；由于实体是POJO，不建议注入依赖，因此手动调用生成器仍是更稳妥的选择。

情况2：想在保存前（调用save后但未插入数据库时）获取ID #

这种场景可以通过修改自定义ID生成器的实现，让Hibernate在持久化阶段提前生成ID：

修改SnowFlakeIdGenerator，重写getGenerationTiming()方法，指定ID生成时机为持久化时而非插入数据库时：

public class SnowFlakeIdGenerator implements IdentifierGenerator {

    // 你的SnowFlake ID生成逻辑
    private Long generateSnowFlakeId() {
        // 实现你的SnowFlake生成逻辑
        return ...;
    }

    @Override
    public Serializable generate(SharedSessionContractImplementor session, Object object) throws HibernateException {
        return generateSnowFlakeId();
    }

    @Override
    public GenerationTiming getGenerationTiming() {
        // 指定在持久化时立即生成ID
        return GenerationTiming.IMMEDIATE;
    }
}

修改后，当你调用shipmentRepository.save(shipment)时，Hibernate会在执行数据库插入操作前就调用生成器为实体赋值ID，此时在save方法返回前就能获取到ID值：

public ShipmentEntity createShipment(CreateShipmentParam createShipmentParam) {
    ShipmentEntity shipment = new ShipmentEntity();
    shipment.setShippingDate(createShipmentParam.getShippingDate());
    ...
    ShipmentEntity savedShipment = shipmentRepository.save(shipment);
    System.out.println(savedShipment.getId()); // 此处已能获取到ID
    return savedShipment;
}

内容的提问来源于stack exchange，提问作者Eric Xin Zhang