add get_resource_unchecked_mut_by_id

Author: jakobhellermann

crates/bevy_ecs/src/world/mod.rs

@@ -1450,6 +1450,25 @@ impl World {
     /// use this in cases where the actual types are not known at compile time.**
     #[inline]
     pub fn get_resource_mut_by_id(&mut self, component_id: ComponentId) -> Option<MutUntyped<'_>> {
+        // SAFETY: unique world access
+        unsafe { self.get_resource_unchecked_mut_by_id(component_id) }
+    }
+
+    /// Gets a resource to the resource with the id [`ComponentId`] if it exists.
+    /// The returned pointer may be used to modify the resource, as long as the mutable borrow
+    /// of the [`World`] is still valid.
+    ///
+    /// **You should prefer to use the typed API [`World::get_resource_mut`] where possible and only
+    /// use this in cases where the actual types are not known at compile time.**
+    ///
+    /// # Safety
+    /// This will allow aliased mutable access to the given resource type. The caller must ensure
+    /// that there is either only one mutable access or multiple immutable accesses at a time.
+    #[inline]
+    pub unsafe fn get_resource_unchecked_mut_by_id(
+        &self,
+        component_id: ComponentId,
+    ) -> Option<MutUntyped<'_>> {
         let info = self.components.get_info(component_id)?;
         if !info.is_send_and_sync() {
             self.validate_non_send_access_untyped(info.name());
@@ -1462,14 +1481,14 @@ impl World {
             // SAFETY:
             // - index is in-bounds because the column is initialized and non-empty
             // - no other reference to the ticks of the same row can exist at the same time
-            component_ticks: unsafe { ticks.deref_mut() },
+            component_ticks: ticks.deref_mut(),
             last_change_tick: self.last_change_tick(),
             change_tick: self.read_change_tick(),
         };
 
         Some(MutUntyped {
             // SAFETY: This function has exclusive access to the world so nothing aliases `ptr`.
-            value: unsafe { ptr.assert_unique() },
+            value: ptr.assert_unique(),
             ticks,
         })
     }
@@ -1524,7 +1543,7 @@ impl World {
         component_id: ComponentId,
     ) -> Option<MutUntyped<'_>> {
         self.components().get_info(component_id)?;
-        // SAFETY: entity_location is valid, component_id is valid as checked by the line above
+        // SAFETY: entity_location is valid, component_id is valid as checked by the line above, world access is unique
         unsafe {
             get_mut_by_id(
                 self,