churn-v2/crates/churn/src/agent/handlers/scheduled_tasks.rs
kjuulh d6fdda0e4e
All checks were successful
continuous-integration/drone/push Build is passing
feat: add abstraction around task
Signed-off-by: kjuulh <contact@kjuulh.io>
2024-12-01 14:25:24 +01:00

43 lines
931 B
Rust

use std::collections::BTreeMap;
use crate::agent::{
actions::Plan,
task::{ConcreteTask, IntoTask},
};
#[derive(Clone)]
pub struct ScheduledTasks {}
impl ScheduledTasks {
pub fn new() -> Self {
Self {}
}
pub async fn handle(
&self,
task: &str,
_properties: BTreeMap<String, String>,
) -> anyhow::Result<()> {
tracing::info!("scheduling: {}", task);
let plan = Plan::new();
let tasks: Vec<ConcreteTask> = plan
.tasks()
.await?
.into_iter()
.map(|i| i.into_task())
.collect();
for task in tasks {
if !task.should_run().await? {
tracing::debug!(task = task.id(), "skipping run");
continue;
}
tracing::info!(task = task.id(), "executing task");
task.execute().await?;
}
Ok(())
}
}