(更新时间)2021年5月19日 仓库温控系统(Winform) 04 获取特性值的扩展方法AttributeHelper

更新时间)2021年5月19日 仓库温控系统(Winform) 04 获取特性值的扩展方法AttributeHelper"/>

(更新时间)2021年5月19日 仓库温控系统(Winform) 04 获取特性值的扩展方法AttributeHelper

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);}
}