Merge pull request '更新 Admin.NET/Admin.NET.Core/Extension/EnumExtension.cs' (#343) from coolcalf/Admin.NET.Pro:v2 into v2

Reviewed-on: https://code.adminnet.top/Admin.NET/Admin.NET.Pro/pulls/343
This commit is contained in:
zuohuaijun 2025-05-23 02:01:57 +08:00
commit fcd9b2e88f

View File

@ -1,239 +1,242 @@
// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。 // Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
// //
// 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。 // 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
// //
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任! // 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任!
namespace Admin.NET.Core; namespace Admin.NET.Core;
/// <summary> /// <summary>
/// 枚举拓展方法 /// 枚举拓展方法
/// </summary> /// </summary>
public static class EnumExtension public static class EnumExtension
{ {
// 枚举显示字典缓存 // 枚举显示字典缓存
private static readonly ConcurrentDictionary<Type, Dictionary<int, string>> EnumDisplayValueDict = new(); private static readonly ConcurrentDictionary<Type, Dictionary<int, string>> EnumDisplayValueDict = new();
// 枚举值字典缓存 // 枚举值字典缓存
private static readonly ConcurrentDictionary<Type, Dictionary<int, string>> EnumNameValueDict = new(); private static readonly ConcurrentDictionary<Type, Dictionary<int, string>> EnumNameValueDict = new();
// 枚举类型缓存 // 枚举类型缓存
private static ConcurrentDictionary<string, Type> _enumTypeDict; private static ConcurrentDictionary<string, Type> _enumTypeDict;
/// <summary> /// <summary>
/// 获取枚举对象Key与名称的字典缓存 /// 获取枚举对象Key与名称的字典缓存
/// </summary> /// </summary>
/// <param name="enumType"></param> /// <param name="enumType"></param>
/// <returns></returns> /// <returns></returns>
public static Dictionary<int, string> GetEnumDictionary(this Type enumType) public static Dictionary<int, string> GetEnumDictionary(this Type enumType)
{ {
if (!enumType.IsEnum) if (!enumType.IsEnum)
throw new ArgumentException("Type '" + enumType.Name + "' is not an enum."); throw new ArgumentException("Type '" + enumType.Name + "' is not an enum.");
// 查询缓存 // 查询缓存
var enumDic = EnumNameValueDict.TryGetValue(enumType, out var value) ? value : new Dictionary<int, string>(); var enumDic = EnumNameValueDict.TryGetValue(enumType, out var value) ? value : new Dictionary<int, string>();
if (enumDic.Count != 0) if (enumDic.Count != 0)
return enumDic; return enumDic;
// 取枚举类型的Key/Value字典集合 // 取枚举类型的Key/Value字典集合
enumDic = GetEnumDictionaryItems(enumType); enumDic = GetEnumDictionaryItems(enumType);
// 缓存 // 缓存
EnumNameValueDict[enumType] = enumDic; EnumNameValueDict[enumType] = enumDic;
return enumDic; return enumDic;
} }
/// <summary> /// <summary>
/// 获取枚举对象Key与名称的字典 /// 获取枚举对象Key与名称的字典
/// </summary> /// </summary>
/// <param name="enumType"></param> /// <param name="enumType"></param>
/// <returns></returns> /// <returns></returns>
private static Dictionary<int, string> GetEnumDictionaryItems(this Type enumType) private static Dictionary<int, string> GetEnumDictionaryItems(this Type enumType)
{ {
// 获取类型的字段,初始化一个有限长度的字典 // 获取类型的字段,初始化一个有限长度的字典
var enumFields = enumType.GetFields(BindingFlags.Public | BindingFlags.Static); var enumFields = enumType.GetFields(BindingFlags.Public | BindingFlags.Static);
Dictionary<int, string> enumDic = new(enumFields.Length); Dictionary<int, string> enumDic = new(enumFields.Length);
// 遍历字段数组获取key和name // 遍历字段数组获取key和name
foreach (var enumField in enumFields) foreach (var enumField in enumFields)
{ {
var intValue = (int)enumField.GetValue(enumType)!; var intValue = (int)enumField.GetValue(enumType)!;
enumDic[intValue] = enumField.Name; enumDic[intValue] = enumField.Name;
} }
return enumDic; return enumDic;
} }
/// <summary> /// <summary>
/// 获取枚举类型key与描述的字典缓存 /// 获取枚举类型key与描述的字典缓存
/// </summary> /// </summary>
/// <param name="enumType"></param> /// <param name="enumType"></param>
/// <returns></returns> /// <returns></returns>
/// <exception cref="Exception"></exception> /// <exception cref="Exception"></exception>
public static Dictionary<int, string> GetEnumDescDictionary(this Type enumType) public static Dictionary<int, string> GetEnumDescDictionary(this Type enumType)
{ {
if (!enumType.IsEnum) if (!enumType.IsEnum)
throw new ArgumentException("Type '" + enumType.Name + "' is not an enum."); throw new ArgumentException("Type '" + enumType.Name + "' is not an enum.");
// 查询缓存 // 查询缓存
var enumDic = EnumDisplayValueDict.TryGetValue(enumType, out var value) var enumDic = EnumDisplayValueDict.TryGetValue(enumType, out var value)
? value ? value
: new Dictionary<int, string>(); : new Dictionary<int, string>();
if (enumDic.Count != 0) if (enumDic.Count != 0)
return enumDic; return enumDic;
// 取枚举类型的Key/Value字典集合 // 取枚举类型的Key/Value字典集合
enumDic = GetEnumDescDictionaryItems(enumType); enumDic = GetEnumDescDictionaryItems(enumType);
// 缓存 // 缓存
EnumDisplayValueDict[enumType] = enumDic; EnumDisplayValueDict[enumType] = enumDic;
return enumDic; return enumDic;
} }
/// <summary> /// <summary>
/// 获取枚举类型key与描述的字典没有描述则获取name /// 获取枚举类型key与描述的字典没有描述则获取name
/// </summary> /// </summary>
/// <param name="enumType"></param> /// <param name="enumType"></param>
/// <returns></returns> /// <returns></returns>
/// <exception cref="Exception"></exception> /// <exception cref="Exception"></exception>
private static Dictionary<int, string> GetEnumDescDictionaryItems(this Type enumType) private static Dictionary<int, string> GetEnumDescDictionaryItems(this Type enumType)
{ {
// 获取类型的字段,初始化一个有限长度的字典 // 获取类型的字段,初始化一个有限长度的字典
var enumFields = enumType.GetFields(BindingFlags.Public | BindingFlags.Static); var enumFields = enumType.GetFields(BindingFlags.Public | BindingFlags.Static);
Dictionary<int, string> enumDic = new(enumFields.Length); Dictionary<int, string> enumDic = new(enumFields.Length);
// 遍历字段数组获取key和name // 遍历字段数组获取key和name
foreach (var enumField in enumFields) foreach (var enumField in enumFields)
{ {
var intValue = (int)enumField.GetValue(enumType)!; var intValue = (int)enumField.GetValue(enumType)!;
var desc = enumField.GetDescriptionValue<DescriptionAttribute>(); var desc = enumField.GetDescriptionValue<DescriptionAttribute>();
enumDic[intValue] = desc != null && !string.IsNullOrEmpty(desc.Description) ? desc.Description : enumField.Name; enumDic[intValue] = desc != null && !string.IsNullOrEmpty(desc.Description) ? desc.Description : enumField.Name;
} }
return enumDic; return enumDic;
} }
/// <summary> /// <summary>
/// 从程序集中查找指定枚举类型 /// 从程序集中查找指定枚举类型
/// </summary> /// </summary>
/// <param name="assembly"></param> /// <param name="assembly"></param>
/// <param name="typeName"></param> /// <param name="typeName"></param>
/// <returns></returns> /// <returns></returns>
public static Type TryToGetEnumType(Assembly assembly, string typeName) public static Type TryToGetEnumType(Assembly assembly, string typeName)
{ {
// 枚举缓存为空则重新加载枚举类型字典 // 枚举缓存为空则重新加载枚举类型字典
_enumTypeDict ??= LoadEnumTypeDict(assembly); _enumTypeDict ??= LoadEnumTypeDict(assembly);
// 按名称查找 // 按名称查找
return _enumTypeDict.TryGetValue(typeName, out var value) ? value : null; return _enumTypeDict.TryGetValue(typeName, out var value) ? value : null;
} }
/// <summary> /// <summary>
/// 从程序集中加载所有枚举类型 /// 从程序集中加载所有枚举类型
/// </summary> /// </summary>
/// <param name="assembly"></param> /// <param name="assembly"></param>
/// <returns></returns> /// <returns></returns>
private static ConcurrentDictionary<string, Type> LoadEnumTypeDict(Assembly assembly) private static ConcurrentDictionary<string, Type> LoadEnumTypeDict(Assembly assembly)
{ {
// 取程序集中所有类型 // 取程序集中所有类型
var typeArray = assembly.GetTypes(); var typeArray = assembly.GetTypes();
// 过滤非枚举类型,转成字典格式并返回 // 过滤非枚举类型,转成字典格式并返回
var dict = typeArray.Where(o => o.IsEnum).ToDictionary(o => o.Name, o => o); var dict = typeArray.Where(o => o.IsEnum).ToDictionary(o => o.Name, o => o);
ConcurrentDictionary<string, Type> enumTypeDict = new(dict); ConcurrentDictionary<string, Type> enumTypeDict = new(dict);
return enumTypeDict; return enumTypeDict;
} }
/// <summary> /// <summary>
/// 获取枚举的Description /// 获取枚举的Description
/// </summary> /// </summary>
/// <param name="value"></param> /// <param name="value"></param>
/// <returns></returns> /// <returns></returns>
public static string GetDescription(this Enum value) public static string GetDescription(this Enum value)
{ {
return value.GetType().GetField(value.ToString())?.GetCustomAttribute<DescriptionAttribute>()?.Description; if (value == null) return "";
} return value.GetType().GetField(value.ToString())?.GetCustomAttribute<DescriptionAttribute>()?.Description;
}
/// <summary>
/// 获取枚举的Description /// <summary>
/// </summary> /// 获取枚举的Description
/// <param name="value"></param> /// </summary>
/// <returns></returns> /// <param name="value"></param>
public static string GetDescription(this object value) /// <returns></returns>
{ public static string GetDescription(this object value)
return value.GetType().GetField(value.ToString()!)?.GetCustomAttribute<DescriptionAttribute>()?.Description; {
} if (value == null) return "";
return value.GetType().GetField(value.ToString()!)?.GetCustomAttribute<DescriptionAttribute>()?.Description;
/// <summary> }
/// 获取枚举的Theme
/// </summary> /// <summary>
/// <param name="value"></param> /// 获取枚举的Theme
/// <returns></returns> /// </summary>
public static string GetTheme(this object value) /// <param name="value"></param>
{ /// <returns></returns>
return value.GetType().GetField(value.ToString()!)?.GetCustomAttribute<ThemeAttribute>()?.Theme; public static string GetTheme(this object value)
} {
if (value == null) return "";
/// <summary> return value.GetType().GetField(value.ToString()!)?.GetCustomAttribute<ThemeAttribute>()?.Theme;
/// 将枚举转成枚举信息集合 }
/// </summary>
/// <param name="type"></param> /// <summary>
/// <returns></returns> /// 将枚举转成枚举信息集合
public static List<EnumEntity> EnumToList(this Type type) /// </summary>
{ /// <param name="type"></param>
if (!type.IsEnum) /// <returns></returns>
throw new ArgumentException("Type '" + type.Name + "' is not an enum."); public static List<EnumEntity> EnumToList(this Type type)
var arr = Enum.GetNames(type); {
return arr.Select(sl => if (!type.IsEnum)
{ throw new ArgumentException("Type '" + type.Name + "' is not an enum.");
var item = Enum.Parse(type, sl); var arr = Enum.GetNames(type);
return new EnumEntity return arr.Select(sl =>
{ {
Name = item.ToString(), var item = Enum.Parse(type, sl);
Describe = item.GetDescription() ?? item.ToString(), return new EnumEntity
Theme = item.GetTheme() ?? string.Empty, {
Value = item.GetHashCode() Name = item.ToString(),
}; Describe = item.GetDescription() ?? item.ToString(),
}).ToList(); Theme = item.GetTheme() ?? string.Empty,
} Value = item.GetHashCode()
};
/// <summary> }).ToList();
/// 枚举ToList }
/// </summary>
/// <typeparam name="T"></typeparam> /// <summary>
/// <param name="type"></param> /// 枚举ToList
/// <returns></returns> /// </summary>
public static List<T> EnumToList<T>(this Type type) /// <typeparam name="T"></typeparam>
{ /// <param name="type"></param>
if (!type.IsEnum) /// <returns></returns>
throw new ArgumentException("Type '" + type.Name + "' is not an enum."); public static List<T> EnumToList<T>(this Type type)
var arr = Enum.GetNames(type); {
return arr.Select(name => (T)Enum.Parse(type, name)).ToList(); if (!type.IsEnum)
} throw new ArgumentException("Type '" + type.Name + "' is not an enum.");
} var arr = Enum.GetNames(type);
return arr.Select(name => (T)Enum.Parse(type, name)).ToList();
/// <summary> }
/// 枚举实体 }
/// </summary>
public class EnumEntity /// <summary>
{ /// 枚举实体
/// <summary> /// </summary>
/// 枚举的描述 public class EnumEntity
/// </summary> {
public string Describe { get; set; } /// <summary>
/// 枚举的描述
/// <summary> /// </summary>
/// 枚举的样式 public string Describe { get; set; }
/// </summary>
public string Theme { get; set; } /// <summary>
/// 枚举的样式
/// <summary> /// </summary>
/// 枚举名称 public string Theme { get; set; }
/// </summary>
public string Name { get; set; } /// <summary>
/// 枚举名称
/// <summary> /// </summary>
/// 枚举对象的值 public string Name { get; set; }
/// </summary>
public int Value { get; set; } /// <summary>
/// 枚举对象的值
/// </summary>
public int Value { get; set; }
} }