librt-rs 0.1.0

Runtime primitives
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

use tokio::signal::unix::{SignalKind, signal};

#[cfg(feature = "tracing")]
use tracing::info;

pub async fn shutdown() {
    tokio::select! {
        // SIGINT  - To allow Ctrl-c to emulate SIGTERM while developing.
        () = sig(SignalKind::interrupt(), "SIGINT") => {
            std::process::exit(0);
        }
        // SIGTERM - Kubernetes sends this to start a graceful shutdown.
        () = sig(SignalKind::terminate(), "SIGTERM") => {}
    };
}

async fn sig(kind: SignalKind, name: &'static str) {
    // Create a Future that completes the first
    // time the process receives 'sig'.
    signal(kind)
        .expect("Failed to register signal handler")
        .recv()
        .await;
    info!(
        // use target to remove 'imp' from output
        target: "rt::signal",
        "received {}, starting shutdown",
        name,
    );
}