pass-it-on 0.17.3

A library that provides a simple notification client and server that receives messages and passes them on to endpoints
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
64
65
66
67
68
69
70
71
72
73

use tracing::{error, info};
use std::time::Duration;
use tokio::sync::watch;

#[cfg(unix)]
pub(crate) async fn listen_for_shutdown(
    shutdown_tx: watch::Sender<bool>,
    shutdown: Option<watch::Receiver<bool>>,
    seconds_to_wait: u64,
) {
    use tokio::signal::unix::{signal, SignalKind};
    // Listen for SIGTERM and SIGINT to know when shutdown
    let mut sigterm = signal(SignalKind::terminate()).expect("unable to listen for terminate signal");
    let mut sigint = signal(SignalKind::interrupt()).expect("unable to listen for interrupt signal");

    if let Some(mut shutdown_rx) = shutdown {
        tokio::select! {
        _ = sigterm.recv() => info!("Received SIGTERM."),
        _ = sigint.recv() => info!("Received SIGINT."),
        _ = shutdown_rx.changed() => info!( "Received shutdown channel signal."),
        }
    } else {
        tokio::select! {
        _ = sigterm.recv() => info!("Received SIGTERM."),
        _ = sigint.recv() => info!("Received SIGINT."),
        }
    }

    // Send shutdown signal
    if let Err(error) = shutdown_tx.send(true) {
        error!("Unable to send shutdown signal: {}", error)
    }

    info!("Starting Shutdown");
    // Allow time for cleanup
    tokio::time::sleep(Duration::from_secs(seconds_to_wait)).await;
    info!("Shutdown Complete")
}

#[cfg(windows)]
pub(crate) async fn listen_for_shutdown(
    shutdown_tx: watch::Sender<bool>,
    shutdown: Option<watch::Receiver<bool>>,
    seconds_to_wait: u64,
) {
    use tokio::signal::windows::{ctrl_break, ctrl_c};
    // Listen for CTRL-C and CTRL-BREAK to know when shutdown
    let mut sig_ctrl_break = ctrl_break().expect("unable to listen for ctrl-break signal");
    let mut sig_ctrl_c = ctrl_c().expect("unable to listen for ctrl-c signal");

    if let Some(mut shutdown_rx) = shutdown {
        tokio::select! {
        _ = sig_ctrl_break.recv() => info!("Received CTRL-BREAK."),
        _ = sig_ctrl_c.recv() => info!("Received CTRL-C."),
        _ = shutdown_rx.changed() => info!("Received shutdown channel signal."),
        }
    } else {
        tokio::select! {
            _ = sig_ctrl_break.recv() => info!("Received CTRL-BREAK."),
            _ = sig_ctrl_c.recv() => info!("Received CTRL-C."),
        }
    }

    // Send shutdown signal
    if let Err(error) = shutdown_tx.send(true) {
        error!("Unable to send shutdown signal: {}", error)
    }

    info!("Starting Shutdown");
    // Allow time for cleanup
    tokio::time::sleep(Duration::from_secs(seconds_to_wait)).await;
    info!("Shutdown Complete")
}