Fix message resize not affecting pending messages

RiptideNetworking · Nov 22, 2023 · caf0344 · caf0344
1 parent c7f0d10
commit caf0344
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/RiptideNetworking/RiptideNetworking/Message.cs b/RiptideNetworking/RiptideNetworking/Message.cs
@@ -69,6 +69,7 @@ public static int MaxPayloadSize
                 maxArraySize = MaxSize / sizeof(ulong) + (MaxSize % sizeof(ulong) == 0 ? 0 : 1);
                 ByteBuffer = new byte[MaxSize];
                 TrimPool(); // When ActiveSocketCount is 0, this clears the pool
+                PendingMessage.ClearPool();
             }
         }
         /// <summary>An intermediary buffer to help convert <see cref="data"/> to a byte array when sending.</summary>

diff --git a/RiptideNetworking/RiptideNetworking/PendingMessage.cs b/RiptideNetworking/RiptideNetworking/PendingMessage.cs
@@ -75,6 +75,12 @@ private static PendingMessage RetrieveFromPool()
             return message;
         }
 
+        /// <summary>Empties the pool. Does not affect <see cref="PendingMessage"/> instances which are actively pending and therefore not in the pool.</summary>
+        public static void ClearPool()
+        {
+            pool.Clear();
+        }
+
         /// <summary>Returns the <see cref="PendingMessage"/> instance to the pool so it can be reused.</summary>
         private void Release()
         {