register actor with state

Author: EnvOut

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "uactor"
-version = "0.13.4"
+version = "0.14.1"
 edition = "2021"
 repository = "https://github.com/EnvOut/uactor"
 license = "MIT"

src/uactor/Cargo.toml

@@ -1,7 +1,6 @@
 use crate::actor::context::ActorContext;
 use crate::actor::message::Message;
 use std::future::Future;
-use std::sync::Arc;
 use crate::actor::select::{ActorSelect, SelectError, SelectResult};
 
 pub trait State: std::any::Any + Send + 'static {}
@@ -18,11 +17,7 @@ pub trait Actor: Sized + Unpin + 'static {
 
     type Inject: Inject + Sized;
 
-    type State: Default + Send + Sync + Clone;
-
-    fn create_state(&mut self) -> Arc<Self::State> {
-        Arc::new(Default::default())
-    }
+    type State: Send + Sync + Clone;
 
     fn on_start(
         &mut self,

src/uactor/src/actor/abstract_actor.rs

@@ -25,11 +25,27 @@ impl GlobalSystem {
     ) -> (R, tokio::sync::mpsc::UnboundedReceiver<M>)
     where
         A: Actor,
+        A::State: Default,
         M: Message + Send + 'static,
         R: From<(ActorName, UnboundedSender<M>, A::State)>,
     {
-        let mut system  = GLOBAL_SYSTEM.write().await;
-        system.register_ref::<A, M , R>(actor_name).await
+        let mut system = GLOBAL_SYSTEM.write().await;
+        system.register_ref::<A, M, R>(actor_name).await
+    }
+
+    pub async fn register_ref_with_state<A, M, R>(
+        &mut self,
+        actor_name: &str,
+        state: A::State,
+    ) -> (R, tokio::sync::mpsc::UnboundedReceiver<M>)
+    where
+        A: Actor,
+        A::State: Default,
+        M: Message + Send + 'static,
+        R: From<(ActorName, UnboundedSender<M>, A::State)>,
+    {
+        let mut system = GLOBAL_SYSTEM.write().await;
+        system.register_ref_with_state::<A, M, R>(actor_name, state).await
     }
 
     pub async fn spawn_actor<A, S>(

src/uactor/src/system/global.rs

@@ -40,9 +40,10 @@ impl System {
         self
     }
 
-    pub async fn register_ref<A, M, R>(
+    pub async fn register_ref_with_state<A, M, R>(
         &mut self,
         actor_name: &str,
+        state: A::State,
     ) -> (R, tokio::sync::mpsc::UnboundedReceiver<M>)
     where
         A: Actor,
@@ -52,7 +53,6 @@ impl System {
         let (tx, rx) = tokio::sync::mpsc::unbounded_channel::<M>();
 
         let actor_name: Arc<str> = actor_name.to_owned().into();
-        let state = A::State::default();
         let old_ref = self.actor_registry.register_ref::<A, M, UnboundedSender<M>>(actor_name.clone(), tx.clone(), state.clone());
         if let Some(_old_ref) = old_ref {
             tracing::warn!("The actor: {actor_name:?} has already been registered, old ref has been replaced");
@@ -61,6 +61,20 @@ impl System {
         (R::from((actor_name, tx, state)), rx)
     }
 
+    pub async fn register_ref<A, M, R>(
+        &mut self,
+        actor_name: &str,
+    ) -> (R, tokio::sync::mpsc::UnboundedReceiver<M>)
+    where
+        A: Actor,
+        A::State: Default,
+        M: Message + Send + 'static,
+        R: From<(ActorName, UnboundedSender<M>, A::State)>,
+    {
+        let state = A::State::default();
+        self.register_ref_with_state::<A, M, R>(actor_name, state).await
+    }
+
     pub async fn spawn_actor<A, S>(
         &mut self,
         actor_name: Arc<str>,