ds8-solution-pro/ds-wms-service/DS.Module.Core/Utils/EnumUtil.cs

using Fasterflect;
using System.Collections.Concurrent;
using System.ComponentModel;
using System.Reflection;

namespace DS.Module.Core;

/// <summary>
/// 枚举工具类
/// </summary>
public static class EnumUtil
{
    // 枚举显示字典缓存
    private static readonly ConcurrentDictionary<Type, Dictionary<string, string>> EnumShowNameDict = new ConcurrentDictionary<Type, Dictionary<string, string>>();

    /// <summary>
    /// 字符串转Enum
    /// </summary>
    /// <typeparam name="T">枚举</typeparam>
    /// <param name="str">字符串</param>
    /// <returns>转换的枚举</returns>
    public static T ToEnum<T>(this string str)
    {
        try
        {
            return (T)Enum.Parse(typeof(T), str);
        }
        catch
        {
            return default(T);
        }
    }

    /// <summary>
    /// 获取到对应枚举的描述-没有描述信息，返回枚举值
    /// </summary>
    /// <param name="enum"></param>
    /// <returns></returns>
    public static string EnumDescription(this Enum @enum)
    {
        Type type = @enum.GetType();
        string name = Enum.GetName(type, @enum);
        if (name == null)
        {
            return null;
        }
        FieldInfo field = type.GetField(name);
        if (!(Attribute.GetCustomAttribute(field, typeof(DescriptionAttribute)) is DescriptionAttribute attribute))
        {
            return name;
        }
        return attribute?.Description;
    }

    public static int ToEnumInt(this Enum e)
    {
        try
        {
            return e.GetHashCode();
        }
        catch (Exception)
        {
            return 0;
        }
    }


    /// <summary>
    /// 获取枚举对象Key与名称的字典（缓存）
    /// </summary>
    /// <param name="enumType"></param>
    /// <returns></returns>
    public static Dictionary<string, string> GetEnumDictionaryWithKey(this Type enumType)
    {
        if (!enumType.IsEnum)
            throw new ArgumentException("Type '" + enumType.Name + "' is not an enum.");

        // 查询缓存
        var enumDic = EnumShowNameDict.ContainsKey(enumType) ? EnumShowNameDict[enumType] : new Dictionary<string, string>();
        if (enumDic.Count != 0)
            return enumDic;

        string[] fieldstrs = Enum.GetNames(enumType);
        // 取枚举类型的Key/Value字典集合
        enumDic = fieldstrs.Select(item =>
        {
            object[] arr = enumType.GetField(item).GetCustomAttributes(typeof(DescriptionAttribute), true);

            return new
            {
                Key = item,
                Val = ((DescriptionAttribute)arr[0]).Description
            };
        }).ToDictionary(a => a.Key, b => b.Val);

        // 缓存
        EnumShowNameDict[enumType] = enumDic;

        return enumDic;
    }

    /// <summary>
    /// 取得枚举值数
    /// </summary>
    /// <param name="EnumObject">枚举对象</param>
    /// <returns>枚举值数</returns>
    public static int GetEnumValue(this System.Enum EnumObject)
    {
        int intValue = (int)System.Enum.Parse(EnumObject.GetType(), EnumObject.ToString(), true);
        return intValue;
    }
    /// <summary>
    /// 
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <returns></returns>
    /// <exception cref="ArgumentException"></exception>
    public static Dictionary<string, int> EnumToDictionary<T>()
    {
        if (!typeof(T).IsEnum)
        {
            throw new ArgumentException("T must be an enumerated type");
        }
        var enumDictionary = new Dictionary<string, int>();
        Array enumValues = Enum.GetValues(typeof(T));
        var enumNames = Enum.GetNames(typeof(T));
        foreach (System.Enum value in enumValues)
        {
            enumDictionary.Add(EnumDescription(value), GetEnumValue(value));
        }

        //for (int i = 0; i < enumValues.Length; i++)
        //{
        //    enumDictionary.Add(EnumDescription(enumNames[i]), (int)enumValues.GetValue(i));
        //}
        return enumDictionary;
    }
}