Stop server immediately as well

Author: Shatur

bevy_replicon_example_backend/src/server.rs

@@ -20,7 +20,6 @@ impl Plugin for RepliconExampleServerPlugin {
         app.add_systems(
             PreUpdate,
             (
-                set_stopped.run_if(resource_removed::<ExampleServer>),
                 set_running.run_if(resource_added::<ExampleServer>),
                 receive_packets.run_if(resource_exists::<ExampleServer>),
             )
@@ -29,9 +28,14 @@ impl Plugin for RepliconExampleServerPlugin {
         )
         .add_systems(
             PostUpdate,
-            send_packets
-                .in_set(ServerSet::SendPackets)
-                .run_if(resource_exists::<ExampleServer>),
+            (
+                set_stopped
+                    .before(ServerSet::Send)
+                    .run_if(resource_removed::<ExampleServer>),
+                send_packets
+                    .in_set(ServerSet::SendPackets)
+                    .run_if(resource_exists::<ExampleServer>),
+            ),
         );
     }
 }

src/shared/backend/replicon_server.rs

@@ -5,7 +5,11 @@ use log::{debug, trace, warn};
 /// Stores information about the server independent from the messaging backend.
 ///
 /// The messaging backend is responsible for updating this resource:
-/// - When the server is started or stopped, [`Self::set_running`] should be used to reflect this.
+/// - When the server starts or stops, [`Self::set_running`] should be used to update its status:
+///   - Set to `true` in [`ServerSet::ReceivePackets`](crate::server::ServerSet::ReceivePackets) to
+///     allow immediate processing of received messages.
+///   - Set to `false` before [`ServerSet::Send`](crate::server::ServerSet::Send) to ensure
+///     the server stops immediately on the same frame.
 /// - For receiving messages, [`Self::insert_received`] should be used.
 ///   A system to forward messages from the backend to Replicon should run in [`ServerSet::ReceivePackets`](crate::server::ServerSet::ReceivePackets).
 /// - For sending messages, [`Self::drain_sent`] should be used to drain all sent messages.