Merge pull request #302 from VelvetThePanda/feat/socket-optimizations

Fix buffer issue from #300
Remora · Apr 20, 2023 · 9eac98d · 9eac98d
2 parents a337681 + b89ba38
commit 9eac98d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Backend/Remora.Discord.Gateway/Transport/WebSocketPayloadTransportService.cs b/Backend/Remora.Discord.Gateway/Transport/WebSocketPayloadTransportService.cs
@@ -229,12 +229,11 @@ public async Task<Result<IPayload>> ReceivePayloadAsync(CancellationToken ct = d
 
         using var bufferWriter = new ArrayPoolBufferWriter<byte>(); // Backed by the ArrayPool; the only allocation is the class itself
 
-        var buffer = bufferWriter.GetMemory(4096);
-
         ValueWebSocketReceiveResult result;
 
         do
         {
+            var buffer = bufferWriter.GetMemory(4096);
             result = await _clientWebSocket.ReceiveAsync(buffer, ct);
 
             if (_clientWebSocket.CloseStatus.HasValue)