🐛Registry<T> now prunes null items before returning its list

troublecatstudios · Aug 28, 2021 · de52c76 · de52c76
1 parent 82d10dc
commit de52c76
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/Chonks/Registry.cs b/src/Chonks/Registry.cs
@@ -7,6 +7,7 @@ public class Registry<T> : IRegistry<T> {
         public event ItemRegistered<T> OnItemRegistered;
 
         public T[] List() {
+            Prune();
             return _items.ToArray();
         }
 
@@ -17,6 +18,17 @@ public class Registry<T> : IRegistry<T> {
             }
         }
 
+        private void Prune() {
+            int idx = 0;
+            while (idx < _items.Count) {
+                if (_items[idx] == null) {
+                    _items.RemoveAt(idx);
+                    continue;
+                }
+                idx++;
+            }
+        }
+
         public void Unregister(T instance) {
             _items.Remove(instance);
         }