使用MySQL和Oracle进行Hibernate自动密钥生成

Question

我正在使用 Java 应用程序，该应用程序应该在具有相同数据库架构的两个不同数据库上执行CRUD操作（使用 Hibernate
4.3.8）。有一个 MySQL （版本5.1.73）和一个 Oracle （11g Express
Edition版本11.2.0.2.0-64位）数据库。

使用Hibernate Code Generation从数据库表中生成带有 JPA批注的 Java类。

问题在于，我们现在需要使用自动主键生成，而MySQL使用 GenerationType.IDENTITY， 而Oracle使用
GenerationType.SEQUENCE 。此外，在极少数情况下，我们需要能够自行手动设置主键。

带注释的类中的followig代码可为两个数据库自动生成密钥，但如果主键是自设置的，则失败。

@GeneratedValue(strategy=GenerationType.AUTO, generator="sequence_generator")
@SequenceGenerator(name="sequence_generator", sequenceName="SEQUENCE1")
@Column(name = "id", unique = true, nullable = false)
public Integer getId() {
    return this.id;
}

没有 @GeneratedValue 和 @SequenceGenerator 批注，可以手动设置主键，但是自动生成无效。

Answer 1

即使您使用时GenerationType.AUTO没有任何SEQUENCE特定参数，也将无法保存分配的标识符。

如果您愿意做出一些妥协，则有一些解决方法：

1. 一种方法是切换到分配的标识符。您可以使用适用于MySQL和Oracle的UUID标识符，也可以手动分配值。

2. 另一种方法是使用自定义表生成器。

首先，您定义一个可识别的接口：

    public interface Identifiable<T extends Serializable> {
    T getId();
}

然后扩展表生成器：

    public class AssignedTableGenerator extends TableGenerator {

    @Override
    public Serializable generate(SessionImplementor session, Object obj) {
        if(obj instanceof Identifiable) {
            Identifiable identifiable = (Identifiable) obj;
            Serializable id = identifiable.getId();
            if(id != null) {
                return id;
            }
        }
        return super.generate(session, obj);
    }
}

此生成器能够将分配的标识符与合成的生成的标识符混合：

    doInTransaction(session -> {
    for (int i = 0; i < 5; i++) {
        session.persist(new AssignTableSequenceIdentifier());
    }
    AssignTableSequenceIdentifier tableSequenceIdentifier = new AssignTableSequenceIdentifier();
    tableSequenceIdentifier.id = -1L;
    session.merge(tableSequenceIdentifier);
    session.flush();
});

生成以下语句：

    select tbl.next_val from sequence_table tbl where tbl.sequence_name=default for update
insert into sequence_table (sequence_name, next_val)  values (default,1)
update sequence_table set next_val=2  where next_val=1 and sequence_name=default
select tbl.next_val from sequence_table tbl where tbl.sequence_name=default for update
update sequence_table set next_val=3  where next_val=2 and sequence_name=default
select tbl.next_val from sequence_table tbl where tbl.sequence_name=default for update
update sequence_table set next_val=4  where next_val=3 and sequence_name=default
select tbl.next_val from sequence_table tbl where tbl.sequence_name=default for update
update sequence_table set next_val=5  where next_val=4 and sequence_name=default
select tbl.next_val from sequence_table tbl where tbl.sequence_name=default for update
update sequence_table set next_val=6  where next_val=5 and sequence_name=default
select identityvs0_.id as id1_0_0_ from assigneTableIdentifier identityvs0_ where identityvs0_.id=-1
insert into assigneTableIdentifier (id) values (1, 2)
insert into assigneTableIdentifier (id) values (2, 4)
insert into assigneTableIdentifier (id) values (5, -1)

对于Oracle，您可以结合SEQUENCE和分配的生成器，如本文所述。

简而言之，请考虑以下生成器：

public class AssignedSequenceStyleGenerator 
    extends SequenceStyleGenerator {

    @Override
    public Serializable generate(SessionImplementor session, 
        Object obj) {
        if(obj instanceof Identifiable) {
            Identifiable identifiable = (Identifiable) obj;
            Serializable id = identifiable.getId();
            if(id != null) {
                return id;
            }
        }
        return super.generate(session, obj);
    }
}

您可以将其映射到您的实体，如下所示：

@Id
@GenericGenerator(
    name = "assigned-sequence",
    strategy = "com.vladmihalcea.book.hpjp.hibernate.identifier.AssignedSequenceStyleGenerator",
    parameters = @org.hibernate.annotations.Parameter(
        name = "sequence_name", 
        value = "post_sequence"
    )
)
@GeneratedValue(
    generator = "assigned-sequence", 
    strategy = GenerationType.SEQUENCE
)
private Long id;

所有代码都可以在GitHub上使用，并且像一个超级按钮一样工作。