快速租用移除容器

KumoKyaku · Jan 9, 2024 · c885b66 · c885b66
1 parent 94b8c31
commit c885b66
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/Megumin.Explosion/Collections/Generic/DictionaryExtension.cs b/Megumin.Explosion/Collections/Generic/DictionaryExtension.cs
@@ -278,5 +278,15 @@ public static Remover<K, V> GetForeachRemover<K, V>(this IDictionary<K, V> sourc
         {
             return new Remover<K, V>(source);
         }
+
+        public static List<K> RentRemoveList<K, V>(this IDictionary<K, V> source)
+        {
+            return ListPool<K>.Shared.Rent();
+        }
+
+        public static void ReturnRemoveList<K, V>(this IDictionary<K, V> source, ref List<K> element, bool forceSafeCheck = false)
+        {
+            ListPool<K>.Shared.Return(ref element, forceSafeCheck);
+        }
     }
 }
diff --git a/Megumin.Explosion/Collections/Generic/IEnumerableExtension.cs b/Megumin.Explosion/Collections/Generic/IEnumerableExtension.cs
@@ -171,6 +171,16 @@ public static Remover<T> GetForeachRemover<T>(this ICollection<T> list)
         {
             return new Remover<T>(list);
         }
+
+        public static List<T> RentRemoveList<T>(this ICollection<T> list)
+        {
+            return ListPool<T>.Shared.Rent();
+        }
+
+        public static void ReturnRemoveList<T>(this ICollection<T> list, ref List<T> element, bool forceSafeCheck = false)
+        {
+            ListPool<T>.Shared.Return(ref element, forceSafeCheck);
+        }
     }
 }