alap

Assets/Mirror/Core/NetworkReaderPool.cs

@@ -0,0 +1,45 @@
+// API consistent with Microsoft's ObjectPool<T>.
+using System;
+using System.Runtime.CompilerServices;
+
+namespace Mirror
+{
+    /// <summary>Pool of NetworkReaders to avoid allocations.</summary>
+    public static class NetworkReaderPool
+    {
+        // reuse Pool<T>
+        // we still wrap it in NetworkReaderPool.Get/Recyle so we can reset the
+        // position and array before reusing.
+        static readonly Pool<NetworkReaderPooled> Pool = new Pool<NetworkReaderPooled>(
+            // byte[] will be assigned in GetReader
+            () => new NetworkReaderPooled(new byte[]{}),
+            // initial capacity to avoid allocations in the first few frames
+            1000
+        );
+
+        /// <summary>Get the next reader in the pool. If pool is empty, creates a new Reader</summary>
+        public static NetworkReaderPooled Get(byte[] bytes)
+        {
+            // grab from pool & set buffer
+            NetworkReaderPooled reader = Pool.Get();
+            reader.SetBuffer(bytes);
+            return reader;
+        }
+
+        /// <summary>Get the next reader in the pool. If pool is empty, creates a new Reader</summary>
+        public static NetworkReaderPooled Get(ArraySegment<byte> segment)
+        {
+            // grab from pool & set buffer
+            NetworkReaderPooled reader = Pool.Get();
+            reader.SetBuffer(segment);
+            return reader;
+        }
+
+        /// <summary>Returns a reader to the pool.</summary>
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public static void Return(NetworkReaderPooled reader)
+        {
+            Pool.Return(reader);
+        }
+    }
+}