Wrap pipeline failures in RecordConnectionFailed

839b8f10 · Nick Craver · 98dc7df2 · 839b8f10
Commit 839b8f10 authored Jul 12, 2018 by Nick Craver
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 2 deletions

PhysicalConnection.cs ...kExchange.Redis/StackExchange/Redis/PhysicalConnection.cs +10 -2

No files found.
--- a/StackExchange.Redis/StackExchange/Redis/PhysicalConnection.cs
+++ b/StackExchange.Redis/StackExchange/Redis/PhysicalConnection.cs
@@ -698,8 +698,16 @@ internal static int WriteRaw(Span<byte> span, long value, bool withLengthPrefix
        internal void WakeWriterAndCheckForThrottle()
        {
-            var flush = _ioPipe.Output.FlushAsync();
+            try
-            if (!flush.IsCompletedSuccessfully) flush.AsTask().Wait();
+            {
+                var flush = _ioPipe.Output.FlushAsync();
+                if (!flush.IsCompletedSuccessfully) flush.AsTask().Wait();
+            }
+            catch (ConnectionResetException ex)
+            {
+                RecordConnectionFailed(ConnectionFailureType.SocketClosed, ex);
+                throw;
+            }
        }
        private static readonly byte[] NullBulkString = Encoding.ASCII.GetBytes("$-1\r\n"), EmptyBulkString = Encoding.ASCII.GetBytes("$0\r\n\r\n");