背景与挑战
在基于Hibernate的应用开发中，为实体生成唯一标识符是常见需求。虽然Hibernate提供了多种内置的ID生成策略（如自增、UUID等），但在某些业务场景下，我们需要更灵活的自定义ID生成规则，例如为不同类型的实体生成带有特定前缀的UUID。一个典型的问题是，当有多个实体（如Product、Order、User等）都需要这种前缀+UUID的ID格式，但前缀各不相同时，开发者往往会为每个实体创建独立的IdentifierGenerator实现类，导致代码冗余和维护成本增加。
例如，原始方法可能如下所示，为Product实体创建一个专门的ID生成器：
import java.io.Serializable;
import java.util.UUID;
import org.hibernate.HibernateException;
import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.id.IdentifierGenerator;
public class ProductIdGenerator implements IdentifierGenerator {
public static final String GENERATOR_NAME = "productIdGenerator"; // 注意：原始问题中的拼写错误已修正
@Override
public Serializable generate(SharedSessionContractImplementor session, Object entity) throws HibernateException {
String prefix = "PROD"; // 硬编码的前缀
String uuid = UUID.randomUUID().toString().substring(0, 8);
return prefix + uuid;
}
}并在实体中使用：
import org.hibernate.annotations.GenericGenerator;
import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;
@Entity
public class Product {
@Id
@GeneratedValue(generator = ProductIdGenerator.GENERATOR_NAME)
@GenericGenerator(name = ProductIdGenerator.GENERATOR_NAME, strategy = "net.ddns.mrq.util.ProductIdGenerator") // 替换为实际包名
@Column(name = "product_id")
private String id;
private String name;
// ... 其他字段
}当有八个甚至更多的实体需要不同前缀时，创建八个类似的生成器类显然不是最优解。本文将介绍两种更优雅的解决方案。
解决方案一：通过实体接口动态获取前缀
这种方法的核心思想是让需要自定义ID前缀的实体实现一个通用接口，该接口定义一个方法来返回对应实体的前缀。IdentifierGenerator在生成ID时，可以尝试将传入的实体对象转换为这个接口类型，并调用其方法获取前缀。
实现思路：
-
定义通用接口： 创建一个接口，例如PrefixableEntity，包含一个getPrefix()方法。
public interface PrefixableEntity { String getPrefix(); } -
实体实现接口： 各个实体类实现PrefixableEntity接口，并根据自身需求返回相应的前缀。
@Entity public class Product implements PrefixableEntity { // ... @Id 和其他字段 @Override public String getPrefix() { return "PROD"; } } @Entity public class Order implements PrefixableEntity { // ... @Id 和其他字段 @Override public String getPrefix() { return "ORD"; } } -
修改ID生成器： 在generate方法中，检查实体是否实现了PrefixableEntity接口，并据此获取前缀。
import java.io.Serializable; import java.util.UUID; import org.hibernate.HibernateException; import org.hibernate.engine.spi.SharedSessionContractImplementor; import org.hibernate.id.IdentifierGenerator; public class DynamicPrefixIdGenerator implements IdentifierGenerator { public static final String GENERATOR_NAME = "dynamicPrefixIdGenerator"; @Override public Serializable generate(SharedSessionContractImplementor session, Object entity) throws HibernateException { String prefix = ""; if (entity instanceof PrefixableEntity) { prefix = ((PrefixableEntity) entity).getPrefix(); } else { // 如果实体未实现接口，可以抛出异常或使用默认前缀 throw new HibernateException("Entity " + entity.getClass().getName() + " does not implement PrefixableEntity."); } String uuid = UUID.randomUUID().toString().substring(0, 8); return prefix + uuid; } } -
实体使用生成器：
@Entity public class Product implements PrefixableEntity { @Id @GeneratedValue(generator = DynamicPrefixIdGenerator.GENERATOR_NAME) @GenericGenerator(name = DynamicPrefixIdGenerator.GENERATOR_NAME, strategy = "com.yourpackage.DynamicPrefixIdGenerator") // 替换为实际包名 @Column(name = "product_id") private String id; // ... @Override public String getPrefix() { return "PROD"; } }
优点： 逻辑清晰，实体直接声明其前缀。 缺点： 实体类需要实现一个额外的接口，增加了实体与ID生成逻辑的耦合。
解决方案二：利用Hibernate Configurable 接口进行参数化配置（推荐）
这是更推荐的方法，它利用了Hibernate IdentifierGenerator 可以实现 org.hibernate.id.Configurable 接口的特性。通过这个接口，我们可以在 @GenericGenerator 注解中通过 @Parameter 传递配置参数，从而实现同一个生成器类在不同实体上使用不同的前缀。
实现步骤：
-
修改ID生成器： 让自定义的IdentifierGenerator实现org.hibernate.id.Configurable接口。
- Configurable接口要求实现一个configure(Type type, Properties params, ServiceRegistry serviceRegistry)方法。
- 在这个方法中，我们可以从params（一个Properties对象）中读取通过注解传递的参数。
import java.io.Serializable; import java.util.Properties; import java.util.UUID; import org.hibernate.HibernateException; import org.hibernate.MappingException; import org.hibernate.engine.spi.SharedSessionContractImplementor; import org.hibernate.id.Configurable; import org.hibernate.id.IdentifierGenerator; import org.hibernate.service.ServiceRegistry; import org.hibernate.type.Type; public class ParameterizedPrefixIdGenerator implements IdentifierGenerator, Configurable { public static final String GENERATOR_NAME = "parameterizedPrefixIdGenerator"; public static final String PREFIX_PARAM = "prefix"; // 定义参数名 private String prefix = ""; // 存储从配置中读取的前缀 @Override public void configure(Type type, Properties params, ServiceRegistry serviceRegistry) throws MappingException { // 从参数中获取前缀，如果未设置则默认为空字符串 this.prefix = params.getProperty(PREFIX_PARAM, ""); if (this.prefix.isEmpty()) { // 可以选择抛出异常或提供一个默认值 throw new MappingException("Prefix parameter '" + PREFIX_PARAM + "' is required for " + GENERATOR_NAME); } } @Override public Serializable generate(SharedSessionContractImplementor session, Object entityObject) throws HibernateException { String uuid = UUID.randomUUID().toString().substring(0, 8); return prefix + uuid; } } -
实体使用生成器并传递参数： 在实体中使用 @GenericGenerator 注解时，通过 @Parameter 注解来指定前缀。
import org.hibernate.annotations.GenericGenerator; import org.hibernate.annotations.Parameter; import javax.persistence.Column; import javax.persistence.Entity; import javax.persistence.GeneratedValue; import javax.persistence.Id; @Entity public class Product { @Id @GeneratedValue(generator = ParameterizedPrefixIdGenerator.GENERATOR_NAME) @GenericGenerator( name = ParameterizedPrefixIdGenerator.GENERATOR_NAME, strategy = "com.yourpackage.ParameterizedPrefixIdGenerator", // 替换为实际包名 parameters = { @Parameter(name = ParameterizedPrefixIdGenerator.PREFIX_PARAM, value = "PROD") // 为Product实体设置前缀 } ) @Column(name = "product_id") private String id; private String name; // ... 其他字段 }对于另一个实体，例如Order，可以这样配置：
@Entity public class Order { @Id @GeneratedValue(generator = ParameterizedPrefixIdGenerator.GENERATOR_NAME) @GenericGenerator( name = ParameterizedPrefixIdGenerator.GENERATOR_NAME, strategy = "com.yourpackage.ParameterizedPrefixIdGenerator", // 替换为实际包名 parameters = { @Parameter(name = ParameterizedPrefixIdGenerator.PREFIX_PARAM, value = "ORD") // 为Order实体设置不同前缀 } ) @Column(name = "order_id") private String id; private String description; // ... 其他字段 }
优点：
- 高复用性： 只需要一个IdentifierGenerator类即可服务所有需要前缀的实体。
- 低耦合： 实体类无需实现额外接口，ID生成逻辑与实体本身解耦。
- 配置灵活： 前缀直接在实体注解中配置，清晰明了，易于修改。
- 可扩展性： 可以通过@Parameter传递更多配置，例如UUID的长度、分隔符等。
注意事项：
- 确保strategy属性中的类路径是正确的。
- @Parameter注解中的name属性必须与IdentifierGenerator实现类中用于读取参数的键名一致（例如PREFIX_PARAM）。
- 在configure方法中，应处理参数缺失的情况，例如提供默认值或抛出MappingException。
总结
通过实现org.hibernate.id.Configurable接口并利用@GenericGenerator的@Parameter功能，我们可以极大地优化Hibernate自定义ID生成器的设计，实现单一生成器类服务多实体、多前缀的需求。这种方法不仅减少了代码冗余，提高了代码的复用性和可维护性，也使得ID生成策略的配置更加灵活和声明式。相较于通过实体接口获取前缀的方案，参数化配置提供了更好的解耦性和更清晰的配置方式，是处理此类问题的首选策略。
