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

Question

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

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

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

注释类中的后续代码适用于两个数据库的自动密钥生成，但如果主键是自设置的，则会失败。

@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

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

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

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

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

首先定义一个 Identifiable 接口：

    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 上找到，就像一个魅力。

Answer 2

试试这样的：

@Id
@Column( name = "ID" )
@TableGenerator( 
        name = "AppSeqStore", 
        table = "APP_SEQ_STORE", 
        pkColumnName = "APP_SEQ_NAME", 
        pkColumnValue = "LISTENER_PK", 
        valueColumnName = "APP_SEQ_VALUE", 
        initialValue = 1, 
        allocationSize = 1 )
@GeneratedValue( strategy = GenerationType.TABLE, generator = "AppSeqStore" )

以及数据库中的这张表：

CREATE TABLE APP_SEQ_STORE (
    APP_SEQ_NAME VARCHAR(255) NOT NULL,
    APP_SEQ_VALUE NUMBER(10) NOT NULL,
    PRIMARY KEY(APP_SEQ_NAME)
)

INSERT INTO APP_SEQ_STORE VALUES ('LISTENER_PK', 0)

这一切都可以在 Oracle、MS Sql Server 和 MySql 中使用 JBoss 作为 App Server。

更多信息在这里： http://www.developerscrappad.com/408/java/java-ee/ejb3-jpa-3-ways-of-generating-primary-key-through-generatedvalue/