泛型,枚举和自定义属性 - 是否可能?

时间:2013-08-08 03:30:39

标签: c# generics

对代码量表示歉意,但更容易以这种方式解释。

我有一个自定义属性CustomUserData实现如下:

public class CustomUserData : Attribute
{
   public CustomUserData(object aUserData)
   {
      UserData = aUserData;
   }
   public object UserData { get; set; }
}

和枚举的扩展方法为:

public static class EnumExtensions
{
    public static TAttribute GetAttribute<TAttribute>(this Enum aValue) where TAttribute : Attribute
    {
        Type type = aValue.GetType();
        string name = Enum.GetName(type, aValue);
        return type.GetField(name)
                   .GetCustomAttributes(false)
                   .OfType<TAttribute>()
                   .SingleOrDefault();
    }

    public static object GetCustomUserData(this Enum aValue)
    {
        CustomUserData userValue = GetAttribute<CustomUserData>(aValue);
        return userValue != null ? userValue.UserData : null;
    }
}

然后我有一个帮助器类,它序列化/反序列化具有与之关联的自定义数据的枚举,如下所示:

public static class ParameterDisplayModeEnumListHelper
{
    public static List<ParameterDisplayModeEnum> FromDatabase(string aDisplayModeString)
    {
        //Default behaviour
        List<ParameterDisplayModeEnum> result = new List<ParameterDisplayModeEnum>();

        //Split the string list into a list of strings
        List<string> listOfDisplayModes = new List<string>(aDisplayModeString.Split(','));

        //Iterate the enum looking for matches in the list
        foreach (ParameterDisplayModeEnum displayModeEnum in Enum.GetValues(typeof (ParameterDisplayModeEnum)))
        {
            if (listOfDisplayModes.FindIndex(item => item == (string)displayModeEnum.GetCustomUserData()) >= 0)
            {
                result.Add(displayModeEnum);
            }
        }

        return result;
    }

    public static string ToDatabase(List<ParameterDisplayModeEnum> aDisplayModeList)
    {
        string result = string.Empty;

        foreach (ParameterDisplayModeEnum listItem in aDisplayModeList)
        {
            if (result != string.Empty)
                result += ",";
            result += listItem.GetCustomUserData();
        }

        return result;
    }
}

然而具体 ParameterDisplayModeEnum ,我有一堆枚举,我需要这样处理序列化/反序列化,所以我宁愿有一个通用的为:

public static class EnumListHelper<TEnum>
{
    public static List<TEnum> FromDatabase(string aDisplayModeString)
    {
        //Default behaviour
        List<TEnum> result = new List<TEnum>();

        //Split the string list into a list of strings
        List<string> listOfDisplayModes = new List<string>(aDisplayModeString.Split(','));

        //Iterate the enum looking for matches in the list
        foreach (TEnum displayModeEnum in Enum.GetValues(typeof (TEnum)))
        {
            if (listOfDisplayModes.FindIndex(item => item == (string)displayModeEnum.GetCustomUserData()) >= 0)
            {
                result.Add(displayModeEnum);
            }
        }

        return result;
    }

    public static string ToDatabase(List<TEnum> aDisplayModeList)
    {
        string result = string.Empty;

        foreach (TEnum listItem in aDisplayModeList)
        {
            if (result != string.Empty)
                result += ",";
            result += listItem.GetCustomUserData();
        }

        return result;
    }
}

但是这不起作用,因为无法调用GetCustomUserData()。有什么建议?我无法更改自定义属性的使用或枚举的使用。我正在寻找一种通用的方法来进行序列化/反序列化,而不必每次都编写具体的列表助手类。

所有建议都表示赞赏。

2 个答案:

答案 0 :(得分:2)

理想情况下,您希望将TEnum限制为枚举,但由于您无法将泛型限制为枚举Microsoft,因此无法工作。但请尝试以下操作,它可能会起作用。 ..

if (listOfDisplayModes.FindIndex(item => 
  item == (string)(displayModeEnum as Enum).GetCustomUserData()) >= 0)

答案 1 :(得分:2)

试试这段代码:

public static class EnumListHelper
{
    private static void EnsureIsEnum<TEnum>()
    {
        if (!typeof(TEnum).IsEnum)
            throw new InvalidOperationException(string.Format("The {0} type is not an enum.", typeof(TEnum)));
    }

    public static List<TEnum> FromDatabase<TEnum>(string aDisplayModeString)
        where TEnum : struct
    {
        EnsureIsEnum<TEnum>();
        //Default behaviour
        List<TEnum> result = new List<TEnum>();

        //Split the string list into a list of strings
        List<string> listOfDisplayModes = new List<string>(aDisplayModeString.Split(','));

        //Iterate the enum looking for matches in the list
        foreach (Enum displayModeEnum in Enum.GetValues(typeof(TEnum)))
        {
            if (listOfDisplayModes.FindIndex(item => item == (string)displayModeEnum.GetCustomUserData()) >= 0)
            {
                result.Add((TEnum)(object)displayModeEnum);
            }
        }

        return result;
    }

    public static string ToDatabase<TEnum>(List<TEnum> aDisplayModeList)
        where TEnum : struct
    {
        EnsureIsEnum<TEnum>();
        string result = string.Empty;

        foreach (var listItem in aDisplayModeList.OfType<Enum>())
        {
            if (result != string.Empty)
                result += ",";
            result += listItem.GetCustomUserData();
        }

        return result;
    }
}

var fromDatabase = EnumListHelper.FromDatabase<TestEnum>("test");
EnumListHelper.ToDatabase(fromDatabase);

更新0

要清楚,因为我们不能将泛型限制为Enum,我们应该检查类型TEnum是枚举,如果不是则抛出异常。 当我们使用FromDatabase方法时,我们知道TEnum是枚举,我们可以编写此代码以将枚举转换为指定的TEnum

result.Add((TEnum)(object)displayModeEnum)

ToDatabase方法中,我们也知道TEnum是枚举,我们可以编写此代码以将TEnum转换为Enum类型:

aDisplayModeList.OfType<Enum>()