Merge pull request #54 from GrahameGW/main

Expose Clear function on EntityTable
chickensoft-games · Oct 28, 2024 · 762b992 · 762b992
2 parents 6bbfbc4 + 51072be
commit 762b992
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/Chickensoft.Collections.Tests/src/EntityTableTest.cs b/Chickensoft.Collections.Tests/src/EntityTableTest.cs
@@ -26,4 +26,20 @@ public void StoresValues() {
     table.Get<string>("a").ShouldBeNull();
     table.Get<object>("b").ShouldNotBeNull();
   }
+
+  [Fact]
+  public void Clears() {
+    var table = new EntityTable();
+
+    table.Set("a", "one");
+    table.Set("b", new object());
+
+    table.Get<string>("a").ShouldBe("one");
+    table.Get<object>("b").ShouldNotBeNull();
+
+    table.Clear();
+
+    table.Get<string>("a").ShouldBeNull();
+    table.Get<object>("b").ShouldBeNull();
+  }
 }
diff --git a/Chickensoft.Collections/src/collections/entity_table/EntityTableOfTId.cs b/Chickensoft.Collections/src/collections/entity_table/EntityTableOfTId.cs
@@ -28,6 +28,11 @@ public void Remove(TId? id) {
     _entities.TryRemove(id, out _);
   }
 
+  /// <summary>
+  /// Clears (but does not dispose) all entities from the table.
+  /// </summary>
+  public void Clear() => _entities.Clear();
+
   /// <summary>
   /// Retrieve an entity from the table.
   /// </summary>