ProtoWriter: Add ReadRawBytes(Span) methods

KonataDev · Mar 26, 2024 · 0d1b421 · 0d1b421
1 parent b296a31
commit 0d1b421
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/KonataNT.Proto/Serialization/ProtoReader.cs b/KonataNT.Proto/Serialization/ProtoReader.cs
@@ -119,6 +119,22 @@ private unsafe T ReadRawIntegerUnchecked<T>(ref T ptr, nuint position) where T :
         return result;
     }
 
+    public ReadOnlySpan<byte> ReadRawBytesSpan(uint length)
+    {
+        if (Length - _position < length)
+        {
+            ThrowMalformedMessage();
+        }
+        var span = MemoryMarshal.CreateReadOnlySpan(ref Unsafe.Add(ref First, _position), (int)length);
+        _position += length;
+        return span;
+    }
+
+    public byte[] ReadRawBytes(uint length)
+    {
+        return ReadRawBytesSpan(length).ToArray();
+    }
+
     [DoesNotReturn]
     public static void ThrowMalformedMessage()
     {