93 lines
3.0 KiB
C#
93 lines
3.0 KiB
C#
using System;
|
|
using System.Reflection;
|
|
using System.Threading.Tasks;
|
|
using NLog;
|
|
using NzbDrone.Common.Composition;
|
|
|
|
namespace NzbDrone.Common.Messaging
|
|
{
|
|
[Singleton]
|
|
public class MessageAggregator : IMessageAggregator
|
|
{
|
|
private readonly Logger _logger;
|
|
private readonly IServiceFactory _serviceFactory;
|
|
|
|
public MessageAggregator(Logger logger, IServiceFactory serviceFactory)
|
|
{
|
|
_logger = logger;
|
|
_serviceFactory = serviceFactory;
|
|
}
|
|
|
|
public void PublishEvent<TEvent>(TEvent @event) where TEvent : IEvent
|
|
{
|
|
var eventName = GetEventName(@event.GetType());
|
|
|
|
_logger.Trace("Publishing {0}", eventName);
|
|
|
|
//call synchronous handlers first.
|
|
foreach (var handler in _serviceFactory.BuildAll<IHandle<TEvent>>())
|
|
{
|
|
try
|
|
{
|
|
_logger.Debug("{0} -> {1}", eventName, handler.GetType().Name);
|
|
handler.Handle(@event);
|
|
_logger.Debug("{0} <- {1}", eventName, handler.GetType().Name);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.ErrorException(string.Format("{0} failed while processing [{1}]", handler.GetType().Name, eventName), e);
|
|
}
|
|
}
|
|
|
|
foreach (var handler in _serviceFactory.BuildAll<IHandleAsync<TEvent>>())
|
|
{
|
|
var handlerLocal = handler;
|
|
Task.Factory.StartNew(() =>
|
|
{
|
|
_logger.Debug("{0} ~> {1}", eventName, handlerLocal.GetType().Name);
|
|
handlerLocal.HandleAsync(@event);
|
|
_logger.Debug("{0} <~ {1}", eventName, handlerLocal.GetType().Name);
|
|
});
|
|
}
|
|
}
|
|
|
|
|
|
private static string GetEventName(Type eventType)
|
|
{
|
|
if (!eventType.IsGenericType)
|
|
{
|
|
return eventType.Name;
|
|
}
|
|
|
|
return string.Format("{0}<{1}>", eventType.Name.Remove(eventType.Name.IndexOf('`')), eventType.GetGenericArguments()[0].Name);
|
|
}
|
|
|
|
|
|
public void PublishCommand<TCommand>(TCommand command) where TCommand : ICommand
|
|
{
|
|
var handlerContract = typeof(IExecute<>).MakeGenericType(command.GetType());
|
|
|
|
_logger.Trace("Publishing {0}", command.GetType().Name);
|
|
|
|
var handler = _serviceFactory.Build(handlerContract);
|
|
|
|
_logger.Debug("{0} -> {1}", command.GetType().Name, handler.GetType().Name);
|
|
|
|
try
|
|
{
|
|
handlerContract.GetMethod("Execute").Invoke(handler, new object[] { command });
|
|
}
|
|
catch (TargetInvocationException e)
|
|
{
|
|
if (e.InnerException != null)
|
|
{
|
|
throw e.InnerException;
|
|
}
|
|
throw;
|
|
}
|
|
|
|
_logger.Debug("{0} <- {1}", command.GetType().Name, handler.GetType().Name);
|
|
}
|
|
}
|
|
}
|