use crate::actors::connector::RillConnector;
use crate::config::EngineConfig;
use anyhow::Error;
use async_trait::async_trait;
use meio::{Actor, Context, Eliminated, IdOf, InterruptedBy, StartedBy};
use rill_protocol::io::provider::EntryId;
use strum::{EnumIter, IntoEnumIterator};
pub struct RillEngine {
name: EntryId,
config: Option<EngineConfig>,
}
#[derive(Debug, Clone, PartialEq, Eq, Hash, EnumIter)]
pub enum Group {
Connector,
}
impl Actor for RillEngine {
type GroupBy = Group;
fn name(&self) -> String {
format!("RillEngine({})", &self.name)
}
}
impl RillEngine {
pub fn new(config: EngineConfig) -> Self {
let name = config.provider_name();
Self {
name,
config: Some(config),
}
}
}
#[async_trait]
impl<T: Actor> StartedBy<T> for RillEngine {
async fn handle(&mut self, ctx: &mut Context<Self>) -> Result<(), Error> {
ctx.termination_sequence(Group::iter().collect());
let config = self.config.take().unwrap();
let connector = RillConnector::new(config);
ctx.spawn_actor(connector, Group::Connector);
Ok(())
}
}
#[async_trait]
impl<T: Actor> InterruptedBy<T> for RillEngine {
async fn handle(&mut self, ctx: &mut Context<Self>) -> Result<(), Error> {
ctx.shutdown();
Ok(())
}
}
#[async_trait]
impl Eliminated<RillConnector> for RillEngine {
async fn handle(
&mut self,
_id: IdOf<RillConnector>,
ctx: &mut Context<Self>,
) -> Result<(), Error> {
ctx.shutdown();
Ok(())
}
}