rayfish 0.1.1

P2P mesh VPN powered by iroh — connect peers by cryptographic identity, not IP address
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

//! Graceful shutdown via SIGINT/SIGTERM.
//!
//! Returns a [`CancellationToken`] that fires when a shutdown signal is received.
//! All long-running tasks select on this token to exit cleanly.

use tokio_util::sync::CancellationToken;

/// Creates a cancellation token that fires on SIGINT or SIGTERM.
pub fn token() -> CancellationToken {
    let token = CancellationToken::new();
    let t = token.clone();
    tokio::spawn(async move {
        signal_listener().await;
        tracing::info!("shutdown signal received");
        t.cancel();
    });
    token
}

async fn signal_listener() {
    #[cfg(unix)]
    {
        use tokio::signal::unix::{SignalKind, signal};
        let mut sigterm =
            signal(SignalKind::terminate()).expect("failed to register SIGTERM handler");
        tokio::select! {
            _ = tokio::signal::ctrl_c() => {}
            _ = sigterm.recv() => {}
        }
    }

    #[cfg(not(unix))]
    {
        tokio::signal::ctrl_c().await.ok();
    }
}