Hibernate自定义UUID生成器：手动/自动ID分配实现方案问询

我定义了如下Car实体类：

@Getter
@Setter
@Entity
@Table(name="car", schema="my_schema")
@Builder // 及其他注解...
public class Car{

@NotNull
@Id
@Column(columnDefinition= "uuid", name="id", nullable=false)
@GeneratedValue(generator="CarGenerator")
@GenericGenerator(
    name="CarGenerator",
    strategy = "com.myCars.utils.CarGenerator"
)
private UUID id;

}

以及自定义的CarGenerator类：

public class CarGenerator extends AbstractUUIDGenerator{

    @Override
    public Object generate(SharedSessionContractImplementator sharedSessionContractImplementator, Object o) throws HibernateException{
        if (o instanceof Car car && car.getId() != null){
            return car.getId();
        }
        return UUID.randomUUID();   
    }   

}

我的需求是：当未手动指定ID时，自动生成UUID；当手动指定ID时，使用指定的值存入数据库。例如在测试中：

Car car = Car.builder()
.brand("Toyota")
.model("Camry")
.build();
carRepository.save(car);
// 期望此处自动分配随机UUID

Car car = Car.builder()
.id(UUID.fromString("08f2364e-1fe3-4807-8378-09006f85e837"))
.brand("Toyota")
.model("Camry")
.build();
carRepository.save(car);
// 期望此处使用手动指定的ID存入数据库

我希望这两种场景都能正常工作，目前考虑自定义CarBuilder，但想了解是否有更优雅的实现方案？

你当前的CarGenerator逻辑已经能满足需求，无需额外自定义Builder，只需调整几个细节就能让功能完全生效：

1. 移除ID字段的@NotNull注解
   你给id加了@NotNull，但未指定ID时该字段为null，会触发JSR-380校验失败。数据库层面的非空约束已经通过@Column(nullable = false)保证，无需额外加@NotNull。修改后的ID字段定义：

   @Id
   @Column(columnDefinition = "uuid", name = "id", nullable = false)
   @GeneratedValue(generator = "CarGenerator")
   @GenericGenerator(
       name = "CarGenerator",
       strategy = "com.myCars.utils.CarGenerator"
   )
   private UUID id;
   

2. 确认CarGenerator逻辑有效性
   你的Generator逻辑完全正确：Hibernate执行save()时会先调用generate()方法，若实体已设置ID则直接返回该值，否则生成随机UUID。这正好匹配你的两种场景需求，无需修改Generator代码。

3. 直接用现有测试代码验证
   运行你提供的测试用例：

   • 未设置ID时，Generator自动生成随机UUID并存入数据库；
   • 手动指定ID时，Generator复用该值完成入库。

可选优化方案 #

如果想进一步简化实现，也可以用Hibernate内置能力替代自定义Generator：

• 使用uuid2策略的内置生成器，配合@GeneratedValue注解：

  @Id
  @Column(columnDefinition = "uuid", name = "id", nullable = false)
  @GeneratedValue(strategy = GenerationType.AUTO, generator = "uuid2")
  @GenericGenerator(name = "uuid2", strategy = "uuid2")
  private UUID id;
  

  这种情况下，Hibernate会在ID为null时自动生成UUID，若手动设置了ID则直接使用。不过自定义Generator的方式更灵活，方便后续扩展生成规则。

如果需要在内存中就生成ID（而非等到持久化阶段），可以自定义Builder：

@Builder
public class Car {
    // 字段定义...

    public static class CarBuilder {
        public Car build() {
            if (this.id == null) {
                this.id = UUID.randomUUID();
            }
            return new Car(this.id, this.brand, this.model);
        }
    }
}

但这种方式将ID生成逻辑从持久层移到了实体层，适合需要提前获取ID的场景，按需选择即可。

内容的提问来源于stack exchange，提问作者user7303261