【Hibernate】Hibernate中查询表名、字段名以及字段类型等信息

谁借莪１个温暖的怀抱￠ 2022-08-21 10:56 294阅读 0赞

Hibernate中查询表名、字段名以及字段类型等信息的工具类

package com.lmb.ui.util;
    
    import org.hibernate.cfg.Configuration;
    import org.hibernate.mapping.Column;
    import org.hibernate.mapping.PersistentClass;
    import org.hibernate.mapping.Property;
    
    import java.util.Iterator;
    
    /**
     * 功能描述：根据实体类得到对应的表名、主键名、字段名工具类
     */
    public class HibernateConfigurationHelper {
        
    
        private static Configuration hibernateConf;
    
        private static Configuration getHibernateConf() {
            if (hibernateConf == null) {
                return new Configuration();
            }
            return hibernateConf;
        }
    
        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 表名
         */
        public static String getTableName(Class clazz) {
            return getPersistentClass(clazz).getTable().getName();
        }
    
        /**
         * 功能描述：获取实体对应表的主键字段名称
         *
         * @param clazz 实体类
         * @return 主键字段名称
         */
        public static String getPkColumnName(Class clazz) {
    
            return getPersistentClass(clazz).getTable().getPrimaryKey()
                    .getColumn(0).getName();
    
        }
    
        /**
         * 功能描述：通过实体类和属性，获取实体类属性对应的表字段名称
         *
         * @param clazz        实体类
         * @param propertyName 属性名称
         * @return 字段名称
         */
        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 clazz        实体类
         * @param propertyName 属性名称
         * @return 字段名称
         */
        public static String getColumnType(Class clazz, String propertyName) {
            PersistentClass persistentClass = getPersistentClass(clazz);
            Property property = persistentClass.getProperty(propertyName);
            String type = property.getType().getName();
            if (type.equals("timestamp")) {
                return "java.sql." + type.substring(0, 1).toUpperCase() + type.substring(1);
            } else {
                return "java.lang." + type.substring(0, 1).toUpperCase() + type.substring(1);
            }
    
        }
    
        /**
         * 获得实体类的主键属性名称，如果没有返回null
         *
         * @param clazz 实体类的class对象
         * @return 实体类主键属性名称
         */
        public static String getPrimaryKeyName(Class clazz) {
            PersistentClass persistentClass = getPersistentClass(clazz);
            Property property = persistentClass.getIdentifierProperty();
            if (property != null) {
                return property.getName();
            }
            return null;
        }
    }