SimulationServerYL/KYNetwork/Network/Message/MessageDispatcherComponent.cs


namespace KYFramework.Network
{
    [ObjectSystem]
    public class MessageDispatcherComponentAwakeSystem : AwakeSystem<MessageDispatcherComponent>
    {
        public override void Awake(MessageDispatcherComponent t)
        {
            t.Awake();
        }
    }

    [ObjectSystem]
    public class MessageDispatcherComponentLoadSystem : LoadSystem<MessageDispatcherComponent>
    {
        public override void Load(MessageDispatcherComponent self)
        {
            self.Load();
        }
    }
    /// <summary>
    /// 消息分发组件
    /// </summary>
    public class MessageDispatcherComponent : Component
    {
        private readonly Dictionary<ushort, List<IMHandler>> handlers = new Dictionary<ushort, List<IMHandler>>();

        public void Awake()
        {
            this.Load();

            Log.Info("消息分发组件初始化完毕！");
        }

        public void Load()
        {
            this.handlers.Clear();

            List<Type> types = Game.EventSystem.GetTypes(typeof(MessageHandlerAttribute));

            foreach (Type type in types)
            {
                object[] attrs = type.GetCustomAttributes(typeof(MessageHandlerAttribute), false);
                if (attrs.Length == 0)
                {
                    continue;
                }

                IMHandler iMHandler = Activator.CreateInstance(type) as IMHandler;
                if (iMHandler == null)
                {
                    Log.Error($"message handle {type.Name} 需要继承 IMHandler");
                    continue;
                }
                Type messageType = iMHandler.GetMessageType();
                ushort opcode = this.Entity.GetComponent<OpcodeTypeComponent>().GetOpcode(messageType);
                if (opcode == 0)
                {
                    Log.Error($"消息opcode为0: {messageType.Name}");
                    continue;
                }
                this.RegisterHandler(opcode, iMHandler);
            }
        }

        public void RegisterHandler(ushort opcode, IMHandler handler)
        {
            if (!this.handlers.ContainsKey(opcode))
            {
                this.handlers.Add(opcode, new List<IMHandler>());
            }
            this.handlers[opcode].Add(handler);
        }

        public void Handle(Session session, MessageInfo messageInfo)
        {
            List<IMHandler> actions;
            if (!this.handlers.TryGetValue(messageInfo.Opcode, out actions))
            {
                Log.Error($"消息没有处理: {messageInfo.Opcode} {JsonHelper.ToJson(messageInfo.Message)}");
                return;
            }

            foreach (IMHandler ev in actions)
            {
                try
                {
                    ev.Handle(session, messageInfo.Message);
                }
                catch (Exception e)
                {
                    Log.Error(e);
                }
            }
        }

        public override void Dispose()
        {
            if (this.IsDisposed)
            {
                return;
            }

            base.Dispose();
        }
    }
}