(更新时间)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);
    }
}

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
  
23
  
24
  
25
  
26
  
27
  
28
  
29
  
30
  
31
  
32
  
33
  
34
  
35
  
36
  
37
  
38
  
39
  
40
  
41
  
42
  
43
  
44
  
45
  
46
  
47
  
48
  
49
  
50
  
51
  
52
  
53
  
54
  
55
  
56
  
57
  
58
  
59
  
60
  
61
  
62
  
63
  
64
  
65
  
66
  
67
  
68
  
69
  
70
  
71
  
72
  
73
  
74
  
75
  
76
  
77
  
78
  
79
  
80
  
81
  
82
  
83
  
84
  
85
  
86
  
87
  
88
  
89
  
90
  
91
  
92
  
93
  
94
  
95
  
96
  
97
  
98
  
99
  
100
  
101
  
102
  
103
  
104
  
105
  
106
  
107
 

文章来源: codeboy.blog.csdn.net，作者：愚公搬代码，版权归原作者所有，如需转载，请联系作者。

原文链接：codeboy.blog.csdn.net/article/details/117046610