Put register code in separate function

beamable · Feb 14, 2024 · 57d3418 · 57d3418
1 parent 37421ec
commit 57d3418
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/bevy_forge/src/main.rs b/bevy_forge/src/main.rs
@@ -20,7 +20,7 @@ fn main() {
             DefaultPlugins,
             DebugPlugin,
             BeamPlugin,
-            microservice::RequestSayHiPlugin,
+            microservice::MicroservicePlugin,
         ))
         // Never attempts to look up meta files. The default meta configuration will be used for each asset.
         .insert_resource(AssetMetaCheck::Never)

diff --git a/bevy_forge/src/microservice.rs b/bevy_forge/src/microservice.rs
@@ -9,19 +9,25 @@ use bevy::{prelude::*, tasks::IoTaskPool};
 #[cfg(target_family = "wasm")]
 use crossbeam_channel::{bounded, Receiver};
 
-pub struct RequestSayHiPlugin;
+pub struct MicroservicePlugin;
 
-impl Plugin for RequestSayHiPlugin {
+impl Plugin for MicroservicePlugin {
     fn build(&self, app: &mut App) {
-        app.add_event::<RequestSayHiEvent>()
-            .add_event::<RequestSayHiEventCompleted>()
-            .add_systems(Update, (request, handle_request));
+        RequestSayHiEvent::register(app);
     }
 }
 
 #[derive(Event, Deref, DerefMut)]
 pub struct RequestSayHiEvent(pub models::SayHiRequestArgs);
 
+impl RequestSayHiEvent {
+    pub fn register(app: &mut App) {
+        app.add_event::<RequestSayHiEvent>()
+            .add_event::<RequestSayHiEventCompleted>()
+            .add_systems(Update, (request, handle_request));
+    }
+}
+
 #[derive(Event, Deref, DerefMut)]
 pub struct RequestSayHiEventCompleted(
     pub Result<String, beam_microservice::apis::Error<SayHiPostError>>,