hibernate中自定義主鍵生成器

Hibernate(目前使用的版本是3.2)中提供了多種生成主鍵的方式。

然而當前的這麼多種生成方式未必能滿足我們的要求.

比如increment,可以在一個hibernate實例的應用上很方便的時候,但是在集群的時候就不行了.

再如 identity ,sequence ,native 是數據局提供的主鍵生成方式,往往也不是我們需要,而且在程序跨數據庫方面也體現出不足.

還有基於算法的生成方式生成出來的主鍵基本都是字符串的.

我們現在需要一種生成方式:使用Long作為主鍵類型,自動增,支持集群.

那麼我們需要自定義一個我們的主鍵生成器才能實現了.

實現代碼:

package hibernate;

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);

}

}

}

}

配置:

在對應的hbm文件裡面將id的配置如下:

<id name="id" type="long" column="id" >

<generator class="hibernate.IncrementGenerator" />

</id>