From 9a67ac5341e2b3ef753102d067c37362de132cf4 Mon Sep 17 00:00:00 2001
From: BrennanConroy <brecon@microsoft.com>
Date: Fri, 15 Sep 2017 08:49:18 -0700
Subject: [PATCH] Make sure socket closed is logged

---
 .../Internal/Transports/WebSocketsTransport.cs         | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/Microsoft.AspNetCore.Sockets.Http/Internal/Transports/WebSocketsTransport.cs b/src/Microsoft.AspNetCore.Sockets.Http/Internal/Transports/WebSocketsTransport.cs
index 4d93111c3e..6ac1755f3e 100644
--- a/src/Microsoft.AspNetCore.Sockets.Http/Internal/Transports/WebSocketsTransport.cs
+++ b/src/Microsoft.AspNetCore.Sockets.Http/Internal/Transports/WebSocketsTransport.cs
@@ -51,9 +51,15 @@ namespace Microsoft.AspNetCore.Sockets.Internal.Transports
             {
                 _logger.SocketOpened(_connection.ConnectionId);
 
-                await ProcessSocketAsync(ws);
+                try
+                {
+                    await ProcessSocketAsync(ws);
+                }
+                finally
+                {
+                    _logger.SocketClosed(_connection.ConnectionId);
+                }
             }
-            _logger.SocketClosed(_connection.ConnectionId);
         }
 
         public async Task ProcessSocketAsync(WebSocket socket)