Блог

Проблема

У меня есть определение сообщения protobuf с MessageType полем, которое является перечислением. Учитывая входящее сообщение protobuf, я хотел бы разрешить некоторые IMessageHandler файлы из контейнера IoC на основе MessageType . Проблема двоякая: как мне выразить MessageType ограничение при написании IMessageHandler и как мне разрешить только нужные обработчики из контейнера IoC?

Я использую Autofac, но мне интересно услышать решения для любого контейнера.

Мои мысли:

Для выражения ограничения я вижу два варианта: свойство или атрибут (не могу использовать общие, потому что это значение enum). Мне нравится свойство, потому что оно делает невозможным запись IMessageHandler без указания ограничения, но недостатком является то, что его необходимо создать, прежде чем вы сможете увидеть значение свойства.

В моем текущем коде я использую подход свойств. Я разрешаю все IMessageHandler и выполняю фильтрацию вручную, но, похоже, должен быть способ получше. Не то чтобы я слишком беспокоился о производительности, просто похоже, что я разрешаю экземпляры, которые не используются.

Обновить

Чтобы сделать это немного более понятным, вот что я сейчас делаю

 public interface IMessageHandler
{
    MessageType TargetType { get; }

    void Handle(IMessage message);
}

public class SomeHandler : IMessageHandler
{
    public MessageType TargetType
    {
        get { return MessageType.Type1;  }
    }

    public void Handle(IMessage message)
    {
        // implementation
    }
}
  

Поэтому, когда я получаю сообщение protobuf, я разрешаю все IMessageHandler сообщения и вызываю те, которые TargetType соответствуют MessageType входящему сообщению.

Я предлагаю вам зарегистрировать один класс IMessageHandler-Factory с помощью autofac и реализовать метод на этой фабрике, который принимает ваше перечисление и возвращает правильную реализацию IMessageHandler.

Я понял, что сервисы с ключами Autofac могут мне помочь. Я думаю, что я собираюсь использовать этот подход.

1. Используйте атрибут, чтобы объявить, в чем MessageType данный IMessageHandler объект заинтересован.
2. Регистрируйте каждый IMessageHandler введенный в MessageType
3. Используйте ResolveKeyed , чтобы получить только те IMessageHandler , которые меня интересуют int.

Приятно то, что если кто-то забудет использовать атрибут, мы можем перехватить это при сборке контейнера. Вот полный пример. Любые предложения приветствуются!

 class Program
{
    static IContainer container;

    static void Main(string[] args)
    {
        var builder = new ContainerBuilder();
        builder.RegisterAssemblyTypes(typeof(Program).Assembly)
            .AssignableTo<IMessageHandler>()
            .Keyed<IMessageHandler>(t => GetMessageType(t));

        container = builder.Build();

        InvokeHandlers(MessageType.Type1);
        InvokeHandlers(MessageType.Type2);

        Console.ReadKey();
    }

    static MessageType GetMessageType(Type type)
    {
        var att = type.GetCustomAttributes(true).OfType<MessageHandlerAttribute>().FirstOrDefault();
        if (att == null)
        {
            throw new Exception("Somone forgot to put the MessageHandlerAttribute on an IMessageHandler!");
        }

        return att.MessageType;
    }

    static void InvokeHandlers(MessageType type)
    {
        using (var lifetime = container.BeginLifetimeScope())
        {
            // I'm impressed that Autofac knows what I mean here!
            var handlers = lifetime.ResolveKeyed<IEnumerable<IMessageHandler>>(type);
            foreach (var handler in handlers)
            {
                handler.Handle();
            }
        }
    }
}

public enum MessageType
{
    Type1,
    Type2,
}

public interface IMessageHandler
{
    void Handle();
}

public class MessageHandlerAttribute : Attribute
{
    public MessageHandlerAttribute(MessageType messageType)
    {
        MessageType = messageType;
    }

    public MessageType MessageType { get; private set; }
}

[MessageHandler(MessageType.Type1)]
public class Handler1 : IMessageHandler
{
    public void Handle()
    {
        Console.WriteLine("A handler for Type1");
    }
}

[MessageHandler(MessageType.Type1)]
public class Handler2 : IMessageHandler
{
    public void Handle()
    {
        Console.WriteLine("Another handler for Type1");
    }
}

[MessageHandler(MessageType.Type2)]
public class Handler3 : IMessageHandler
{
    public void Handle()
    {
        Console.WriteLine("A handler for Type2");
    }
}