feat: adding support for Span<T>

MirageNet · Mar 12, 2024 · 3c48db8 · 3c48db8
1 parent 68ba92c
commit 3c48db8
Showing 1 changed file with 30 additions and 2 deletions.
diff --git a/Assets/Mirage/Runtime/Serialization/CollectionExtensions.cs b/Assets/Mirage/Runtime/Serialization/CollectionExtensions.cs
@@ -90,7 +90,7 @@ public static void WriteArraySegment<T>(this NetworkWriter writer, ArraySegment<
         }
 
         [WeaverSerializeCollection]
-        public static void WriteGodotDictionary<TKey, TValue>(this NetworkWriter writer, Dictionary<TKey, TValue> dictionary)
+        public static void WriteDictionary<TKey, TValue>(this NetworkWriter writer, Dictionary<TKey, TValue> dictionary)
         {
             WriteCountPlusOne(writer, dictionary?.Count);
 
@@ -104,6 +104,17 @@ public static void WriteArraySegment<T>(this NetworkWriter writer, ArraySegment<
             }
         }
 
+        [WeaverSerializeCollection]
+        public static void WriteSpan<T>(this NetworkWriter writer, Span<T> span) => WriteReadOnlySpan<T>(writer, span);
+        [WeaverSerializeCollection]
+        public static void WriteReadOnlySpan<T>(this NetworkWriter writer, ReadOnlySpan<T> span)
+        {
+            var length = span.Length;
+            writer.WritePackedUInt32(checked((uint)length));
+            for (var i = 0; i < length; i++)
+                writer.Write(span[i]);
+        }
+
         /// <returns>array or null</returns>
         public static byte[] ReadBytesAndSize(this NetworkReader reader)
         {
@@ -174,7 +185,7 @@ public static ArraySegment<T> ReadArraySegment<T>(this NetworkReader reader)
         }
 
         [WeaverSerializeCollection]
-        public static Dictionary<TKey, TValue> ReadGodotDictionary<TKey, TValue>(this NetworkReader reader)
+        public static Dictionary<TKey, TValue> ReadDictionary<TKey, TValue>(this NetworkReader reader)
         {
             var hasValue = ReadCountPlusOne(reader, out var length);
             if (!hasValue)
@@ -192,6 +203,23 @@ public static ArraySegment<T> ReadArraySegment<T>(this NetworkReader reader)
             return result;
         }
 
+        [WeaverSerializeCollection]
+        public static Span<T> ReadSpan<T>(this NetworkReader reader)
+        {
+            var length = checked((int)reader.ReadPackedUInt32());
+            ValidateSize(reader, length);
+
+            var result = new T[length]; // todo remove allocation
+            for (var i = 0; i < length; i++)
+            {
+                result[i] = reader.Read<T>();
+            }
+            return result;
+        }
+        [WeaverSerializeCollection]
+        public static ReadOnlySpan<T> ReadReadOnlySpan<T>(this NetworkReader reader) => ReadSpan<T>(reader);
+
+
         /// <summary>Writes null as 0, and all over values as +1</summary>
         /// <param name="count">The real count or null if collection is is null</param>
         internal static void WriteCountPlusOne(NetworkWriter writer, int? count)