Add try-catch when processing data received

KirisameMarisa · Feb 18, 2024 · 620aeec · 620aeec
1 parent c346505
commit 620aeec
Showing 1 changed file with 15 additions and 11 deletions.
diff --git a/Plugins/Unity/project/Assets/BocchiTracker/Runtime/BocchiTrackerTcpSocket.cs b/Plugins/Unity/project/Assets/BocchiTracker/Runtime/BocchiTrackerTcpSocket.cs
@@ -46,13 +46,13 @@ private IEnumerator Connect(string inIPAddress, int inPort)
             yield break;
         }
 
-        /// <summary>
-        /// Disconnects the socket.
-        /// </summary>
-        public void DisConnect()
+
+        private void OnDestroy()
         {
-            if(IsConnect())
+            if (IsConnect())
+            {
                 socket.Disconnect(true);
+            }
         }
 
         /// <summary>
@@ -94,14 +94,18 @@ private IEnumerator ProcessReceiveData()
             var task = socket.ReceiveAsync(receivedData, SocketFlags.None);
             yield return new WaitUntil(() => task.IsCompleted);
 
-            int bytesRead = task.Result;
-            if (task.Result > 0)
+            try
             {
-                Console.WriteLine("ProcessReceiveData::Success, size=" + bytesRead);
-                List<byte> receivedDataList = new List<byte>(receivedData);
-                receivedDataList.RemoveRange(bytesRead, receivedData.Length - bytesRead);
-                ReciveCallback?.Invoke(receivedDataList);
+                int bytesRead = task.Result;
+                if (task.Result > 0)
+                {
+                    Console.WriteLine("ProcessReceiveData::Success, size=" + bytesRead);
+                    List<byte> receivedDataList = new List<byte>(receivedData);
+                    receivedDataList.RemoveRange(bytesRead, receivedData.Length - bytesRead);
+                    ReciveCallback?.Invoke(receivedDataList);
+                }
             }
+            catch { }
         }
 
         /// <summary>