开源 - Ideal库 - 枚举扩展计划思路及实现难点（三） - ToB企服应用市场:ToB评测及商务社交产业平台

//根据枚举名称转换成枚举，转换失败则返回空
public static TEnum? ToEnumByName<TEnum>(this string name)
where TEnum : struct, Enum
{
//转换成功则返回结果，否则返回空
if (Enum.TryParse<TEnum>(name, out var result))
{
//检查是否为有效的枚举名称字符串，
if (Enum.IsDefined(typeof(TEnum), name))
{
//返回枚举
return result;
}
else
{
//计算是否为有效的位标志组合项
var isValidFlags = IsValidFlagsMask<ulong, TEnum>(result.ToEnumValue<ulong>());
//如果是有效的位标志组合项则返回枚举，否则返回空
return isValidFlags ? result : default(TEnum?);
}
}
//返回空
return default;
}

复制代码

//根据枚举描述转换成枚举，转换失败返回空
public static TEnum? ToEnumByDesc<TEnum>(this string description)
where TEnum : struct, Enum
{
var type = typeof(TEnum);
var info = GetEnumTypeInfo(type);
//不是位标志枚举的情况处理
if (!info.IsFlags)
{
return ToEnumDesc<TEnum>(description);
}
//是位标志枚举的情况处理
//不是组合位的情况,本身可能就包含[,]
var tenum = ToEnumDesc<TEnum>(description);
if (tenum.HasValue)
{
return tenum;
}
//如果不包含[,]，则直接返回
if (!description.Contains(','))
{
return default;
}
//是组合位的情况
var names = description.Split(',');
var values = Enum.GetValues(type);
//记录有效枚举描述个数
var count = 0;
ulong mask = 0L;
//变量枚举所有项
foreach (var name in names)
{
foreach (Enum value in values)
{
//取枚举项描述与目标描述相比较，相同则返回该枚举项
if (value.ToEnumDesc() == name)
{
//有效枚举个数加1
count++;
//将枚举值转为long类型
var valueLong = Convert.ToUInt64(value);
// 过滤掉负数或无效的值，规范的位标志枚举应该都为非负数
if (valueLong >= 0)
{
//合并枚举值至mask
mask |= valueLong;
}
break;
}
}
}
//如果两者不相等，说明描述字符串不是一个有效组合项
if (count != names.Length)
{
return default;
}
var underlyingType = Enum.GetUnderlyingType(type);
if (underlyingType == typeof(byte))
{
return ((byte)mask).ToEnumByValue<TEnum>();
}
else if (underlyingType == typeof(sbyte))
{
return ((sbyte)mask).ToEnumByValue<TEnum>();
}
else if (underlyingType == typeof(short))
{
return ((short)mask).ToEnumByValue<TEnum>();
}
else if (underlyingType == typeof(ushort))
{
return ((ushort)mask).ToEnumByValue<TEnum>();
}
else if (underlyingType == typeof(int))
{
return ((int)mask).ToEnumByValue<TEnum>();
}
else if (underlyingType == typeof(uint))
{
return ((uint)mask).ToEnumByValue<TEnum>();
}
else if (underlyingType == typeof(long))
{
return ((long)mask).ToEnumByValue<TEnum>();
}
else if (underlyingType == typeof(ulong))
{
return mask.ToEnumByValue<TEnum>();
}
return default;
}

复制代码

public static partial class EnumExtension
{
//枚举类型基础信息
private sealed class EnumTypeInfo
{
//是否是位标志
public bool IsFlags { get; set; }
//枚举掩码
public ulong Mask { get; set; }
//枚举项集合
public List<Enum> Items { get; set; } = [];
}
//存储枚举项对应的描述
private static readonly ConcurrentDictionary<Enum, string> _descs = new();
//存储枚举相关信息

复制代码

//获取枚举名称+枚举描述
public static Dictionary<string, string> ToEnumNameDescs(this Type type)
{
//根据type获取枚举类型基础信息
var info = GetEnumTypeInfo(type);
//通过枚举项集合转为目标类型
return info.Items.ToDictionary(r => r.ToString(), r => r.ToEnumDesc());
}

复制代码