improve the internal close

Paulyoufu · May 31, 2017 · 07feb51 · 07feb51
1 parent e1d09b4
commit 07feb51
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Cowboy/Cowboy.Sockets/Tcp/Client/APM/TcpSocketClient.cs b/Cowboy/Cowboy.Sockets/Tcp/Client/APM/TcpSocketClient.cs
@@ -247,6 +247,12 @@ private void HandleTcpServerConnected()
                     Close(true); // user side handle tcp connection error occurred
                 }
             }
+            catch (TimeoutException ex)
+            {
+                _log.Error(ex.Message, ex);
+                Close(false); // timeout exception
+                throw;
+            }
             catch (Exception ex)
             {
                 _log.Error(ex.Message, ex);
@@ -374,7 +380,7 @@ private void HandleDataReceived(IAsyncResult ar)
             if (this.State != TcpSocketConnectionState.Connected
                 || _stream == null)
             {
-                Close(false);
+                Close(false); // receive buffer callback
                 return;
             }