Loop style.

Gremlinq · Mar 14, 2024 · e9f10e9 · e9f10e9
1 parent ad68be4
commit e9f10e9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/Providers.Core/Extensions/ClientWebSocketExtensions.cs b/src/Providers.Core/Extensions/ClientWebSocketExtensions.cs
@@ -13,23 +13,23 @@ public static async Task<MemoryOwner<byte>> ReceiveAsync(this ClientWebSocket cl
 
             try
             {
-                while (true)
+                ValueWebSocketReceiveResult result;
+
+                do
                 {
                     ct.ThrowIfCancellationRequested();
 
                     if (read == bytes.Memory.Length)
                         bytes = bytes.Double();
 
-                    var result = await client.ReceiveAsync(bytes.Memory[read..], ct);
+                    result = await client.ReceiveAsync(bytes.Memory[read..], ct);
 
                     if (result.MessageType == WebSocketMessageType.Close)
                         throw new ObjectDisposedException(client.GetType().Name);
 
                     read += result.Count;
-
-                    if (result.EndOfMessage)
-                        break;
                 }
+                while (!result.EndOfMessage);
 
                 return bytes[..read];
             }