hibernate中自定义主键(集群)

在HIBERNATE中，identity ,sequence ,native 是数据局提供的主键生成方式,往往也不是我们需要,而且在程序跨数据库方面也体现出不足.还有基于算法的生成方式生成出来的主键基本都是字符串的.

　　我们现在需要一种生成方式:使用Long作为主键类型,需要自动增,支持集群.那么我们需要自定义一个我们的主键生成器才能实现了.

 主键生成器代码 类：

     package com.ta.jiang;
　　
　　import java.io.Serializable;
　　
　　import java.sql.Connection;
　　
　　import java.sql.PreparedStatement;
　　
　　import java.sql.ResultSet;
　　
　　import java.sql.SQLException;
　　
　　import java.util.Properties;
　　
　　import org.apache.commons.logging.Log;
　　
　　import org.apache.commons.logging.LogFactory;
　　
　　import org.hibernate.HibernateException;
　　
　　import org.hibernate.MappingException;
　　
　　import org.hibernate.dialect.Dialect;
　　
　　import org.hibernate.engine.SessionImplementor;
　　
　　import org.hibernate.id.Configurable;
　　
　　import org.hibernate.id.IdentifierGenerator;
　　
　　import org.hibernate.id.PersistentIdentifierGenerator;
　　
　　import org.hibernate.type.Type;
　　
　　public class IncrementGenerator implements IdentifierGenerator, Configurable {
　　
　　private static final Log log = LogFactory.getLog(IncrementGenerator.class);
　　
　　private Long next;
　　
　　private String sql;
　　
　　public Serializable generate(SessionImplementor session, Object object)
　　
　　throws HibernateException {
　　
　　if (sql!=null) {
　　
　　 getNext( session.connection() );
　　
　　}
　　
　　return next;
　　
　　}
　　
　　public void configure(Type type, Properties params, Dialect d) throws MappingException {
　　
　　String table = params.getProperty("table");
　　
　　if (table==null) table = params.getProperty(PersistentIdentifierGenerator.TABLE);
　　
　　 String column = params.getProperty("column");
　　
　　if (column==null) column = params.getProperty(PersistentIdentifierGenerator.PK);
　　
　　String schema = params.getProperty(PersistentIdentifierGenerator.SCHEMA);
　　
　　sql = "select max("+column +") from " + ( schema==null ? table : schema + '.' + table );
　　
　　log.info(sql);
　　
　　}
　　
　　 private void getNext(Connection conn) throws HibernateException {
　　
　　try {
　　
　　PreparedStatement st = conn.prepareStatement(sql);
　　
　　ResultSet rs = st.executeQuery();
　　
　　if ( rs.next() ) {
　　
　　next = rs.getLong(1) + 1;
　　
　　}
　　
　　else {
　　
　　next = 1l;
　　
　　}
　　
　　}catch(SQLException e)
　　
　　{
　　
　　throw new HibernateException(e);
　　
　　}
　　
　　finally {
　　
　　try{
　　
　　conn.close();
　　
　　}catch(SQLException e)
　　
　　{
　　
　　throw new HibernateException(e);
　　
　　}
　　
　　}
　　
　　}
　　
　　}

  代码二：

  Hibernate 配置文件：

  <id name="id" type="long" column="id" >
　　
　　<generator class="com.ta.jiang.IncrementGenerator" />
　　
　　</id>

评论

1 楼 soartju 2010-10-09  

貌似不支持集群，多个应用如果取到了某个时间的同一个id，岂不是就出错了？