[entity] fix use storeAlias to nullify (#69)

pjechris · Jul 4, 2024 · e55e794 · e55e794
1 parent 055a2d2
commit e55e794
Showing 1 changed file with 7 additions and 17 deletions.
diff --git a/Sources/CohesionKit/EntityStore.swift b/Sources/CohesionKit/EntityStore.swift
@@ -200,7 +200,7 @@ public class EntityStore {
         return node
     }
 
-    private func storeAlias<T>(content: T, key: AliasKey<T>, modifiedAt: Stamp?) {
+    private func storeAlias<T>(content: T?, key: AliasKey<T>, modifiedAt: Stamp?) {
         let aliasNode = refAliases[safe: key, onChange: registry.enqueueChange(for:)]
         let aliasContainer = AliasContainer(key: key, content: content)
 
@@ -350,29 +350,19 @@ extension EntityStore {
     /// Removes an alias from the storage
     public func removeAlias<T>(named: AliasKey<T>) {
         transaction {
-            if let alias = refAliases[named] {
-                do {
-                    try alias.updateEntity(AliasContainer(key: named, content: nil), modifiedAt: nil)
-                    logger?.didUnregisterAlias(named)
-                }
-                catch {
-
-                }
+            if refAliases[named] != nil {
+                storeAlias(content: nil, key: named, modifiedAt: nil)
+                logger?.didUnregisterAlias(named)
             }
         }
     }
 
     /// Removes an alias from the storage
     public func removeAlias<C: Collection>(named: AliasKey<C>) {
         transaction {
-            if let alias = refAliases[named] {
-                do {
-                    try alias.updateEntity(AliasContainer(key: named, content: nil), modifiedAt: nil)
-                    logger?.didUnregisterAlias(named)
-                }
-                catch {
-
-                }
+            if refAliases[named] != nil {
+                storeAlias(content: nil, key: named, modifiedAt: nil)
+                logger?.didUnregisterAlias(named)
             }
         }