Hibernate—根据实体类得到对应的表名、主键名、字段名

时间:2022-09-15 13:48:53

import java.util.Iterator;

import org.hibernate.cfg.Configuration;
import org.hibernate.mapping.Column;
import org.hibernate.mapping.PersistentClass;
import org.hibernate.mapping.Property;

import com.iman.gdms.ruledata.po.IntelligenceNetVPNAndSCP;

/**
 * 功能描述:根据实体类得到对应的表名、主键名、字段名工具类
 * </p>
 * 注:po类名须与对应映射文件名一致,即Student.java与Student.hbm.xml
 *
 * @Date:Nov 10, 2008
 * @Time:3:13:07 PM
 *
 */
public class HibernateConfigurationHelper {

 private static Configuration hibernateConf;

 @SuppressWarnings("unused")
 private static Configuration getHibernateConf() {
  if (hibernateConf == null) {
   return new Configuration();
  }
  return hibernateConf;
 }

 @SuppressWarnings("unchecked")
 private static PersistentClass getPersistentClass(Class clazz) {
  synchronized (HibernateConfigurationHelper.class) {
   PersistentClass pc = getHibernateConf().getClassMapping(
     clazz.getName());
   if (pc == null) {
    hibernateConf = getHibernateConf().addClass(clazz);
    pc = getHibernateConf().getClassMapping(clazz.getName());

   }
   return pc;
  }
 }

 /**
  * 功能描述:获取实体对应的表名
  *
  * @param clazz
  *            实体类
  * @return 表名
  */
 @SuppressWarnings("unchecked")
 public static String getTableName(Class clazz) {
  return getPersistentClass(clazz).getTable().getName();
 }

 /**
  * 功能描述:获取实体对应表的主键字段名称
  *
  * @param clazz
  *            实体类
  * @return 主键字段名称
  */
 @SuppressWarnings("unchecked")
 public static String getPkColumnName(Class clazz) {

  return getPersistentClass(clazz).getTable().getPrimaryKey()
    .getColumn(0).getName();

 }

 /**
  * 功能描述:通过实体类和属性,获取实体类属性对应的表字段名称
  *
  * @param clazz
  *            实体类
  * @param propertyName
  *            属性名称
  * @return 字段名称
  */
 @SuppressWarnings("unchecked")
 public static String getColumnName(Class clazz, String propertyName) {
  PersistentClass persistentClass = getPersistentClass(clazz);
  Property property = persistentClass.getProperty(propertyName);
  Iterator it = property.getColumnIterator();
  if (it.hasNext()) {
   Column column = (Column) it.next();
   return column.getName();
  }
  return null;
 }

 /**
  * @param args
  */
 public static void main(String[] args) {
  System.out.println(getTableName(IntelligenceNetVPNAndSCP.class));
  System.out.println(getColumnName(IntelligenceNetVPNAndSCP.class,
    "vsCode"));
 }

}