Merge pull request #145 from code0-tech/80-cron-adapter

Cron Adapter

Author: raphael-goetz

Cargo.lock

@@ -1,5 +1,5 @@
 [workspace]
-members = ["adapter/rest", "crates/base"]
+members = [ "adapter/cron","adapter/rest", "crates/base"]
 resolver = "3"
 
 [workspace.package]
@@ -22,6 +22,8 @@ anyhow = "1.0.98"
 prost = "0.14.0"
 tonic-health = "0.14.0"
 futures-lite = "2.6.1"
+chrono = "0.4.42"
+cron = "0.15.0"
 
 [workspace.dependencies.base]
 path = "../draco/crates/base"

Cargo.toml

@@ -0,0 +1,14 @@
+[package]
+name = "cron"
+version.workspace = true
+edition.workspace = true
+
+[dependencies]
+tokio = {workspace = true}
+chrono = {workspace = true}
+cron = {workspace = true}
+base = {workspace = true}
+tucana = {workspace = true}
+async-trait =  {workspace = true}
+log = { workspace = true }
+anyhow = {workspace = true}

adapter/cron/Cargo.toml

@@ -0,0 +1,142 @@
+use async_trait::async_trait;
+use base::runner::{ServerContext, ServerRunner};
+use base::store::FlowIdentifyResult;
+use base::traits::{IdentifiableFlow, LoadConfig, Server};
+use chrono::{DateTime, Datelike, Timelike, Utc};
+use cron::Schedule;
+use std::str::FromStr;
+use tucana::shared::ValidationFlow;
+use tucana::shared::value::Kind;
+
+#[derive(Default)]
+struct Cron {}
+
+#[derive(Clone)]
+struct CronConfig {}
+
+impl LoadConfig for CronConfig {
+    fn load() -> Self {
+        Self {}
+    }
+}
+
+#[tokio::main]
+async fn main() {
+    let server = Cron::default();
+    let runner = ServerRunner::new(server).await.unwrap();
+    runner.serve().await.unwrap();
+}
+
+struct Time {
+    now: DateTime<Utc>,
+}
+
+fn extract_flow_setting_field(flow: &ValidationFlow, name: &str) -> Option<String> {
+    flow.settings
+        .iter()
+        .find(|s| s.flow_setting_id == name)
+        .and_then(|s| s.value.as_ref())
+        .and_then(|v| v.kind.as_ref())
+        .and_then(|k| match k {
+            Kind::StringValue(s) => Some(s.clone()),
+            _ => None,
+        })
+}
+
+impl IdentifiableFlow for Time {
+    fn identify(&self, flow: &tucana::shared::ValidationFlow) -> bool {
+        let Some(minute) = extract_flow_setting_field(flow, "CRON_MINUTE") else {
+            return false;
+        };
+        let Some(hour) = extract_flow_setting_field(flow, "CRON_HOUR") else {
+            return false;
+        };
+        let Some(dom) = extract_flow_setting_field(flow, "CRON_DAY_OF_MONTH") else {
+            return false;
+        };
+        let Some(month) = extract_flow_setting_field(flow, "CRON_MONTH") else {
+            return false;
+        };
+        let Some(dow) = extract_flow_setting_field(flow, "CRON_DAY_OF_WEEK") else {
+            return false;
+        };
+
+        let expression = format!("* {} {} {} {} {}", minute, hour, dom, month, dow);
+        let schedule = match Schedule::from_str(expression.as_str()) {
+            Ok(s) => s,
+            Err(e) => {
+                log::error!(
+                    "Could not create schedule from expression ({}). Reason: {:?}",
+                    expression,
+                    e
+                );
+                return false;
+            }
+        };
+        let next = match schedule.upcoming(Utc).next() {
+            Some(n) => n,
+            None => {
+                log::error!("Could not find any upcomming schedules");
+                return false;
+            }
+        };
+
+        self.now.year() == next.year()
+            && self.now.month() == next.month()
+            && self.now.day() == next.day()
+            && self.now.hour() == next.hour()
+            && self.now.minute() == next.minute()
+    }
+}
+
+#[async_trait]
+impl Server<CronConfig> for Cron {
+    async fn init(&mut self, _ctx: &ServerContext<CronConfig>) -> anyhow::Result<()> {
+        Ok(())
+    }
+
+    async fn run(&mut self, ctx: &ServerContext<CronConfig>) -> anyhow::Result<()> {
+        log::info!("Starting Cron adapter");
+        let expression = "0 * * * * *";
+        let schedule = Schedule::from_str(expression)?;
+        let pattern = "CRON.<";
+
+        loop {
+            let now = Utc::now();
+            log::info!("Scheduled: {:?}", now);
+            if let Some(next) = schedule.upcoming(Utc).take(1).next() {
+                let until_next = next - now;
+                tokio::time::sleep(until_next.to_std()?).await;
+
+                let time = Time { now: next };
+                match ctx
+                    .adapter_store
+                    .get_possible_flow_match(pattern.to_string(), time)
+                    .await
+                {
+                    FlowIdentifyResult::None => {
+                        log::debug!("No Flow identified for this schedule");
+                    }
+                    FlowIdentifyResult::Single(flow) => {
+                        log::debug!("One Flow identified for this schedule");
+                        ctx.adapter_store
+                            .validate_and_execute_flow(flow, None)
+                            .await;
+                    }
+                    FlowIdentifyResult::Multiple(flows) => {
+                        log::debug!("Multiple Flows identified for this schedule");
+                        for flow in flows {
+                            ctx.adapter_store
+                                .validate_and_execute_flow(flow, None)
+                                .await;
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    async fn shutdown(&mut self, _ctx: &ServerContext<CronConfig>) -> anyhow::Result<()> {
+        Ok(())
+    }
+}