juipnet/Infrastructure/Hncore.Infrastructure/EventBus/EventBus.cs

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;  

namespace Hncore.Infrastructure.Events
{
    /// <summary>
    /// 事件总线
    /// </summary>
    public class EventBus
    {
        private static EventBus _eventBus = null;

        public static EventBus Default
        {
            get { return _eventBus ?? (_eventBus = new EventBus()); }
        }

        /// <summary>
        /// 定义线程安全集合
        /// </summary>
        private readonly ConcurrentDictionary<Type, List<IEventHandler>> _eventAndHandlerMapping;

        public EventBus()
        {
            _eventAndHandlerMapping = new ConcurrentDictionary<Type, List<IEventHandler>>();
            MapEventToHandler();
        }

        /// <summary>
        ///通过反射，将事件源与事件处理绑定
        /// </summary>
        private void MapEventToHandler()
        {
            //  Assembly assembly = Assembly.GetEntryAssembly();

            var allAssembly = AppDomain.CurrentDomain.GetAssemblies().Where(item => item.FullName.Contains("Microkj."));
            if (!allAssembly.Any())
            {
                return;
            }

            foreach (var assembly in allAssembly)
            {
                foreach (var type in assembly.GetTypes())
                {
                    if (!type.IsGenericType && typeof(IEventHandler).IsAssignableFrom(type)) //判断当前类型是否实现了IEventHandler接口
                    {
                        Type handlerInterface = type.GetInterface("IEventHandler`1"); //获取该类实现的泛型接口
                        if (handlerInterface != null)
                        {
                            Type eventDataType = handlerInterface.GetGenericArguments()[0]; // 获取泛型接口指定的参数类型

                            if (_eventAndHandlerMapping.ContainsKey(eventDataType))
                            {
                                List<IEventHandler> handlerTypes = _eventAndHandlerMapping[eventDataType];
                                handlerTypes.Add(Activator.CreateInstance(type) as IEventHandler);
                                _eventAndHandlerMapping[eventDataType] = handlerTypes;
                            }
                            else
                            {
                                var handlerTypes = new List<IEventHandler>
                                {
                                    Activator.CreateInstance(type) as IEventHandler
                                };
                                _eventAndHandlerMapping[eventDataType] = handlerTypes;
                            }
                        }
                    }
                }
            }
        }

        /// <summary>
        /// 手动绑定事件源与事件处理
        /// </summary>
        /// <typeparam name="TEventData"></typeparam>
        /// <param name="eventHandler"></param>
        public void Register<TEventData>(IEventHandler eventHandler)
        {
            if (_eventAndHandlerMapping.Keys.Contains(typeof (TEventData)))
            {
                List<IEventHandler> handlerTypes = _eventAndHandlerMapping[typeof (TEventData)];
                if (!handlerTypes.Contains(eventHandler))
                {
                    handlerTypes.Add(eventHandler);
                    _eventAndHandlerMapping[typeof (TEventData)] = handlerTypes;
                }
            }
            else
            {
                _eventAndHandlerMapping.GetOrAdd(typeof (TEventData), (type) => new List<IEventHandler>())
                    .Add(eventHandler);
            }
        }


        public void Register<TEventData>(Action<TEventData> action) where TEventData : IEventData
        {
            var actionHandler = new ActionEventHandler<TEventData>(action);
            Register<TEventData>(actionHandler);
        }

        /// <summary>
        /// 手动解除事件源与事件处理的绑定
        /// </summary>
        /// <typeparam name="TEventData"></typeparam>
        /// <param name="eventHandler"></param>
        public void UnRegister<TEventData>(Type eventHandler)
        {
            List<IEventHandler> handlerTypes = _eventAndHandlerMapping[typeof (TEventData)];

            _eventAndHandlerMapping.GetOrAdd(typeof (TEventData), (type) => new List<IEventHandler>())
                .RemoveAll(t => t.GetType() == eventHandler);
        }

        /// <summary>
        /// 根据事件源触发绑定的事件处理
        /// </summary>
        /// <typeparam name="TEventData"></typeparam>
        /// <param name="eventData"></param>
        public static void Publish<TEventData>(TEventData eventData) where TEventData : IEventData
        {
            List<IEventHandler> handlers = Default._eventAndHandlerMapping[typeof (TEventData)];

            if (handlers != null && handlers.Count > 0)
            {
                foreach (var handler in handlers)
                {
                    var eventHandler = handler as IEventHandler<TEventData>;
                    if (eventHandler.Ansyc)
                    {
                        Task.Run(() =>
                        {
                            eventHandler.HandleEvent(eventData);
                        });
                    }
                    else
                    {
                        eventHandler.HandleEvent(eventData);
                    }
                }
            }
        }
    }
}