Fixed a bug not able to reconnect

inworld-ai · Dec 15, 2023 · a3c6c2d · a3c6c2d
1 parent b46a0f8
commit a3c6c2d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/Assets/Inworld/Inworld.AI/Scripts/InworldWebSocketClient.cs b/Assets/Inworld/Inworld.AI/Scripts/InworldWebSocketClient.cs
@@ -264,11 +264,13 @@ string _GetSessionFullName(string sceneFullName)
         protected IEnumerator _StartSession()
         {
             string url = m_ServerConfig.SessionURL(m_Token.sessionId);
-            if (!IsTokenValid || WebSocketManager.Instance.Contains(url))
+            if (!IsTokenValid)
                 yield break;
             yield return new WaitForEndOfFrame();
             string[] param = {m_Token.type, m_Token.token};
-            m_Socket = new WebSocket(url, param);
+            m_Socket = WebSocketManager.Instance.GetWebSocket(url);
+            if (m_Socket == null)
+                m_Socket = new WebSocket(url, param);
             m_Socket.OnOpen += OnSocketOpen;
             m_Socket.OnMessage += OnMessageReceived;
             m_Socket.OnClose += OnSocketClosed;

diff --git a/...orld/Inworld.AI/UnityWebSocket/Scripts/Runtime/Implementation/NoWebGL/WebSocketManager.cs b/...orld/Inworld.AI/UnityWebSocket/Scripts/Runtime/Implementation/NoWebGL/WebSocketManager.cs
@@ -35,6 +35,7 @@ public static void CreateInstance()
         readonly List<WebSocket> sockets = new List<WebSocket>();
 
         public bool Contains(string sessionURL) => sockets.Any(s => s.Address == sessionURL);
+        public WebSocket GetWebSocket(string sessionURL) => sockets.FirstOrDefault(s => s.Address == sessionURL);
         public void Add(WebSocket socket)
         {
             if (!Contains(socket.Address))