tulpje_framework/handler/
task_handler.rs

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
use std::{future::Future, pin::Pin};

use async_cron_scheduler::cron::Schedule;
use chrono::{DateTime, Utc};

use crate::context::TaskContext;
use crate::Error;

pub(crate) type TaskFunc<T> =
    fn(TaskContext<T>) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send>>;

#[derive(Clone)]
pub struct TaskHandler<T: Clone + Send + Sync> {
    pub module: String,
    pub name: String,
    pub cron: Schedule,
    pub func: TaskFunc<T>,
}

impl<T: Clone + Send + Sync> TaskHandler<T> {
    pub async fn run(&self, ctx: TaskContext<T>) -> Result<(), Error> {
        // can add more handling/parsing/etc here in the future
        (self.func)(ctx).await
    }

    pub fn next_run(&self) -> Option<DateTime<Utc>> {
        self.cron.upcoming(Utc).next()
    }
}