JPA中的自定义主键生成器



我使用org.hibernate.id.IdentifierGenerator生成主键列,如下所示。在下面的给定示例中,目前它只是按顺序递增INT(11)(MySQL)类型的键,即它与MySQL中的auto_increment类似,但它可以用于生成任何自定义模式的值,如E0001、E0002、E0003。。。e010。。。e100。。。E1000。。。E12345。。。

import java.io.Serializable;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import org.hibernate.HibernateException;
import org.hibernate.engine.spi.SessionImplementor;
import org.hibernate.id.IdentifierGenerator;
public final class TestIdGenerator implements IdentifierGenerator {
    @Override
    public Serializable generate(SessionImplementor session, Object object) throws HibernateException {
        try {
            Connection connection = session.connection();
            PreparedStatement ps = connection.prepareStatement("SELECT MAX(id) AS id FROM test");
            ResultSet rs = ps.executeQuery();
            if (rs.next()) {
                int id = rs.getInt("id");
                return id <= 0 ? 1 : id + 1;
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return null;
    }
}

使用上述id生成器的实体Test

public class Test implements Serializable {
    @Id
    @GenericGenerator(name = "test_sequence", strategy = "com.example.TestIdGenerator")
    @GeneratedValue(generator = "test_sequence")
    @Basic(optional = false)
    @Column(name = "id", nullable = false)
    private Integer id;
    //...
}

然而,这是Hibernate特有的特性。有没有一种方法可以使it提供者不可知,即JPA(2.1)中是否有这样的功能?

我使用的是Hibernate 5.1.0最终版本JPA 2.1。


这种方法在持久化、合并和删除实体时是否会导致一些并发问题,并需要某种锁定?

你不能像你希望的那样用注释来完成它。您可以使用@PrePersist挂钩。

@PrePersist
public void ensureId() {
    id = ...
}

也就是说,您可能无法从@PrePersist方法访问DB。。。JPA2规范(JSR317)规定了以下内容:

通常,可移植应用程序的生命周期方法不应在同一持久性上下文中调用EntityManager或Query操作、访问其他实体实例或修改关系。

就实现而言,Hibernate明确禁止它:

回调方法不能调用EntityManager或Query方法!

我的建议是:如果需要指定类型,请使用序列并有一个单独的列。此外,您还可以随时修改SELECT以收回您选择的模式:

select 'E' || id as id from schema.table;

或者你可以使用@PostLoad/@PostPersist来撤回一个瞬态字段"formattedId":

@PostLoad
@PostPersist
private void setFormattedId() {
    this.formattedId = "E" + id;
}

相关内容

  • 没有找到相关文章

最新更新