Attempt to remove component from render world if not extracted.

bevyengine · Oct 14, 2024 · 9ec9c7b · 9ec9c7b
1 parent d7b2713
commit 9ec9c7b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/crates/bevy_render/src/extract_component.rs b/crates/bevy_render/src/extract_component.rs
@@ -205,6 +205,8 @@ fn extract_components<C: ExtractComponent>(
     for (entity, query_item) in &query {
         if let Some(component) = C::extract_component(query_item) {
             values.push((entity, component));
+        } else {
+            commands.entity(entity).remove::<C>();
         }
     }
     *previous_len = values.len();
@@ -222,6 +224,8 @@ fn extract_visible_components<C: ExtractComponent>(
         if view_visibility.get() {
             if let Some(component) = C::extract_component(query_item) {
                 values.push((entity, component));
+            } else {
+                commands.entity(entity).remove::<C>();
             }
         }
     }