serein-flow/Library/Network/WebSocket/Handle/WebSocketHandleModule.cs

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

namespace Serein.Library.Network.WebSocketCommunication.Handle
{
    /// <summary>
    /// Json消息处理模块
    /// </summary>
    public class WebSocketHandleModule
    {
        /// <summary>
        /// Json消息处理模块
        /// </summary>
        public WebSocketHandleModule(string ThemeJsonKey, string DataJsonKey)
        {
            this.ThemeJsonKey = ThemeJsonKey;
            this.DataJsonKey = DataJsonKey;
        }

        /// <summary>
        /// 指示处理模块该使用json中的哪个key作为业务区别字段
        /// </summary>
        public string ThemeJsonKey { get; }

        /// <summary>
        /// 指示处理模块该使用json中的哪个key作为业务数据字段
        /// </summary>
        public string DataJsonKey { get; }

        /// <summary>
        /// 存储处理数据的配置
        /// </summary>
        public ConcurrentDictionary<string, JsonMsgHandleConfig> MyHandleConfigs = new ConcurrentDictionary<string, JsonMsgHandleConfig>();

        internal void AddHandleConfigs(SocketHandleModel model, ISocketHandleModule instance, MethodInfo methodInfo
            , Action<Exception, Action<object>> onExceptionTracking)
        {
            if (!MyHandleConfigs.ContainsKey(model.ThemeValue))
            {
                var jsonMsgHandleConfig = new JsonMsgHandleConfig(model,instance, methodInfo, onExceptionTracking);
                MyHandleConfigs[model.ThemeValue] = jsonMsgHandleConfig;
            }
        }

        /// <summary>
        /// 移除某个处理模块
        /// </summary>
        /// <param name="socketControlBase"></param>
        /// <returns></returns>
        public bool RemoveConfig(ISocketHandleModule socketControlBase)
        {
            foreach (var kv in MyHandleConfigs.ToArray())
            {
                var config = kv.Value;
                if (config.HandleGuid.Equals(socketControlBase.HandleGuid))
                {
                    MyHandleConfigs.TryRemove(kv.Key, out _);
                }
            }
            return MyHandleConfigs.Count == 0;
        }

        /// <summary>
        /// 卸载当前模块的所有配置
        /// </summary>
        public void UnloadConfig()
        {
            var temp = MyHandleConfigs.Values;
            MyHandleConfigs.Clear();
            foreach (var config in temp)
            {
                config.Clear();
            }
        }

        /// <summary>
        /// 处理JSON数据
        /// </summary>
        /// <param name="RecoverAsync"></param>
        /// <param name="jsonObject"></param>
        public void HandleSocketMsg(Func<string, Task> RecoverAsync, JObject jsonObject)
        {
            // 获取到消息
            string themeKeyName = jsonObject.GetValue(ThemeJsonKey)?.ToString();
            if (!MyHandleConfigs.TryGetValue(themeKeyName, out var handldConfig))
            {
                // 没有主题
                return;
            }
            if (jsonObject[DataJsonKey] is JObject dataJsonObject)
            {
                handldConfig.Handle(RecoverAsync, dataJsonObject);
            }
        }
    }

}