C#给枚举加自定义特性

通常我们需要定义一组特定值。采用枚举再好不过了。它可以让我们很方便直观的管理一组固定的值。如果我们需要对应输出枚举值的汉语意思或者颜色样式等，我们可以这样实现：

首先定义一个枚举类

enum @enum
{
    Update = 1,
    Insert = 2
}

方式一（if语句）

if (@enum == 1)
{
    Console.Write("更新");
}
else (@enum == 2){
    Console.Write("新增");
}

方式二（switch语句）

switch(@enum)
{
    case 1:
        Console.Write("更新");
        break;
    case 2:
        Console.Write("新增");
        break;
}

方式三（采用字典）

Dictionary<int, string> dic = new Dictionary<int, string>
{
    [1] = "更新",
    [2] = "新增"
};
Console.Write(dic[@enum]);

现在我们用一种更加优雅的方式来实现，给枚举加特性

在System.ComponentModel命名空间下有一个特性Description，用来指定属性或事件的描述。

enum @enum
{
    [Description("修改")]
    Update = 1,
    [Description("新增")]
    Insert = 2
}

我们可以写一个扩展，用来获取Description特性，这用反射来读取Description的值，这个是复用性的

/// <summary>
/// 获取特性 (DescriptionAttribute) 的说明；如果未使用该特性，则返回枚举的名称。可指定的默认值。
/// </summary>
/// <param name="enum"></param>
/// <param name="def">默认值</param>
/// <returns></returns>
public static string Description(this Enum @enum, string def = "")
{
    Type enumType = @enum.GetType();
    int value = int.Parse(Enum.Format(enumType, Enum.Parse(enumType, @enum.ToString()), "d"));
    FieldInfo fieldInfo = enumType.GetField(Enum.GetName(enumType, value));
    if (fieldInfo.GetCustomAttribute(typeof(DescriptionAttribute), false) is DescriptionAttribute descriptionAttribute)
    {
        return descriptionAttribute.Description;
    }
    return def != "" ? def : @enum.ToString();
}

那么在使用的时候我们将可以很方便的获取枚举值对应的描述信息

@enum.Update.Description();

这样每次使用都只需要这一句代码便可调出枚举值的描述信息那么，如果还有其它需求怎么办，比方说，这个枚举值还需要有代表色，样式等用来渲染HTML的属性怎么办，这里就用到了自定义特性，我们先说说怎么自定义特性。其实很简单。
新建一个类，注意所有的特性类必须继承自 Attribute，代码如下：

/// <summary>
/// 指定属性的样式
/// </summary>
public class ClassAttribute : Attribute
{
    /// <summary>
    /// 样式
    /// </summary>
    public string Class { get; set; }
    /// <summary>
    /// 样式
    /// </summary>
    /// <param name="value"></param>
    public ClassAttribute(string value)
    {
        Class = value;
    }
}

这样一个自定义特性就建好了，那么调用的方法，和Description一样。但是如果每次添加一个自定义特性都得建立一个调取特性值的方法，是不是很累。我们要记住面向对象的三大特性之封装。我们再写一个扩展，这里用到了反射和泛型。

/// <summary>
/// 获取指定特性的值
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="enum"></param>
/// <param name="def">默认值</param>
/// <returns></returns>
public static String GetAttribute<T>(this Enum @enum, String def = "") where T : Attribute
{
    Type type = @enum.GetType();
    MemberInfo[] memInfo = type.GetMember(@enum.ToString());
    object[] attributes = memInfo[0].GetCustomAttributes(typeof(T), false);
    return (attributes.Length > 0) ? attributes[0].ToString() : def;
}

我们需要更改一下枚举

enum @enum
{
    [Description("修改"),Class("className")]
    Update = 1,
    [Description("新增"),Class("className")]
    Insert = 2
}

在使用的时候我们将可以很方便的获取到自定义特性的值

@enum.Update.GetAttribute<ClassAttribute>();

使用的时候注意引用关系。使用的时候根据实际需求做相应的调整。