scalesocket 0.2.4

A collaborative websocket server and autoscaler
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

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

use crate::types::{Event, EventTx, ShutdownTx};

pub async fn handle(routes_shutdown_tx: ShutdownTx, events_shutdown_tx: EventTx) {
    let mut interrupt = signal(SignalKind::interrupt()).expect("failed to create signal");
    let mut terminate = signal(SignalKind::terminate()).expect("failed to create signal");
    let signals = futures::future::select(interrupt.recv().boxed(), terminate.recv().boxed());

    signals.await;

    tracing::info!("received signal, shutting down");
    let _ = routes_shutdown_tx.send(());
    let _ = events_shutdown_tx.send(Event::Shutdown);
}