use crate::message::create_init_lifecycle;
use crate::message::ActorError;
use crate::message::ActorResult;
use crate::message::Envelope;
use crate::message::Message;
use async_trait::async_trait;
use tokio::sync::mpsc;
use tokio::sync::oneshot;
pub type State<T> = std::collections::HashMap<i32, T>;
#[async_trait]
pub trait Actor {
async fn handle_envelope(&mut self, envelope: Envelope);
async fn stop(&self);
}
pub struct Handle {
#[doc(hidden)]
pub sender: mpsc::Sender<Envelope>,
}
impl<'a> Handle {
#[doc(hidden)]
pub async fn send(&self, envelope: Envelope) -> ActorResult<()> {
self.sender.send(envelope).await.map_err(|e| ActorError {
reason: e.to_string(),
})
}
pub async fn tell(&self, message: Message) -> ActorResult<()> {
let envelope = Envelope {
message,
respond_to: None,
..Default::default()
};
log::trace!("tell sending envelope {envelope:?}");
self.send(envelope).await
}
pub async fn ask(&self, message: Message) -> ActorResult<Message> {
let (send, recv) = oneshot::channel();
let envelope = Envelope {
message,
respond_to: Some(send),
..Default::default()
};
log::trace!("ask sending envelope: {envelope:?}");
match self.send(envelope).await {
Ok(_) => recv.await.map_err(|e| ActorError {
reason: e.to_string(),
})?,
Err(e) => Err(e),
}
}
pub async fn integrate(&self, path: String, helper: &Self) -> ActorResult<Message> {
let (send, recv): (
oneshot::Sender<ActorResult<Message>>,
oneshot::Receiver<ActorResult<Message>>,
) = oneshot::channel();
let (init_cmd, load_cmd) = create_init_lifecycle(path, 8, send);
helper.send(load_cmd).await.map_err(|e| ActorError {
reason: e.to_string(),
})?;
self.send(init_cmd).await.map_err(|e| ActorError {
reason: e.to_string(),
})?;
recv.await.map_err(|e| ActorError {
reason: e.to_string(),
})?
}
#[doc(hidden)]
#[must_use]
pub const fn new(sender: mpsc::Sender<Envelope>) -> Self {
Self { sender }
}
}