如何向Enums添加扩展方法

Question

我有这个枚举代码:

enum Duration { Day, Week, Month };

我可以为此枚举添加扩展方法吗？

Answer 1

根据这个网站:

扩展方法提供了一种以团队中其他人可能实际发现和使用的方式为现有类编写方法的方法.鉴于枚举类似于任何其他类,因此您可以扩展它们并不会太令人惊讶,例如:

enum Duration { Day, Week, Month };

static class DurationExtensions 
{
  public static DateTime From(this Duration duration, DateTime dateTime) 
  {
    switch (duration) 
    {
      case Day:   return dateTime.AddDays(1);
      case Week:  return dateTime.AddDays(7);
      case Month: return dateTime.AddMonths(1);
      default:    throw new ArgumentOutOfRangeException("duration");
    }
  }
}

我认为枚举通常不是最好的选择,但至少这可以让你集中一些开关/如果处理并将它们抽象一点,直到你可以做得更好.请记住检查值是否在范围内.

你可以阅读更多在这里在MSDN的Microsft.

Answer 2

您还可以向Enum类型添加扩展方法,而不是Enum的实例:

/// <summary> Enum Extension Methods </summary>
/// <typeparam name="T"> type of Enum </typeparam>
public class Enum<T> where T : struct, IConvertible
{
    public static int Count
    {
        get
        {
            if (!typeof(T).IsEnum)
                throw new ArgumentException("T must be an enumerated type");

            return Enum.GetNames(typeof(T)).Length;
        }
    }
}

您可以通过执行以下操作来调用上面的扩展方法:

var result = Enum<Duration>.Count;

这不是一个真正的扩展方法.它只能起作用,因为Enum <>是一个与System.Enum不同的类型.

Answer 3

当然,例如,您可以使用DescriptionAttribue您的enum值:

using System.ComponentModel.DataAnnotations;

public enum Duration 
{ 
    [Description("Eight hours")]
    Day,

    [Description("Five days")]
    Week,

    [Description("Twenty-one days")] 
    Month 
}

现在你希望能够做到这样的事情:

Duration duration = Duration.Week;
var description = duration.GetDescription(); // will return "Five days"

您的扩展方法GetDescription()可以写成如下:

using System.ComponentModel;
using System.Reflection;

public static string GetDescription(this Enum value)
{
    FieldInfo fieldInfo = value.GetType().GetField(value.ToString());
    if (fieldInfo == null) return null;
    var attribute = (DescriptionAttribute)fieldInfo.GetCustomAttribute(typeof(DescriptionAttribute));
    return attribute.Description;
}

Answer 4

所有答案都很棒,但他们正在讨论将扩展方法添加到特定类型的枚举

如果你想为所有枚举添加方法,比如返回当前值的int而不是显式转换,该怎么办？

public static class EnumExtensions
{
    public static int ToInt<T>(this T soure) where T : IConvertible//enum
    {
        if (!typeof(T).IsEnum)
            throw new ArgumentException("T must be an enumerated type");

        return (int) (IConvertible) soure;
    }

    //ShawnFeatherly funtion (above answer) but as extention method
    public static int Count<T>(this T soure) where T : IConvertible//enum
    {
        if (!typeof(T).IsEnum)
            throw new ArgumentException("T must be an enumerated type");

        return Enum.GetNames(typeof(T)).Length;
    }
}

其背后的技巧IConvertible是它的继承层次结构,请参阅MDSN

感谢ShawnFeatherly的回答

Answer 5

您甚至可以为任何事物创建扩展名object(尽管这不是最佳实践).理解扩展方法就像一种public static方法.您可以在方法上使用您喜欢的任何参数类型.

public static class DurationExtensions
{
  public static int CalculateDistanceBetween(this Duration first, Duration last)
  {
    //Do something here
  }
}

Answer 6

一个简单的解决方法。

public static class EnumExtensions
{
    public static int ToInt(this Enum payLoad) {

        return ( int ) ( IConvertible ) payLoad;

    }
}

int num = YourEnum.AItem.ToInt();
Console.WriteLine("num : ", num);

Answer 7

参见MSDN。

public static class Extensions
{
  public static string SomeMethod(this Duration enumValue)
  {
    //Do something here
    return enumValue.ToString("D"); 
  }
}