簡體   English   中英

在枚舉擴展幫助器中返回枚舉列表而不是字符串

[英]Return enum list instead of string in enum extension helper

如何轉換GetWithOrder方法以正確的順序返回實際的枚舉? 它適用於枚舉的字符串,但是我想要實際的Enum值,當然它也需要是通用的。

[AttributeUsage(AttributeTargets.Field, AllowMultiple = false)]
public class EnumOrderAttribute : Attribute
{
    public int Order { get; set; }
}


public static class EnumExtenstions
{
    public static IEnumerable<string> GetWithOrder(this Enum enumVal)
    {
        return enumVal.GetType().GetWithOrder();
    }

    public static IEnumerable<string> GetWithOrder(this Type type)
    {
        if (!type.IsEnum)
        {
            throw new ArgumentException("Type must be an enum");
        }
        // caching for result could be useful
        return type.GetFields()
                               .Where(field => field.IsStatic)
                               .Select(field => new
                                            {
                                                field,
                                                attribute = field.GetCustomAttribute<EnumOrderAttribute>()
                                            })
                                .Select(fieldInfo => new
                                             {
                                                 name = fieldInfo.field.Name,
                                                 order = fieldInfo.attribute != null ? fieldInfo.attribute.Order : 0
                                             })
                               .OrderBy(field => field.order)
                               .Select(field => field.name);
    }
}

public enum TestEnum
{
    [EnumOrder(Order=2)]
    Second = 1,

    [EnumOrder(Order=1)]
    First = 4,

    [EnumOrder(Order=3)]
    Third = 0
}

var names = typeof(TestEnum).GetWithOrder();
var names = TestEnum.First.GetWithOrder();

枚舉類型變量的擴展方法沒有意義,因為您不需要該變量的值-您只需要枚舉類型。 因此,我將使用靜態助手類:

public static class Enum<T>
{
    public static IEnumerable<T> GetOrderedValues()
    {
        var type = typeof(T);
        if (!type.IsEnum)
            throw new InvalidOperationException();

        return from f in type.GetFields(BindingFlags.Static | BindingFlags.Public)
               orderby f.GetCustomAttribute<EnumOrderAttribute>()?.Order ?? Int32.MaxValue
               select (T)f.GetValue(obj: null);
    }
}

注意:枚舉類型具有一個非靜態value字段,該字段實際上保存枚舉的值。 因此,如果要列出枚舉成員,則只需采用靜態字段。

用法

var values = Enum<Fruits>.GetOrderedValues();

對於示例枚舉

public enum Fruits
{
    Banana,
    [EnumOrder(2)]
    Apple,
    [EnumOrder(1)]
    Lemon
}

輸出將是

[ Lemon, Apple, Banana ]

沒有EnumOrder屬性的成員將排在最后,但是如果您使用其他默認順序代替Int32.MaxValue ,則可以更改該行為。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM