aa-runtime 0.0.1-alpha.8

Tokio async runtime wrapper and lifecycle management for Agent Assembly
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

//! Signal handling and graceful shutdown coordination.

/// Waits until the process receives a shutdown signal (SIGTERM or SIGINT).
///
/// Returns as soon as either signal fires. Callers should then trigger
/// cooperative cancellation on all tracked tasks.
pub async fn wait_for_shutdown_signal() {
    let sigterm = sigterm();
    let sigint = tokio::signal::ctrl_c();

    tokio::select! {
        _ = sigterm => {
            tracing::info!("received SIGTERM — initiating graceful shutdown");
        }
        result = sigint => {
            match result {
                Ok(()) => tracing::info!("received SIGINT (Ctrl-C) — initiating graceful shutdown"),
                Err(e) => tracing::error!("SIGINT handler error: {e}"),
            }
        }
    }
}

/// Returns a future that resolves on the first SIGTERM.
///
/// On non-Unix platforms this future never resolves (SIGTERM is Unix-only).
async fn sigterm() {
    #[cfg(unix)]
    {
        use tokio::signal::unix::{signal, SignalKind};
        let mut stream = signal(SignalKind::terminate()).expect("failed to install SIGTERM handler");
        stream.recv().await;
    }
    #[cfg(not(unix))]
    {
        std::future::pending::<()>().await
    }
}