bullrs 0.0.3

A BullMQ compatible Job Queue based on Redis
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
63

use std::{
    sync::{
        Arc,
        atomic::{AtomicBool, Ordering},
    },
    time::Duration,
};

use tokio::{sync::Notify, time::timeout};

/// A switch, which can be only toggled once to shut it down.
#[derive(Clone)]
pub struct ShutdownSwitch {
    toggle: Arc<AtomicBool>,
    notification: Arc<Notify>,
}

impl ShutdownSwitch {
    pub fn new() -> Self {
        Self {
            toggle: Arc::new(false.into()),
            notification: Default::default(),
        }
    }
    pub fn shutdown(&self) {
        self.toggle.store(true, Ordering::Relaxed);
        self.notification.notify_waiters();
    }
    pub fn running(&self) -> bool {
        !self.toggle.load(Ordering::Relaxed)
    }
    async fn block_until_shutdown(&self) {
        let cooldown = Duration::from_secs(1);
        while self.running() {
            // I am unusre, if a notification could be missed.
            // So we wait for at most 1s
            let _ = timeout(cooldown, self.notification.notified()).await;
        }
    }
    /// Sleeps for the given duration, but returns early if shutdown is triggered.
    pub async fn sleep(&self, duration: Duration) -> ShutdownSleep {
        if !self.running() {
            return ShutdownSleep::Shutdown;
        }
        match timeout(duration, self.block_until_shutdown()).await {
            Ok(_) => ShutdownSleep::Shutdown,
            Err(_) => ShutdownSleep::Slept,
        }
    }
}

pub enum ShutdownSleep {
    /// Worker is being shut down
    Shutdown,
    /// Regular sleep finished
    Slept,
}

impl Drop for ShutdownSwitch {
    fn drop(&mut self) {
        self.shutdown();
    }
}