effectum 0.7.0

An embeddable task queue based on SQLite
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

use std::ops::Deref;
use std::sync::Arc;
use std::time::Duration;

use time::OffsetDateTime;
use tokio::sync::mpsc;
use tokio::time::Instant;

use crate::db_writer::DbOperation;
use crate::pending_jobs::ScheduledJobType;
use crate::worker_list::Workers;

pub(crate) struct SharedStateData {
    pub db_write_tx: mpsc::Sender<DbOperation>,
    /// Separate pool for miscellaneous read-only calls so they won't block the writes.
    pub read_conn_pool: deadpool_sqlite::Pool,
    pub workers: tokio::sync::RwLock<Workers>,
    pub close: tokio::sync::watch::Receiver<()>,
    pub time: Time,
    pub pending_jobs_tx: tokio::sync::mpsc::Sender<ScheduledJobType>,
}

#[derive(Clone)]
pub(crate) struct SharedState(pub Arc<SharedStateData>);

impl Deref for SharedState {
    type Target = Arc<SharedStateData>;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

#[derive(Clone)]
pub(crate) struct Time {
    start_instant: tokio::time::Instant,
    start_time: time::OffsetDateTime,
}

impl Time {
    pub fn new() -> Self {
        let start_instant = tokio::time::Instant::now();
        let start_time = time::OffsetDateTime::now_utc();

        Time {
            start_instant,
            start_time,
        }
    }

    pub fn now(&self) -> OffsetDateTime {
        let now = self.start_instant.elapsed();
        self.start_time + now
    }

    pub fn instant_for_timestamp(&self, timestamp: i64) -> Instant {
        let ts = std::cmp::max(timestamp - self.start_time.unix_timestamp(), 0) as u64;
        let duration = Duration::from_secs(ts);
        self.start_instant + duration
    }
}