1. 程式人生 > >Hibernate自定義主鍵生成器

Hibernate自定義主鍵生成器

1:Entity實體配置
 /**
 * Test.java
 * 建立:劉彥亮 2015年4月8日
 */
package com.saving.ecm.resource.entity;

import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;
import javax.persistence.Table;
import javax.xml.bind.annotation.XmlRootElement;

import org.hibernate.annotations.GenericGenerator;
import org.hibernate.annotations.Parameter;

/**
 * @author lyl
 * @version 1.0
 */
@XmlRootElement
@Entity
@Table(name = "test")
public class Test {
	@Id
	@GeneratedValue(generator = "idGenerator")
	@GenericGenerator(name = "idGenerator", strategy = "com.saving.ecm.resource.entity.KeyUtils",
	parameters = { @Parameter(name = "dataCenterID", value = "123456") ,@Parameter(name = "idLength", value = "10")})
	private String code;

	@Column(name = "name", nullable = false, columnDefinition = "varchar(48) default '' comment '名稱'")
	private String name;

	public String getCode() {
		return code;
	}

	public void setCode(String code) {
		this.code = code;
	}

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}


}
2:自定義主鍵生成器
/**
 * AssignedSequenceGenerator.java
 * 建立:劉彥亮 2015年4月8日
 */
package com.saving.ecm.resource.entity;

import java.io.Serializable;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Properties;

import org.hibernate.dialect.Dialect;
import org.hibernate.engine.spi.SessionImplementor;
import org.hibernate.id.AbstractUUIDGenerator;
import org.hibernate.id.Configurable;
import org.hibernate.type.Type;

/**
 * @author lyl
 * @version 1.0
 */
public class KeyUtils extends AbstractUUIDGenerator implements Configurable {

	public String dataCenterID;
	public String idLength;

	public void configure(Type type, Properties params, Dialect d) {
		this.dataCenterID = params.getProperty("dataCenterID");
		this.idLength = params.getProperty("idLength");
	}

	public Serializable generate(SessionImplementor session, Object obj) {
		// Date today = new Date();
		// SimpleDateFormat formatDate = new SimpleDateFormat("yyyyMMddhhmmss");
		// String time = formatDate.format(today);
		return dataCenterID + "|" + idLength;
	}

}
3:執行save
public void insertTest() throws Throwable {
		IPersistAction action = null;
		try {
			Test t = new Test();
			t.setName("dasdasd");
			TestDAO tDao = (TestDAO) TransactionManagerFactory.getDAO(TestDAO.class);
			action = PersistenceFactory.createPersistenceAction(PersistenceFactory.PersistenceUnitNative);
			action.transact();
			tDao.saveEntity(action.getEntityManager(), t);
			action.flush();
		} catch (Exception ex) {
			ex.printStackTrace();
			if (action != null)
				action.rollback();
		} finally {
			if (action != null)
				action.close();
		}

	}
4:檢視結果