Prowlarr/NzbDrone.Common/Messaging/MessageAggregator.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using NLog;

namespace NzbDrone.Common.Messaging
{
    public class MessageAggregator : IMessageAggregator
    {
        private readonly Logger _logger;
        private readonly Func<IEnumerable<IProcessMessage>> _handlers;

        public MessageAggregator(Logger logger, Func<IEnumerable<IProcessMessage>> handlers)
        {
            _logger = logger;
            _handlers = handlers;
        }

        public void Publish<TEvent>(TEvent message) where TEvent : IEvent
        {
            _logger.Trace("Publishing {0}", message.GetType().Name);

            //call synchronous handlers first.
            foreach (var handler in _handlers().OfType<IHandle<TEvent>>())
            {
                _logger.Debug("{0} -> {1}", message.GetType().Name, handler.GetType().Name);
                handler.Handle(message);
                _logger.Debug("{0} <- {1}", message.GetType().Name, handler.GetType().Name);
            }

            foreach (var handler in _handlers().OfType<IHandleAsync<TEvent>>())
            {
                var handlerLocal = handler;
                Task.Factory.StartNew(() =>
                    {
                        _logger.Debug("{0} ~> {1}", message.GetType().Name, handlerLocal.GetType().Name);
                        handlerLocal.HandleAsync(message);
                        _logger.Debug("{0} <~ {1}", message.GetType().Name, handlerLocal.GetType().Name);
                    });
            }
        }

        public void Execute<TCommand>(TCommand message) where TCommand : ICommand
        {
            throw new NotImplementedException();
        }
    }
}