前言

嗨，大家好！

在 C# 的世界里，枚舉（Enum）讓我們可以用更直觀的方式來表示一組相關(guān)的常量，比如 HTTP 方法、狀態(tài)碼等。

通過擴(kuò)展方法，我們可以為枚舉賦予更多的功能，使其在實(shí)際應(yīng)用中更加靈活。

今天，我們要分享的就是 4 個我在工作中常用的枚舉擴(kuò)展方法：獲取枚舉描述、描述轉(zhuǎn)枚舉、字符串轉(zhuǎn)枚舉和數(shù)字轉(zhuǎn)枚舉等。

這些方法在我的工作中幫助挺大的，希望對你也有所啟發(fā)，讓我們一起來看看吧！

枚舉擴(kuò)展方法

以下是完整的枚舉擴(kuò)展方法，留意代碼中的注釋：

public static class EnumUtil
{
    /// <summary>
    /// 獲取枚舉描述
    /// </summary>
    /// <param name="value"></param>
    /// <returns></returns>
    public static string ToDescription(this Enum value)
    {
        if (value == null) return "";

        System.Reflection.FieldInfo fieldInfo = value.GetType().GetField(value.ToString());

        object[] attribArray = fieldInfo.GetCustomAttributes(false);
        if (attribArray.Length == 0)
        {
            return value.ToString();
        }
        else
        {
            return (attribArray[0] as DescriptionAttribute).Description;
        }
    }

    /// <summary>
    /// 根據(jù)描述獲取枚舉值
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="description"></param>
    /// <returns></returns>
    public static T GetEnumByDescription<T>(this string description) where T : Enum
    {
        if (!string.IsNullOrWhiteSpace(description))
        {
            System.Reflection.FieldInfo[] fields = typeof(T).GetFields();
            foreach (System.Reflection.FieldInfo field in fields)
            {
                //獲取描述屬性
                object[] objs = field.GetCustomAttributes(typeof(DescriptionAttribute), false);    
                if (objs.Length > 0 && (objs[0] as DescriptionAttribute).Description == description)
                {
                    return (T)field.GetValue(null);
                }
            }
        }

        throw new ArgumentException(string.Format("{0} 未能找到對應(yīng)的枚舉.", description), "Description");
    }

    /// <summary>
    /// 數(shù)字轉(zhuǎn)枚舉
    /// </summary>
    /// <returns></returns>
    public static T ToEnum<T>(this int intValue) where T : Enum
    {
        return (T)Enum.ToObject(typeof(T), intValue);
    }

    /// <summary>
    /// 字符串轉(zhuǎn)枚舉
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="enumString"></param>
    /// <returns></returns>
    public static T ToEnum<T>(this string enumString) where T : Enum
    {
        return (T)Enum.Parse(typeof(T), enumString);
    }
}

使用

1. 創(chuàng)建枚舉

首先，我們來定義一個枚舉，表示 HTTP 方法類型：

public enum HttpMethodType
{
    [Description("查詢數(shù)據(jù)")]
    GET = 1,
    [Description("保存數(shù)據(jù)")]
    POST = 2,
    [Description("更新數(shù)據(jù)")]
    PUT = 3,
    [Description("刪除數(shù)據(jù)")]
    DELETE = 4
}

2. 使用擴(kuò)展方法

接下來，我們在 Program.cs 文件里使用這些擴(kuò)展方法：

using EumnSample;

// 1. 獲取枚舉的描述
HttpMethodType httpMethodType = HttpMethodType.GET;
string getMethodDesc = httpMethodType.ToDescription();
Console.WriteLine(getMethodDesc);

// 2. 根據(jù)描述獲取相應(yīng)的枚舉
string postMethodDesc = "保存數(shù)據(jù)";
HttpMethodType postMethodType = postMethodDesc.GetEnumByDescription<HttpMethodType>();
Console.WriteLine(postMethodType.ToString());

// 3. 根據(jù)枚舉名稱獲取相應(yīng)的枚舉
string putMethodStr = "PUT";
HttpMethodType putMethodType = putMethodStr.ToEnum<HttpMethodType>();
Console.WriteLine(putMethodType.ToString());

// 4. 根據(jù)枚舉的值獲取相應(yīng)的枚舉
int delMethodValue = 4;
HttpMethodType delMethodType = delMethodValue.ToEnum<HttpMethodType>();
Console.WriteLine(delMethodType.ToString());