0

我有一个基本的 Message 类，以及大约 100 个不同的 Message 子类型类，它们代表可以处理的每种类型的消息。我目前正在考虑做的是使用一个巨大的 switch 语句来创建消息对象。例如：

switch (MsgType)
{
   case MessageType.ChatMsg:
      Msg = new MsgChat(Buf);
      break;
   case MessageType.ResultMsg:
      Msg = new MsgResult(Buf);
      break;
   ... // 98 more case statements
}
Msg.ProcessMsg(); // Use a polymorphic call to process the message.

有一个更好的方法吗？如果是这样，您能否展示一个简单的代码示例。

编辑

所以，我试着这样做：

public class Test
{
   public Test()
   {
      IEnumerable<Type> myEnumerable = GetTypesWith<MyAttribute>(true);
   }

   IEnumerable<Type> GetTypesWith<TAttribute>(bool inherit)
      where TAttribute : System.Attribute
   {
      return from a in AppDomain.CurrentDomain.GetAssemblies()
             from t in a.GetTypes()
             where t.IsDefined(typeof(TAttribute), inherit)
             select t;
   }
}

这似乎有效，因为 myEnumerable 现在包含所有 100 个消息子类型，以及基本消息类型。然而，虽然我不介意在程序开始时使用反射来加载类型，但使用它来实时访问正确的对象可能太慢了。所以，我想尝试使用委托。

@Mark Hildreth 下面评论中的示例：

“所以，你会有一个 > 的字典。然后，你的映射将是 mappings[MessageType.ChatMsg] = x => new MsgChat(x);”

有几种方法可以解释此代码。一种想法是删除所有 100 个子类，只使用一个具有 100 个委托方法的大型类。这是一个遥远的第二选择。另一个想法和我的第一选择是让上面的代码以某种方式创建一个消息子类对象。但是，我不太明白它会如何做到这一点。此外，最好将上述技术保留在我的 Test 类中，以获取所有类型或委托，而不必编写全部 100 个。你或其他人能解释一下如何做到这一点吗？

4

2 回答 2

3

代替使用巨大的switch语句，您可以定义 aDictionary以将每个MessageType值映射到其定义的Message派生类，并使用此映射数据创建一个实例。

字典定义：

Dictionary<int, Type> mappings = new Dictionary<int, Type>();
mappings.Add(MessageType.ChatMsg, typeof(MsgChat));
mappings.Add(MessageType.ResultMsg, typeof(MsgResult));

...

字典消费：

ConstructorInfo ctor = mappings[MessageType.ChatMsg].GetConstructor(new[] { typeof(Buf) });
Message message = (Message)ctor.Invoke(new object[] { Buf });

请注意，我没有编译此代码来验证是否正确。我只想向你展示这个想法。

编辑

有我的新答案来改进第一个。我正在考虑您编辑的问题，使用来自@MikeSW和 @Mark Hildreth的给定想法。

public class FactoryMethodDelegateAttribute : Attribute
{
    public FactoryMethodDelegateAttribute(Type type, string factoryMethodField, Message.MessageType typeId)
    {
        this.TypeId = typeId;
        var field = type.GetField(factoryMethodField);
        if (field != null)
        {
            this.FactoryMethod = (Func<byte[], Message>)field.GetValue(null);
        }
    }

    public Func<byte[], Message> FactoryMethod { get; private set; }
    public Message.MessageType TypeId { get; private set; }
}

public class Message
{
    public enum MessageType
    {
        ChatMsg,
    }
}

[FactoryMethodDelegate(typeof(ChatMsg), "FactoryMethodDelegate", Message.MessageType.ChatMsg)]
public class ChatMsg : Message
{
    public static readonly MessageType MessageTypeId = MessageType.ChatMsg;
    public static readonly Func<byte[], Message> FactoryMethodDelegate = buffer => new ChatMsg(buffer);
    public ChatMsg(byte[] buffer)
    {
        this.Buffer = buffer;
    }

    private byte[] Buffer { get; set; }
 }

public class TestClass
{
    private IEnumerable<Type> GetTypesWith<TAttribute>(bool inherit) where TAttribute : Attribute
    {
        return from a in AppDomain.CurrentDomain.GetAssemblies()
               from t in a.GetTypes()
               where t.IsDefined(typeof(TAttribute), inherit)
               select t;
    }

    [Test]
    public void Test()
    {
        var buffer = new byte[1];
        var mappings = new Dictionary<Message.MessageType, Func<byte[], Message>>();
        IEnumerable<Type> types = this.GetTypesWith<FactoryMethodDelegateAttribute>(true);
        foreach (var type in types)
        {
            var attribute =
                (FactoryMethodDelegateAttribute)
                type.GetCustomAttributes(typeof(FactoryMethodDelegateAttribute), true).First();

            mappings.Add(attribute.TypeId, attribute.FactoryMethod);
        }

        var message = mappings[Message.MessageType.ChatMsg](buffer);
    }
}
于 2013-02-05T22:13:42.170 回答
2

你走在正确的轨道上，使用字典是个好主意。如果反射太慢，你可以使用表达式，像这样（我假设你用 MessageTypeAttribute 装饰 Messages 类）。

public class Test
{
 public Test()
  {
     var dict=new Dictionary<MessageType,Func<Buffer,Mesage>>();
     var types=from a in AppDomain.CurrentDomain.GetAssemblies()
         from t in a.GetTypes()
         where t.IsDefined(MessageTypeAttribute, inherit)
         select t;
    foreach(var t in types) {
      var attr = t.GetCustomAttributes(typeof (MessageTypeAttribute), false).First();
       dict[attr.MessageType] = CreateFactory(t);
       }

      var msg=dict[MessageType.Chat](Buf);
  }

 Func<Buffer,Message> CreateFactory(Type t)
 {
      var arg = Expression.Parameter(typeof (Buffer));
        var newMsg = Expression.New(t.GetConstructor(new[] {typeof (Buffer)}),arg);
        return Expression.Lambda<Func<Buffer, Message>>(newMsg, arg).Compile();
}

}
于 2013-02-06T11:20:14.227 回答