Added periodic check initialization in init() function

Author: ElFantasma

examples/updater/src/main.rs

@@ -8,22 +8,17 @@ mod server;
 
 use std::{thread, time::Duration};
 
-use messages::UpdaterOutMessage;
 use server::{UpdateServerState, UpdaterServer};
 use spawned_concurrency::tasks::GenServer as _;
 use spawned_rt::tasks as rt;
 
 fn main() {
     rt::run(async {
-        let mut update_server = UpdaterServer::start(UpdateServerState {
+        UpdaterServer::start(UpdateServerState {
             url: "https://httpbin.org/ip".to_string(),
             periodicity: Duration::from_millis(1000),
         });
 
-        let result = UpdaterServer::check(&mut update_server).await;
-        tracing::info!("Update check done: {result:?}");
-        assert_eq!(result, UpdaterOutMessage::Ok);
-
         // giving it some time before ending
         thread::sleep(Duration::from_secs(10));
     })

examples/updater/src/server.rs

@@ -15,15 +15,6 @@ pub struct UpdateServerState {
 }
 pub struct UpdaterServer {}
 
-impl UpdaterServer {
-    pub async fn check(server: &mut UpdateServerHandle) -> OutMessage {
-        match server.cast(InMessage::Check).await {
-            Ok(_) => OutMessage::Ok,
-            Err(_) => OutMessage::Error,
-        }
-    }
-}
-
 impl GenServer for UpdaterServer {
     type CallMsg = ();
     type CastMsg = InMessage;
@@ -35,6 +26,16 @@ impl GenServer for UpdaterServer {
         Self {}
     }
 
+    // Initializing GenServer to start periodic checks
+    async fn init(
+        &mut self,
+        handle: &GenServerHandle<Self>,
+        state: Self::State,
+    ) -> Result<Self::State, Self::Error> {
+        send_after(state.periodicity, handle.clone(), InMessage::Check);
+        Ok(state)
+    }
+
     async fn handle_call(
         &mut self,
         _message: Self::CallMsg,

examples/updater_threads/src/main.rs

@@ -8,22 +8,17 @@ mod server;
 
 use std::{thread, time::Duration};
 
-use messages::UpdaterOutMessage;
 use server::{UpdateServerState, UpdaterServer};
 use spawned_concurrency::threads::GenServer as _;
 use spawned_rt::threads as rt;
 
 fn main() {
     rt::run(|| {
-        let mut update_server = UpdaterServer::start(UpdateServerState {
+        UpdaterServer::start(UpdateServerState {
             url: "https://httpbin.org/ip".to_string(),
             periodicity: Duration::from_millis(1000),
         });
 
-        let result = UpdaterServer::check(&mut update_server);
-        tracing::info!("Update check done: {result:?}");
-        assert_eq!(result, UpdaterOutMessage::Ok);
-
         // giving it some time before ending
         thread::sleep(Duration::from_secs(10));
     })

examples/updater_threads/src/server.rs

@@ -16,15 +16,6 @@ pub struct UpdateServerState {
 }
 pub struct UpdaterServer {}
 
-impl UpdaterServer {
-    pub fn check(server: &mut UpdateServerHandle) -> OutMessage {
-        match server.cast(InMessage::Check) {
-            Ok(_) => OutMessage::Ok,
-            Err(_) => OutMessage::Error,
-        }
-    }
-}
-
 impl GenServer for UpdaterServer {
     type CallMsg = ();
     type CastMsg = InMessage;
@@ -36,6 +27,16 @@ impl GenServer for UpdaterServer {
         Self {}
     }
 
+    // Initializing GenServer to start periodic checks.
+    fn init(
+        &mut self,
+        handle: &GenServerHandle<Self>,
+        state: Self::State,
+    ) -> Result<Self::State, Self::Error> {
+        send_after(state.periodicity, handle.clone(), InMessage::Check);
+        Ok(state)
+    }
+
     fn handle_call(
         &mut self,
         _message: Self::CallMsg,