时间戳

首页 > 解决方案 > 如何使用 jpa 和 hibernate 为不同的子类实体生成 id

java - 如何使用 jpa 和 hibernate 为不同的子类实体生成 id

问题描述

我有以下 AbstractCompany 超类实体：

@Entity
@Inheritance(strategy = InheritanceType.JOINED)
public class AbstractCompany {
@Id
    @GenericGenerator(name = "rju_id", strategy = "com.ivc.talonsystem.util.keygens.RjuIdGenerator")
    @GeneratedValue(generator = "rju_id")  
    @Column(name = "id")
    private Integer id;
// other properties and getters-setters for them
}

和子类 Rju：

@Entity
@Table(name="rju")
public class Rju extends AbstractCompany implements Serializable {
    @NotEmpty
    @Column(name = "fullname")
    private String namerju;
    @NotEmpty
    @Column(name = "briefname")
    private String briefname;
    // other properties and getters-setters for them
}

和用于为 Rju 实体生成 id 的 RjuIdGenerator 类：

public class RjuIdGenerator implements IdentifierGenerator {

    @Override
    public Serializable generate(SessionImplementor session, Object object) throws HibernateException {
        Connection connection = session.connection();
        try {
            Statement statement=connection.createStatement();

            ResultSet rs=statement.executeQuery("select max(id) as Id from Rju");

            if(rs.next())
            {
                int id=rs.getInt(1)+1;
                Integer generatedId = new Integer(id);
                return generatedId;
            }
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        return null;
    }

}

这个 RjuIdGenerator 类为所有超类和子类生成 id。如何为特定子实体添加另一个生成器类？提前致谢！！！

标签： javahibernatejpa

解决方案

伙计们，我找到了解决这个问题的方法。我所做的只是检查 RjuIdGenerator 的 generate 方法中的对象实例，然后使用适当的表查询获取特定类的生成 id：

public class RjuIdGenerator implements IdentifierGenerator {

    @Override
    public Serializable generate(SessionImplementor session, Object object) throws HibernateException {
        Connection connection = session.connection();
        try {
            Statement statement=connection.createStatement();
            if (object.getClass().getName()==Rju.class.getName()) {
                ResultSet rs=statement.executeQuery("select max(id) as Id from Rju");

                if(rs.next())
                {
                    int id=rs.getInt(1)+1;
                    Integer generatedId = new Integer(id);
                    return generatedId;
                }
            } else {
                ResultSet rs=statement.executeQuery("select max(id) as Id from AbstractCompany");

                if(rs.next())
                {
                    int id=rs.getInt(1)+100000;
                    Integer generatedId = new Integer(id);
                    return generatedId;
                }
            }

        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return null;
    }

}

推荐阅读