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
20
21
22
23
24
25
26

use crate::errors::SummaResult;
use async_broadcast::Sender;
use tokio::task::JoinHandle;

#[derive(Clone, Debug)]
pub enum ControlMessage {
    Shutdown,
}

/// Holds `JoinHandle` together with its `shutdown_trigger`
#[derive(Debug)]
pub struct ThreadHandler {
    join_handle: JoinHandle<SummaResult<()>>,
    shutdown_trigger: Sender<ControlMessage>,
}

impl ThreadHandler {
    pub fn new(join_handle: JoinHandle<SummaResult<()>>, shutdown_trigger: Sender<ControlMessage>) -> ThreadHandler {
        ThreadHandler { join_handle, shutdown_trigger }
    }

    pub async fn stop(self) -> SummaResult<()> {
        self.shutdown_trigger.broadcast(ControlMessage::Shutdown).await.unwrap();
        self.join_handle.await?
    }
}