celery-rs 0.6.2

Community-maintained Rust implementation of Celery (fork of rusty-celery)
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

//! This module contains the definition of application-provided schedules.
//!
//! These structs have not changed a lot compared to Python: in Python there are three
//! different types of schedules: `schedule` (corresponding to [`DeltaSchedule`]),
//! `crontab` (corresponding to [`CronSchedule`]), `solar` (not implemented yet).
use std::any::Any;
use std::time::{Duration, SystemTime};

mod cron;
pub use cron::CronSchedule;

mod descriptor;
pub use descriptor::ScheduleDescriptor;

/// The trait that all schedules implement.
pub trait Schedule: Any {
    /// Compute when a task should be executed again.
    /// If this method returns `None` then the task should
    /// never run again and it is safe to remove it from the
    /// list of scheduled tasks.
    fn next_call_at(&self, last_run_at: Option<SystemTime>) -> Option<SystemTime>;

    /// Describe this schedule so it can be serialized for persistence.
    fn describe(&self) -> Option<ScheduleDescriptor> {
        None
    }
}

/// A schedule that can be used to execute tasks at regular intervals.
pub struct DeltaSchedule {
    interval: Duration,
}

impl DeltaSchedule {
    /// Create a new time delta schedule which can be used to execute a task
    /// forever, starting immediately and with the given `interval`
    /// between subsequent executions.
    pub fn new(interval: Duration) -> DeltaSchedule {
        DeltaSchedule { interval }
    }

    pub fn interval(&self) -> Duration {
        self.interval
    }
}

impl Schedule for DeltaSchedule {
    fn next_call_at(&self, last_run_at: Option<SystemTime>) -> Option<SystemTime> {
        match last_run_at {
            Some(last_run_at) => Some(
                last_run_at
                    .checked_add(self.interval)
                    .expect("Invalid SystemTime encountered"),
            ),
            None => Some(SystemTime::now()),
        }
    }

    fn describe(&self) -> Option<ScheduleDescriptor> {
        Some(ScheduleDescriptor::delta(self.interval))
    }
}