初始提交

2020-10-07 20:25:03 +08:00
commit d318014316
3809 changed files with 263103 additions and 0 deletions
--- a/Infrastructure/Hncore.Infrastructure/EventBus/EventBus.cs
+++ b/Infrastructure/Hncore.Infrastructure/EventBus/EventBus.cs
@@ -0,0 +1,150 @@
+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);
+                    }
+                }
+            }
+        }
+    }
+}