roder-tasks 0.1.1

Agentic software development tools and SDKs for Roder.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

use std::collections::BTreeMap;
use std::sync::Arc;

use roder_api::extension::TaskExecutorId;
use roder_api::tasks::{TaskExecutor, TaskSpec};

#[derive(Default, Clone)]
pub struct TaskExecutorRegistry {
    executors: BTreeMap<TaskExecutorId, Arc<dyn TaskExecutor>>,
}

impl TaskExecutorRegistry {
    pub fn register(&mut self, executor: Arc<dyn TaskExecutor>) -> anyhow::Result<()> {
        let id = executor.id();
        if self.executors.contains_key(&id) {
            anyhow::bail!("task executor {id:?} is already registered");
        }
        self.executors.insert(id, executor);
        Ok(())
    }

    pub fn get(&self, id: &str) -> Option<Arc<dyn TaskExecutor>> {
        self.executors.get(id).cloned()
    }

    pub fn specs(&self) -> Vec<TaskSpec> {
        self.executors
            .values()
            .map(|executor| executor.spec())
            .collect()
    }

    pub fn is_empty(&self) -> bool {
        self.executors.is_empty()
    }
}