Unity websocket

记录一下Unity WebSocket的简单用法

using System;
using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityWebSocket;

public class WebSocketLogic :MonoBehaviour
{

    private WebSocketLogic() { }
    private WebSocketLogic _instance;
    private bool connected = false;
    private bool inited = false;
    private Dictionary<SocketCode, List<Action<string>>> msgResponseDic = new Dictionary<SocketCode, List<Action<string>>>();
    private Dictionary<SocketCode, List<Action<string>>> msgListenerDic = new Dictionary<SocketCode, List<Action<string>>>();

   

   public WebSocketLogic Instance
    {
        get
        {
            if (_instance == null)
            {
                _instance = new WebSocketLogic();
            }
            return _instance;
        }
        private set { }
    }

    WebSocket webSocket;
    string ip;

    public void Init(string _ip)
    {
        inited = true;
        ip = _ip;
    }

    public void Connect()
    {
        RemoveHandle();
        webSocket = new WebSocket(ip);
        webSocket.OnClose += WebSocketClose;
        webSocket.OnOpen += WebSocketOpen;
        webSocket.OnError += WebSocketError;
        webSocket.OnMessage += WebSocketReceive;
        webSocket.ConnectAsync();
    }

    public void AddListener(SocketCode socketCode,Action<string> callback)
    {
        if (msgListenerDic.ContainsKey(socketCode))
        {
            List<Action<string>> actions = msgListenerDic[socketCode];
            for (int i = 0; i < actions.Count; i++)
            {
                if (actions[i] == callback)
                {
                    Debug.LogError("重复注册监听，消息号:" + socketCode);
                    return;
                }

            }
            msgListenerDic[socketCode].Add(callback);
        }
        else
        {
            List<Action<string>> actions = new List<Action<string>>();
            actions.Add(callback);
            msgListenerDic.Add(socketCode, actions);
        }
    }

    public void RemoveListener(SocketCode socketCode,Action<string> callback)
    {
        if (msgListenerDic.ContainsKey(socketCode))
        {
            List<Action<string>> actions = msgListenerDic[socketCode];
            for (int i = 0; i < actions.Count; i++)
            {
                if (actions[i] == callback)
                {
                    actions.RemoveAt(i);
                    return;
                }
            }
            Debug.LogError("移除监听出错，未包注册监听消息函数，消息号:" + socketCode);
        }
        else
        {
            Debug.LogError("移除监听出错，未包注册监听，消息号:" + socketCode);
        }
    }

    void RemoveHandle()
    {
        msgResponseDic.Clear();
        webSocket = null;
        webSocket.OnClose -= WebSocketClose;
        webSocket.OnOpen -= WebSocketOpen;
        webSocket.OnError -= WebSocketError;
        webSocket.OnMessage -= WebSocketReceive;
    }

    public void Send(SocketCode socketCode, string msg,Action<string> callback)
    {
        if (callback != null)
        {
            if (msgResponseDic.ContainsKey(socketCode))
            {
                msgResponseDic[socketCode].Add(callback);
            }
            else
            {
                List<Action<string>> actions = new List<Action<string>>();
                actions.Add(callback);
                msgResponseDic.Add(socketCode, actions);
            }
        }
        webSocket.SendAsync(string.Format("{0}/{1}", socketCode, msg));
    }

    private void WebSocketReceive(object sender, MessageEventArgs e)
    {
        //优化逻辑
        //heartTime = 0;
        receiveMsgTime = Time.realtimeSinceStartup;
        
         if (e.IsText)
        {
            string[] temp = e.Data.Split('.');
            //发送接口需要返回的消息
            SocketCode socketCode = (SocketCode)int.Parse(temp[0]);
            if (msgResponseDic.ContainsKey(socketCode))
            {
                List<Action<string>> actions = msgResponseDic[socketCode];
                for (int i = 0; i < actions.Count; i++)
                {
                    actions[i](temp[1]);
                }
                msgResponseDic.Remove(socketCode);
            }

            if (msgListenerDic.ContainsKey(socketCode))
            {
                List<Action<string>> actions = msgListenerDic[socketCode];
                for (int i = 0; i < actions.Count; i++)
                {
                    actions[i](temp[1]);
                }
            }
            Debug.LogError(string.Format("接收到消息: {0}", e.Data));
        }
        else if (e.IsBinary)
        {
            Debug.LogError(string.Format("接受到消息 Bytes ({1}): {0}", e.Data, e.RawData.Length));
        }
    }

    private void WebSocketError(object sender, ErrorEventArgs e)
    {
        connected = false;
        throw new NotImplementedException();
    }

    private void WebSocketOpen(object sender, OpenEventArgs e)
    {
        reconnectCount = 0;
        receiveMsgTime = Time.realtimeSinceStartup;
        connected = true;
        Debug.LogError(string.Format("连接成功,IP:{0}",ip));
    }

    private void WebSocketClose(object sender, CloseEventArgs e)
    {
        connected = false;
        Debug.LogError(string.Format("连接关闭: StatusCode: {0}, Reason: {1}", e.StatusCode, e.Reason));
    }

    float heartTime = 0;
    float receiveMsgTime = 0;
    float reconnectCount = 0;
    private void Update()
    {
        if (!inited)
        {
            return;
        }
        heartTime = heartTime + Time.deltaTime;
        if (heartTime >= 5.0f && connected == true)
        {
            heartTime = 0.0f;
            Send(SocketCode.Heart, "heartbeat",null);
        }
        if (receiveMsgTime != 0 &&Time.realtimeSinceStartup - receiveMsgTime > 6.0f)
        {
            if (reconnectCount > 5)
            {
                Debug.LogError("重新连接失败！");
                return;
            }
            receiveMsgTime = Time.realtimeSinceStartup;
            connected = false;
            webSocket.CloseAsync();
            Connect();
            reconnectCount = reconnectCount + 1;
            Debug.LogError(string.Format("连接断开,尝试第{0}次重新连接！", reconnectCount));
        }
    }

    public void Clear()
    {
        connected = false;
        inited = false;
        msgResponseDic.Clear();
        msgListenerDic.Clear();
    }
}