Small refactor

src/deferred_system.rs

@@ -7,6 +7,12 @@ use bevy::{
 };
 use std::sync::Arc;
 
+#[derive(Resource, Default)]
+pub struct DeferredSystemRegistry {
+    pub ref_counts: HashMap<SystemId, Arc<()>>,
+    pub run_queue: Box<Vec<SystemId>>,
+}
+
 #[derive(Clone, Copy)]
 pub struct DeferredSystem {
     id: SystemId,
@@ -14,7 +20,15 @@ pub struct DeferredSystem {
 }
 
 impl DeferredSystem {
-    pub(crate) fn new<S>(system: S, world: &mut World) -> (Self, Arc<()>)
+    pub fn schedule(&self) {
+        unsafe { &mut *self.run_queue }.push(self.id);
+    }
+}
+
+unsafe impl Send for DeferredSystem {}
+unsafe impl Sync for DeferredSystem {}
+
+pub fn new_deferred_system<S>(system: S, world: &mut World) -> (DeferredSystem, Arc<()>)
 where
     S: IntoSystem<(), (), ()> + 'static,
 {
@@ -26,23 +40,10 @@ impl DeferredSystem {
         .ref_counts
         .insert(id, Arc::clone(&ref_count));
 
-        let deferred_system = Self {
+    let deferred_system = DeferredSystem {
         id,
         run_queue: Box::as_mut(&mut system_registry.run_queue),
     };
+
     (deferred_system, ref_count)
 }
-
-    pub fn schedule(&self) {
-        unsafe { &mut *self.run_queue }.push(self.id);
-    }
-}
-
-unsafe impl Send for DeferredSystem {}
-unsafe impl Sync for DeferredSystem {}
-
-#[derive(Resource, Default)]
-pub struct DeferredSystemRegistry {
-    pub ref_counts: HashMap<SystemId, Arc<()>>,
-    pub run_queue: Box<Vec<SystemId>>,
-}

src/hooks.rs

@@ -1,4 +1,7 @@
-use crate::{deferred_system::DeferredSystem, tick::EcsContext};
+use crate::{
+    deferred_system::{new_deferred_system, DeferredSystem},
+    tick::EcsContext,
+};
 use bevy::ecs::{
     system::{IntoSystem, Resource},
     world::World,
@@ -26,7 +29,7 @@ impl DioxusUiHooks for ScopeState {
     where
         S: IntoSystem<(), (), ()> + 'static,
     {
-        self.use_hook(|| DeferredSystem::new(system, EcsContext::get_world(self)))
+        self.use_hook(|| new_deferred_system(system, EcsContext::get_world(self)))
             .0
     }
 }

src/lib.rs

@@ -16,7 +16,7 @@ use bevy::{
 };
 use dioxus_core::ElementId;
 
-pub use self::hooks::DioxusUiHooks;
+pub use self::{deferred_system::DeferredSystem, hooks::DioxusUiHooks};
 pub use dioxus_core::{Element, Scope};
 
 pub struct DioxusUiPlugin;