feat: Add PoisonPill message type

src/actor/actor/actor.rs

@@ -32,7 +32,6 @@ pub trait Actor: Debug + Send + Sync + 'static {
           AutoReceiveMessage::PostRestart => self.post_restart(context_handle).await,
           AutoReceiveMessage::PreStop => self.pre_stop(context_handle).await,
           AutoReceiveMessage::PostStop => self.post_stop(context_handle).await,
-          AutoReceiveMessage::PoisonPill => Ok(()),
           AutoReceiveMessage::Terminated(t) => self.post_child_terminate(context_handle, &t).await,
         },
         _ => self.receive(context_handle.clone(), message_handle).await,

src/actor/context/actor_context.rs

@@ -38,6 +38,7 @@ use crate::actor::message::message_or_envelope::{
   unwrap_envelope_header, unwrap_envelope_sender, wrap_envelope, MessageEnvelope,
 };
 use crate::actor::message::not_influence_receive_timeout::NotInfluenceReceiveTimeoutHandle;
+use crate::actor::message::poison_pill::PoisonPill;
 use crate::actor::message::readonly_message_headers::ReadonlyMessageHeadersHandle;
 use crate::actor::message::receive_timeout::ReceiveTimeout;
 use crate::actor::message::response::ResponseHandle;
@@ -171,7 +172,7 @@ impl ActorContext {
   async fn default_receive(&mut self) -> Result<(), ActorError> {
     let message = self.get_message_handle_opt().await.expect("Failed to retrieve message");
     tracing::debug!("ActorContext::default_receive: message = {:?}", message);
-    if let Some(AutoReceiveMessage::PoisonPill) = message.to_typed::<AutoReceiveMessage>() {
+    if message.to_typed::<PoisonPill>().is_some() {
       let me = self.get_self_opt().await.unwrap();
       self.stop(&me).await;
       Ok(())
@@ -909,10 +910,7 @@ impl StopperPart for ActorContext {
   async fn poison(&mut self, pid: &ExtendedPid) {
     let inner_mg = self.inner.lock().await;
     pid
-      .send_user_message(
-        inner_mg.actor_system.clone(),
-        MessageHandle::new(AutoReceiveMessage::PoisonPill),
-      )
+      .send_user_message(inner_mg.actor_system.clone(), MessageHandle::new(PoisonPill))
       .await;
   }
 

src/actor/context/root_context.rs

@@ -18,10 +18,10 @@ use crate::actor::context::{
   InfoPart, MessagePart, SenderContext, SenderPart, SpawnerContext, SpawnerPart, StopperPart,
 };
 use crate::actor::future::{Future, FutureProcess};
-use crate::actor::message::auto_receive_message::AutoReceiveMessage;
 use crate::actor::message::message_handle::MessageHandle;
 use crate::actor::message::message_headers::MessageHeaders;
 use crate::actor::message::message_or_envelope::MessageEnvelope;
+use crate::actor::message::poison_pill::PoisonPill;
 use crate::actor::message::readonly_message_headers::ReadonlyMessageHeadersHandle;
 use crate::actor::message::system_message::SystemMessage;
 use crate::actor::message::watch::Watch;
@@ -249,10 +249,7 @@ impl StopperPart for RootContext {
 
   async fn poison(&mut self, pid: &ExtendedPid) {
     pid
-      .send_user_message(
-        self.get_actor_system().await.clone(),
-        MessageHandle::new(AutoReceiveMessage::PoisonPill),
-      )
+      .send_user_message(self.get_actor_system().await.clone(), MessageHandle::new(PoisonPill))
       .await
   }
 

src/actor/message.rs

@@ -11,6 +11,7 @@ pub mod message_or_envelope;
 mod message_or_envelope_test;
 pub mod messages;
 pub mod not_influence_receive_timeout;
+pub mod poison_pill;
 pub mod readonly_message_headers;
 pub mod receive_timeout;
 pub mod response;

src/actor/message/auto_receive_message.rs

@@ -14,7 +14,6 @@ pub enum AutoReceiveMessage {
   PostRestart,
   PreStop,
   PostStop,
-  PoisonPill,
   Terminated(TerminateInfo),
 }
 
@@ -41,7 +40,6 @@ impl Display for AutoReceiveMessage {
       AutoReceiveMessage::PostRestart => write!(f, "PostRestart"),
       AutoReceiveMessage::PreStop => write!(f, "PreStop"),
       AutoReceiveMessage::PostStop => write!(f, "PostStop"),
-      AutoReceiveMessage::PoisonPill => write!(f, "PoisonPill"),
       AutoReceiveMessage::Terminated(_) => write!(f, "Terminated"),
     }
   }
@@ -57,7 +55,6 @@ impl Message for AutoReceiveMessage {
       (AutoReceiveMessage::PostRestart, Some(&AutoReceiveMessage::PostRestart)) => true,
       (AutoReceiveMessage::PreStop, Some(&AutoReceiveMessage::PreStop)) => true,
       (AutoReceiveMessage::PostStop, Some(&AutoReceiveMessage::PostStop)) => true,
-      (AutoReceiveMessage::PoisonPill, Some(&AutoReceiveMessage::PoisonPill)) => true,
       (AutoReceiveMessage::Terminated(me), Some(&AutoReceiveMessage::Terminated(ref you))) => *me == *you,
       _ => false,
     }

src/actor/message/poison_pill.rs

@@ -0,0 +1,15 @@
+use crate::actor::message::message::Message;
+use std::any::Any;
+
+#[derive(Debug, Clone, PartialEq)]
+pub struct PoisonPill;
+
+impl Message for PoisonPill {
+  fn eq_message(&self, other: &dyn Message) -> bool {
+    other.as_any().is::<PoisonPill>()
+  }
+
+  fn as_any(&self) -> &(dyn Any + Send + Sync + 'static) {
+    self
+  }
+}