背景:
Hibernate(目前使用的版本是3.2)中提供了多種生成主鍵的方式.在下面的文章中有列出來
[hibernate]Hibernate主鍵生成方式 Key Generator
然而當前的這么多種生成方式未必能滿足我們的要求.
比如
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>
ps:此生成方式僅通過兩個hibernate實例測試,如發現有問題,請留言.
Let life be beautiful like summer flowers and death like autumn leaves.