MqttPacketInspector: add FillReceiveBuffer(ReadOnlySequence<byte>) me…

…thod.
dotnet · Dec 6, 2024 · f2a81ea · f2a81ea
1 parent 8df3a96
commit f2a81ea
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/Source/MQTTnet/Adapter/MqttPacketInspector.cs b/Source/MQTTnet/Adapter/MqttPacketInspector.cs
@@ -3,6 +3,7 @@
 // See the LICENSE file in the project root for more information.
 
 using System;
+using System.Buffers;
 using System.IO;
 using System.Threading.Tasks;
 using MQTTnet.Diagnostics.Logger;
@@ -81,6 +82,22 @@ public void FillReceiveBuffer(ReadOnlySpan<byte> buffer)
         _receivedPacketBuffer?.Write(buffer);
     }
 
+    public void FillReceiveBuffer(ReadOnlySequence<byte> buffer)
+    {
+        if (!_asyncEvent.HasHandlers)
+        {
+            return;
+        }
+
+        if (_receivedPacketBuffer != null)
+        {
+            foreach (var memory in buffer)
+            {
+                _receivedPacketBuffer.Write(memory.Span);
+            }
+        }
+    }
+
     async Task InspectPacket(byte[] buffer, MqttPacketFlowDirection direction)
     {
         try