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

2022-09-03 13:40:07

我正在开发一个Java应用程序,它应该在具有相同数据库模式的两个不同数据库上进行CRUD操作(使用Hibernate 4.3.8)。有一个MySQL(版本5.1.73)和一个Oracle(11g Express Edition Release 11.2.0.2.0 - 64bit)数据库。

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

问题是我们现在需要使用自动主键生成,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注释,则可以手动设置主键,但自动生成不起作用。


答案 1

即使不使用任何特定于 SEQUENCE 的参数,也无法保存分配的标识符。GenerationType.AUTO

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

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

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


答案 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/


推荐