caesura 0.31.0

An all-in-one command line tool to transcode FLAC audio files and upload to gazelle based indexers/trackers
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

//! Cooperative shutdown for graceful SIGINT handling.

use crate::prelude::*;
use std::process::exit;
use std::sync::atomic::{AtomicBool, Ordering};
use tokio::signal::ctrl_c;
use tokio::spawn;

/// Cooperative shutdown signal for graceful SIGINT handling.
///
/// Registered as a singleton in the DI container. Call [`listen`](Shutdown::listen)
/// once at startup to spawn the signal handler task.
pub struct Shutdown {
    /// Whether a graceful shutdown has been requested.
    requested: AtomicBool,
}

#[injectable]
impl Shutdown {
    /// Create a new [`Shutdown`] in the non-requested state.
    #[must_use]
    pub fn new() -> Self {
        Self {
            requested: AtomicBool::new(false),
        }
    }

    /// Check whether a graceful shutdown has been requested.
    pub fn is_requested(&self) -> bool {
        self.requested.load(Ordering::Relaxed)
    }

    /// Mark shutdown as requested.
    fn request(&self) {
        self.requested.store(true, Ordering::Relaxed);
    }

    /// Spawn a task that listens for SIGINT.
    ///
    /// - First signal: sets the shutdown flag and logs a message
    /// - Second signal: calls [`exit`]
    pub fn listen(self: &Ref<Self>) {
        let shutdown = self.clone();
        spawn(async move {
            ctrl_c().await.ok();
            shutdown.request();
            info!(
                "{} graceful shutdown.\nFinishing current item.\nPress Ctrl+C again to exit immediately.",
                "Triggered".bold()
            );
            ctrl_c().await.ok();
            info!("{} immediate shutdown", "Triggered".bold());
            exit(1);
        });
    }
}