summa 0.8.8

Fast full-text search server
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use crate::utils::thread_handler::ControlMessage;
use async_broadcast::{broadcast, Receiver};
use tokio::signal::unix::{signal, SignalKind};
use tokio::task;

/// Spawns a thread for processing `SignalKind` and returns `oneshot::Receiver` for a signal event
pub fn signal_channel() -> Receiver<ControlMessage> {
    let (sender, receiver) = broadcast::<ControlMessage>(1);
    task::spawn(async move {
        let mut sigint = signal(SignalKind::interrupt()).unwrap();
        let mut sigterm = signal(SignalKind::terminate()).unwrap();
        tokio::select! {
            _ = sigint.recv() => {}
            _ = sigterm.recv() => {}
        }
        sender.broadcast(ControlMessage::Shutdown).await.unwrap()
    });
    receiver
}