Added spawn_on_thread and send_message_on

Author: ElFantasma

concurrency/src/tasks/gen_server.rs

@@ -5,7 +5,10 @@ use crate::{
     tasks::InitResult::{NoSuccess, Success},
 };
 use futures::future::FutureExt as _;
-use spawned_rt::tasks::{self as rt, mpsc, oneshot, timeout, CancellationToken};
+use spawned_rt::{
+    tasks::{self as rt, mpsc, oneshot, timeout, CancellationToken, JoinHandle},
+    threads,
+};
 use std::{fmt::Debug, future::Future, panic::AssertUnwindSafe, time::Duration};
 
 const DEFAULT_CALL_TIMEOUT: Duration = Duration::from_secs(5);
@@ -27,7 +30,7 @@ impl<G: GenServer> Clone for GenServerHandle<G> {
 }
 
 impl<G: GenServer> GenServerHandle<G> {
-    pub(crate) fn new(gen_server: G) -> Self {
+    fn new(gen_server: G) -> Self {
         let (tx, mut rx) = mpsc::channel::<GenServerInMsg<G>>();
         let cancellation_token = CancellationToken::new();
         let handle = GenServerHandle {
@@ -51,7 +54,7 @@ impl<G: GenServer> GenServerHandle<G> {
         handle_clone
     }
 
-    pub(crate) fn new_blocking(gen_server: G) -> Self {
+    fn new_blocking(gen_server: G) -> Self {
         let (tx, mut rx) = mpsc::channel::<GenServerInMsg<G>>();
         let cancellation_token = CancellationToken::new();
         let handle = GenServerHandle {
@@ -70,6 +73,25 @@ impl<G: GenServer> GenServerHandle<G> {
         handle_clone
     }
 
+    fn new_on_thread(gen_server: G) -> Self {
+        let (tx, mut rx) = mpsc::channel::<GenServerInMsg<G>>();
+        let cancellation_token = CancellationToken::new();
+        let handle = GenServerHandle {
+            tx,
+            cancellation_token,
+        };
+        let handle_clone = handle.clone();
+        // Ignore the JoinHandle for now. Maybe we'll use it in the future
+        let _join_handle = threads::spawn(|| {
+            threads::block_on(async move {
+                if let Err(error) = gen_server.run(&handle, &mut rx).await {
+                    tracing::trace!(%error, "GenServer crashed")
+                };
+            })
+        });
+        handle_clone
+    }
+
     pub fn sender(&self) -> mpsc::Sender<GenServerInMsg<G>> {
         self.tx.clone()
     }
@@ -153,6 +175,15 @@ pub trait GenServer: Send + Sized {
         GenServerHandle::new_blocking(self)
     }
 
+    /// For some "singleton" GenServers that run througout the whole execution of the
+    /// program, it makes sense to run in their own dedicated thread to avoid interference
+    /// with the rest of the tasks' runtime.
+    /// The use of tokio::task::spawm_blocking is not recommended for these scenarios
+    /// as it is a limited thread pool better suited for blocking IO tasks that eventually end
+    fn start_on_thread(self) -> GenServerHandle<Self> {
+        GenServerHandle::new_on_thread(self)
+    }
+
     fn run(
         self,
         handle: &GenServerHandle<Self>,
@@ -300,6 +331,36 @@ pub trait GenServer: Send + Sized {
     }
 }
 
+/// Spawns a task that awaits on a future and sends messages to a GenServer.
+///
+/// This function returns a handle to the spawned task.
+pub fn send_message_on<T, U>(
+    handle: GenServerHandle<T>,
+    future: U,
+    message: T::CastMsg,
+) -> JoinHandle<()>
+where
+    T: GenServer,
+    U: Future + Send + 'static,
+    <U as Future>::Output: Send,
+{
+    let cancelation_token = handle.cancellation_token();
+    let mut handle_clone = handle.clone();
+    let join_handle = spawned_rt::tasks::spawn(async move {
+        tracing::info!("Ctrl+C listener started");
+        let is_cancelled = core::pin::pin!(cancelation_token.cancelled());
+        let signal = core::pin::pin!(future);
+        match futures::future::select(is_cancelled, signal).await {
+            futures::future::Either::Left(_) => tracing::error!("GenServer stopped"),
+            futures::future::Either::Right(_) => {
+                tracing::info!("Sending shutdown to PeerTable Server");
+                handle_clone.cast(message).await.unwrap();
+            }
+        }
+    });
+    join_handle
+}
+
 #[cfg(debug_assertions)]
 mod warn_on_block {
     use super::*;

concurrency/src/tasks/mod.rs

@@ -12,8 +12,8 @@ mod stream_tests;
 mod timer_tests;
 
 pub use gen_server::{
-    CallResponse, CastResponse, GenServer, GenServerHandle, GenServerInMsg, InitResult,
-    InitResult::NoSuccess, InitResult::Success,
+    send_message_on, CallResponse, CastResponse, GenServer, GenServerHandle, GenServerInMsg,
+    InitResult, InitResult::NoSuccess, InitResult::Success,
 };
 pub use process::{send, Process, ProcessInfo};
 pub use stream::spawn_listener;

examples/blocking_genserver/main.rs

@@ -99,7 +99,7 @@ impl GenServer for WellBehavedTask {
 pub fn main() {
     rt::run(async move {
         // If we change BadlyBehavedTask to start instead, it can stop the entire program
-        let mut badboy = BadlyBehavedTask::new().start_blocking();
+        let mut badboy = BadlyBehavedTask::new().start_on_thread();
         let _ = badboy.cast(()).await;
         let mut goodboy = WellBehavedTask::new(0).start();
         let _ = goodboy.cast(()).await;