fix buffer allocs for benchmark (#16)

petabridge · Apr 15, 2024 · 69b3e61 · 69b3e61
1 parent 074d4be
commit 69b3e61
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/benchmarks/TurboMqtt.Core.Benchmarks/Mqtt311/Mqtt311ConnectCodecBenchmarks.cs b/benchmarks/TurboMqtt.Core.Benchmarks/Mqtt311/Mqtt311ConnectCodecBenchmarks.cs
@@ -42,6 +42,7 @@ public class Mqtt311ConnectCodecBenchmarks
         }
     };
 
+    private byte[] _writeableBytes;
     private ReadOnlyMemory<byte> _encodedConnectPacket;
     private int _estimatedConnectPacketSize;
     private int _estimatedHeaderSize;
@@ -52,6 +53,7 @@ public void Setup()
         var estimate = MqttPacketSizeEstimator.EstimateMqtt3PacketSize(_connectPacket);
         var headerSize =
             MqttPacketSizeEstimator.GetPacketLengthHeaderSize(estimate) + 1; // add 1 for the lead byte
+        _writeableBytes = new byte[estimate + headerSize];
         var memory = new Memory<byte>(new byte[estimate + headerSize]);
         _encodedConnectPacket = memory;
         Mqtt311Encoder.EncodePacket(_connectPacket, ref memory, estimate);
@@ -64,7 +66,7 @@ public void Setup()
     [IterationSetup]
     public void IterationSetup()
     {
-        _writeableBuffer = new Memory<byte>(new byte[_estimatedConnectPacketSize + _estimatedHeaderSize]);
+        _writeableBuffer = new Memory<byte>(_writeableBytes);
     }
 
     [Benchmark]