一个简单的 Generic Factory 类

简单的工厂类的一个使用场景是，假设有一个基类 BaseClass，和一系列的子类 A， B， C，工厂类根据某个参数，例如字符串 “A”, “B”, “C” 创建出相应的子类。举例如下：

public class Factory
{public static BaseClass Create(string name){switch (name){case "A": return new A();case "B": return new B();case "C": return new C();default: throw new ArgumentException("Wrong Name");}}
}

这里的一个问题是，当子类增加或减少时， Factory 类需要相应的改动。有没有办法可以只是改动子类本身，而不用修改Factory类呢，当然有，这里我举一个简单的实现。

基本思想是在每个子类上附加一个 Attribute，定义如下：

[AttributeUsage(AttributeTargets.Class)]
public class FactoryKeyAttribute : Attribute
{public object Key { get; set; }
}

假设我们有基类和子类实现如下

public abstract class BaseClass {}[FactoryKey(Key = "Standard")]
public class Standard : BaseClass {}[FactoryKey(Key = "Enterprise")]
public class Enterprise : BaseClass {}[FactoryKey(Key = "Lite")]
public class Lite : BaseClass {}

假设这些类都在同一个 Assembly中（对于不在同一个Assembly的，实现会稍微复杂些）工厂类需要预先加载 Key => Type 的Mapping，然后根据Key创建不同的实例，实现如下：

public class Factory<TKey, TBaseClass>
{private static readonly IDictionary<TKey, Type> TypeDict = Init();private static IDictionary<TKey, Type> Init(){var dict = from type in Assembly.GetExecutingAssembly().GetTypes()let key = (FactoryKeyAttribute)Attribute.GetCustomAttribute(type, typeof(FactoryKeyAttribute))where key != null && typeof(TBaseClass).IsAssignableFrom(type)select new { Key = key, Value = type };return dict.ToDictionary(kvp => (TKey)kvp.Key.Key, kvp => kvp.Value);}public static TBaseClass CreateInstance(TKey key){Type type;if (TypeDict.TryGetValue(key, out type)){return (TBaseClass)Activator.CreateInstance(type);}throw new ArgumentException("Incorrect Key!");}
}

使用方法也很简单：

BaseClass s = Factory<string, BaseClass>.CreateInstance("Standard");
BaseClass l = Factory<string, BaseClass>.CreateInstance("Lite");
BaseClass e = Factory<string, BaseClass>.CreateInstance("Enterprise");