Fix server crash on client disconnect due to incomplete entity removal

Ralith · Sep 10, 2023 · 1f5780b · 1f5780b
1 parent 27f0dab
commit 1f5780b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/server/src/sim.rs b/server/src/sim.rs
@@ -163,6 +163,9 @@ impl Sim {
     pub fn destroy(&mut self, entity: Entity) {
         let id = *self.world.get::<&EntityId>(entity).unwrap();
         self.entity_ids.remove(&id);
+        if let Ok(position) = self.world.get::<&Position>(entity) {
+            self.graph_entities.remove(position.node, entity);
+        }
         self.world.despawn(entity).unwrap();
         self.despawns.push(id);
     }