IntoIterator for RemovedComponents

Author: Aceeri

crates/bevy_ecs/src/removal_detection.rs

@@ -122,3 +122,15 @@ impl<'w, 's, T: Component> RemovedComponents<'w, 's, T> {
         }
     }
 }
+
+impl<'a, 'w, 's: 'a, T> IntoIterator for &'a mut RemovedComponents<'w, 's, T> 
+where
+    T: Component,
+{
+    type Item = Entity;
+    type IntoIter = Box<dyn Iterator<Item = Entity> + 'a>;
+    fn into_iter(self) -> Self::IntoIter {
+        self.iter()
+    }
+}
+

examples/ecs/removal_detection.rs

@@ -54,7 +54,7 @@ fn remove_component(
 fn react_on_removal(mut removed: RemovedComponents<MyComponent>, mut query: Query<&mut Sprite>) {
     // `RemovedComponents<T>::iter()` returns an interator with the `Entity`s that had their
     // `Component` `T` (in this case `MyComponent`) removed at some point earlier during the frame.
-    for entity in removed.iter() {
+    for entity in &mut removed {
         if let Ok(mut sprite) = query.get_mut(entity) {
             sprite.color.set_r(0.0);
         }