2021-05-21 仓库温控系统(Winform) 04 获取特性值的扩展方法AttributeHelper

系统管理员 2023-01-23 02:52 8阅读 0赞

public static class AttributeHelper
    { 
        /// <summary>
        /// 获取映射表名
        /// </summary>
        /// <param name="type"></param>
        /// <returns></returns>
        public static string GetTName(this Type type)
        { 
            string tableName = string.Empty;
            //获取指定类型的自定义特性
            object[] attrs = type.GetCustomAttributes(false);
            foreach (var  attr in attrs)
            { 
                if(attr is TableAttribute)
                { 
                    TableAttribute tableAttribute = attr as TableAttribute;
                    tableName = tableAttribute.TableName;
                }
            }
            if(string.IsNullOrEmpty(tableName))
            { 
                tableName = type.Name;
            }
            return tableName;
        }
    
        /// <summary>
        /// 获取映射列名
        /// </summary>
        /// <param name="property"></param>
        /// <returns></returns>
        public static string GetColName(this PropertyInfo property)
        { 
            string columnName = string.Empty;
            //获取指定类型的自定义特性
            object[] attrs = property.GetCustomAttributes(false);
            foreach (var attr in attrs)
            { 
                if(attr is ColumnAttribute)
                { 
                    ColumnAttribute colAttr = attr as ColumnAttribute;
                    columnName = colAttr.ColumnName;
                }
            }
            if(string.IsNullOrEmpty(columnName))
            { 
                columnName = property.Name;
            }
            return columnName;
        }
    
        /// <summary>
        /// 判断主键是否自增
        /// </summary>
        /// <param name="type"></param>
        /// <returns></returns>
        public static bool IsIncrement(this Type type)
        { 
            //获取指定类型的自定义特性
            object[] attributes = type.GetCustomAttributes(false);
            foreach (var attr in attributes)
            { 
                if (attr is PrimaryKeyAttribute)
                { 
                    PrimaryKeyAttribute primaryKeyAttr = attr as PrimaryKeyAttribute;
                    return primaryKeyAttr.autoIncrement;
                }
            }
            return false;
        } 
    
        ///<summary> 
        /// 获取主键名 
        /// </summary> 
        /// <param name="type"></param> 
        /// <returns></returns> 
        public static string GetPrimary(this Type type)
        { 
            object[] attributes = type.GetCustomAttributes(false);
            foreach (var attr in attributes)
            { 
                if (attr is PrimaryKeyAttribute)
                { 
                    PrimaryKeyAttribute primaryKeyAttr = attr as PrimaryKeyAttribute;
                    return primaryKeyAttr.PrimaryKey;
                }
            }
            return null;
        }
    
        /// <summary> 
        /// 判断属性是否为主键 
        /// </summary> 
        /// <param name="type"></param> 
        /// <param name="property"></param> 
        /// <returns></returns> 
        public static bool IsPrimary(this Type type, PropertyInfo property)
        { 
            //获取主键名
            string primaryKeyName = type.GetPrimary();
            //获取指定属性的映射列名
            string columnName = property.GetColName();
            //判断是否相等
            return (primaryKeyName == columnName);
        }
    }