task.rs

use crate::{
    package::simulation::init::js_py::{TsInitTask, JsInitTask, PyInitTask},
    task::{
        StoreAccessValidator, TargetedTaskMessage, Task, TaskDistributionConfig, TaskSharedStore,
    },
    worker::WorkerHandler,
    worker_pool::WorkerPoolHandler,
    Error, Result,
};


/// All init package tasks are registered in this enum
#[derive(Clone, Debug)]
pub enum InitTask {
    JsInitTask(JsInitTask),
    TsInitTask(TsInitTask),
    PyInitTask(PyInitTask),
}

impl Task for InitTask {
    fn name(&self) -> &'static str {
        match self {
            Self::JsInitTask(task) => task.name(),
            Self::TsInitTask(task) => task.name(),
            Self::PyInitTask(task) => task.name(),
        }
    }

    fn distribution(&self) -> TaskDistributionConfig {
        match self {
            Self::JsInitTask(task) => task.distribution(),
            Self::TsInitTask(task) => task.distribution(),
            Self::PyInitTask(task) => task.distribution(),
        }
    }
}

impl StoreAccessValidator for InitTask {
    fn verify_store_access(&self, access: &TaskSharedStore) -> Result<()> {
        let state = &access.state;
        let context = access.context();
        if state.is_disabled() && context.is_disabled() {
            Ok(())
        } else {
            Err(Error::access_not_allowed(state, context, "Init".into()))
        }
    }
}

impl WorkerHandler for InitTask {
    fn start_message(&self) -> Result<TargetedTaskMessage> {
        match self {
            Self::JsInitTask(inner) => inner.start_message(),
            Self::TsInitTask(inner) => inner.start_message(),
            Self::PyInitTask(inner) => inner.start_message(),
        }
    }
}

impl WorkerPoolHandler for InitTask {}