Catch SocketException

Author: cnblogs-dudu

Enyim.Caching/Memcached/MemcachedNode.cs

@@ -9,6 +9,7 @@
 using System.Diagnostics;
 using System.IO;
 using System.Net;
+using System.Net.Sockets;
 using System.Runtime.Serialization;
 using System.Security;
 using System.Threading;
@@ -825,7 +826,14 @@ protected virtual async Task<IPooledSocketResult> ExecuteOperationAsync(IOperati
                 {
                     _logger.LogError(nameof(MemcachedNode), e);
 
-                    result.Fail("Exception reading response", e);
+                    result.Fail("IOException reading response", e);
+                    return result;
+                }
+                catch (SocketException e)
+                {
+                    _logger.LogError(nameof(MemcachedNode), e);
+
+                    result.Fail("SocketException reading response", e);
                     return result;
                 }
                 finally

Enyim.Caching/Memcached/PooledSocket.cs

@@ -254,9 +254,12 @@ public int ReadByte()
             {
                 return _inputStream.ReadByte();
             }
-            catch (IOException)
+            catch (Exception ex)
             {
-                _isAlive = false;
+                if (ex is IOException || ex is SocketException)
+                {
+                    _isAlive = false;
+                }
 
                 throw;
             }
@@ -270,9 +273,12 @@ public int ReadByteAsync()
             {
                 return _inputStream.ReadByte();
             }
-            catch (IOException)
+            catch (Exception ex)
             {
-                _isAlive = false;
+                if (ex is IOException || ex is SocketException)
+                {
+                    _isAlive = false;
+                }
                 throw;
             }
         }
@@ -296,9 +302,13 @@ public async Task ReadAsync(byte[] buffer, int offset, int count)
                     offset += currentRead;
                     shouldRead -= currentRead;
                 }
-                catch (IOException)
+                catch (Exception ex)
                 {
-                    _isAlive = false;
+                    if (ex is IOException || ex is SocketException)
+                    {
+                        _isAlive = false;
+                    }
+
                     throw;
                 }
             }
@@ -330,9 +340,12 @@ public void Read(byte[] buffer, int offset, int count)
                     offset += currentRead;
                     shouldRead -= currentRead;
                 }
-                catch (IOException)
+                catch (Exception ex)
                 {
-                    _isAlive = false;
+                    if (ex is IOException || ex is SocketException)
+                    {
+                        _isAlive = false;
+                    }
                     throw;
                 }
             }
@@ -369,9 +382,13 @@ public void Write(IList<ArraySegment<byte>> buffers)
                     ThrowHelper.ThrowSocketWriteError(_endpoint, status);
                 }
             }
-            catch (IOException)
+            catch (Exception ex)
             {
-                _isAlive = false;
+                if (ex is IOException || ex is SocketException)
+                {
+                    _isAlive = false;
+                }
+                _logger.LogError(ex, nameof(PooledSocket.Write));
                 throw;
             }
         }
@@ -390,9 +407,12 @@ public async Task WriteAsync(IList<ArraySegment<byte>> buffers)
                     ThrowHelper.ThrowSocketWriteError(_endpoint);
                 }
             }
-            catch (IOException ex)
+            catch (Exception ex)
             {
-                _isAlive = false;
+                if (ex is IOException || ex is SocketException)
+                {
+                    _isAlive = false;
+                }
                 _logger.LogError(ex, nameof(PooledSocket.WriteAsync));
                 throw;
             }