从枚举类型创建类的最干净的方法?

时间:2018-01-30 10:22:04

标签: c# enums factory system.reflection

我有以下枚举和类:

public enum MyEnum
{
   MyType1,
   MyType2
}

public abstract class MyBaseClass
{
   public abstract MyEnum GetMyType();
}

public class MySubClass : MyBaseClass
{
   public override MyEnum GetMyType()
   {
      return MyEnum.MyType1;
   }
}

我想基于GetMyType()创建一个MySubClass实例,但是不需要将MySubClass“注册”到某种处理程序(或者是这样的方式?)。我知道通过反射可能很容易做到(通过汇编继承MyBaseClass的类并检查它们的MyType),但这是要走的路吗?

现在我正在做一些很长的事情:

public MyBaseClass CreateMyClass(MyEnum myEnum)
{
   if(myEnum == MyEnum.MyType1)
   {
      return new MySubClass();
   }
   else if(myEnum == MyEnum.MyType2)
   {
      return new MyOtherSubClass();
   }
}

最终会导致我忘记手动添加新课程。它将在测试中被捕获,但我不需要添加类,因为已经提供了所需的所有信息。

对所有关于最佳方法的建议持开放态度。

2 个答案:

答案 0 :(得分:4)

您可以创建自定义属性来标识所需的枚举类。

public class MyCustomAttribute : Attribute
{
    public MyEnum EnumType { get; set; }
}

然后是枚举和类;

public enum MyEnum
{
    MyType1,
    MyType2
}
public class MyBaseClass
{

}
[MyCustomAttribute(EnumType = MyEnum.MyType1)]
public class MySubClass : MyBaseClass
{

}
[MyCustomAttribute(EnumType = MyEnum.MyType2)]
public class MyOtherClass : MyBaseClass
{

}

您可以使用Reflection来确定包含枚举的相关类。

private static Dictionary<MyEnum, Type> _myEnumDictionary = new Dictionary<MyEnum, Type>();

public MyBaseClass GetEnumClass(MyEnum enumType)
{
    if (!_myEnumDictionary.ContainsKey(enumType))
    {
        var enumClass = typeof(MySubClass).Assembly
            .GetTypes()
            .FirstOrDefault(x => x.GetCustomAttributes<MyCustomAttribute>()
                .Any(k => k.EnumType == enumType));
        if (enumClass == null)
        {
            throw new Exception("There is no declared class with the enumType" + enumType);
        }
        _myEnumDictionary.Add(enumType, enumClass);
    }
    return (MyBaseClass)Activator.CreateInstance(_myEnumDictionary[enumType]);
}

最后,您可以获得相关类的实例;

var mySubClass = GetEnumClass(MyEnum.MyType1);
var myOtherClass = GetEnumClass(MyEnum.MyType2);
//There is no lookup here. It will get the Type from Dictionary directly.
var mySubClass2 = GetEnumClass(MyEnum.MyType1); 

此外,您可以使用静态Dictionary<MyEnum, Type>字典集合来防止每次查找。

答案 1 :(得分:1)

我希望这个例子可以解决你的问题。

public enum MyEnum
{
    Type1, Type2
}
public abstract class MyBaseClass
{
    public abstract MyEnum GetMyType();
}
public class MySubClass1 : MyBaseClass
{
    public override MyEnum GetMyType()
    {
        // implementation
    }
}
public class MySubClass2 : MyBaseClass
{
    public override MyEnum GetMyType()
    {
        // implementation
    }
}

所以,您MySubClass1MySubClass2作为MyBaseClass的孩子 这是你的方法。

    public MyBaseClass CreateMyClass(MyEnum myEnum)
    {
        var dict = new Dictionary<MyEnum, Type>();
        dict.Add(MyEnum.Type1, typeof(MySubClass1));
        dict.Add(MyEnum.Type2, typeof(MySubClass2));

        var type = dict.Where(x => x.Key == myEnum).Select(x => x.Value);
        var instance = Activator.CreateInstance(type.GetType());
        return (MyBaseClass)instance;
    }

让我简要解释一下。 我们只是在DictionaryEnum之间的Type创建了一个映射 根据我们在参数中得到的EnumActivator创建从映射中找到的类型的实例。

这里唯一的问题是如果我们在MyBaseClass字典中放置除dict的子类以外的任何类型,它将抛出运行时异常。

---------------编辑----------------

如果你想完全避免映射,那么你可以将类型作为函数的string参数传递,并实现如下所示

ObjectType instance = 
 (ObjectType)Activator.CreateInstance("MyAssembly","MyNamespace." + argType);