fix: recycling segments

If I ask for a segment of size n, it should have a buffer of at least n
MirageNet · Oct 18, 2020 · 9d12658 · 9d12658
1 parent e79d6c4
commit 9d12658
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/Assets/Mirror/Runtime/Transport/Kcp/ByteBuffer.cs b/Assets/Mirror/Runtime/Transport/Kcp/ByteBuffer.cs
@@ -1,5 +1,4 @@
 using System;
-using System.Collections.Generic;
 
 namespace Mirror.KCP
 {
@@ -76,6 +75,11 @@ public void WriteBytes(byte[] bytes, int startIndex, int length)
             writeIndex = total;
         }
 
+        public void Resize(int length)
+        {
+            FixSizeAndReset(RawBuffer.Length, length);
+        }
+
         public int ReaderIndex { get; private set; }
 
         public int ReadableBytes => writeIndex - ReaderIndex;

diff --git a/Assets/Mirror/Runtime/Transport/Kcp/Segment.cs b/Assets/Mirror/Runtime/Transport/Kcp/Segment.cs
@@ -27,7 +27,9 @@ public static Segment Get(int size)
         {
             if (msSegmentPool.Count > 0)
             {
-                return msSegmentPool.Pop();
+                Segment buffer =  msSegmentPool.Pop();
+                buffer.data.Resize(size);
+                return buffer;
             }
             return new Segment(size);
         }