add disconnect event

hiram3512 · Sep 2, 2018 · 591f6db · 591f6db
1 parent a26ba75
commit 591f6db
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/HiSocket/TcpConnection/Socket/TcpSocket.cs b/src/HiSocket/TcpConnection/Socket/TcpSocket.cs
@@ -211,7 +211,14 @@ private void EndSend(IAsyncResult ar)
             Array.Copy(SendBuffer.Array, SendBuffer.ReadPosition, sendBytes, 0, sendBytes.Length);
             SocketSendEvent(sendBytes);
             SendBuffer.MoveReadPosition(length);
-            Send();
+            if (length > 0)
+            {
+                Send();
+            }
+            else
+            {
+                DisconnectedEvnet();
+            }
         }
 
         private void Receive()
@@ -247,7 +254,14 @@ private void EndReceive(IAsyncResult ar)
             ReceiveBuffer.MoveWritePosition(length);
             var bytes = ReceiveBuffer.ReadAll();
             SocketReceiveEvent(bytes);
-            Receive();
+            if (length > 0)
+            {
+                Receive();
+            }
+            else
+            {
+                DisconnectedEvnet();
+            }
         }
 
         public void DisConnect()