use ractor::ActorCell;
use ractor::thread_local::{ThreadLocalActor, ThreadLocalActorSpawner};
use crate::discovery::actors::DiscoveryManager;
use crate::discovery::{Builder, Discovery, DiscoveryError};
use crate::supervisor::{ChildActor, ChildActorFut, Supervisor};
impl Builder {
pub async fn spawn_linked(self, supervisor: &Supervisor) -> Result<Discovery, DiscoveryError> {
let discovery = Discovery::new(None, self.build_args());
supervisor.start_child_actor(discovery.clone()).await?;
Ok(discovery)
}
}
impl ChildActor for Discovery {
fn on_start(
&self,
supervisor: ActorCell,
thread_pool: ThreadLocalActorSpawner,
) -> ChildActorFut<'_> {
Box::pin(async move {
let (actor_ref, _) =
DiscoveryManager::spawn_linked(None, self.args.clone(), supervisor, thread_pool)
.await?;
let mut inner = self.inner.write().await;
inner.actor_ref.replace(actor_ref.clone());
Ok(actor_ref.into())
})
}
}