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

use crate::channel::Message;
use once_cell::sync::OnceCell;
use std::process;
use std::sync::mpsc;
use std::sync::mpsc::{Receiver, Sender};
use std::sync::{Arc, Mutex};
use tracing::error;

type ArcSender = Arc<Mutex<Sender<Message>>>;
type ArcReceiver = Arc<Mutex<Receiver<Message>>>;

static CHANNEL: OnceCell<(ArcSender, ArcReceiver)> = OnceCell::new();

pub fn get_sender() -> ArcSender {
    CHANNEL.get().expect("cannot get sender").0.clone()
}
pub fn get_receiver() -> ArcReceiver {
    CHANNEL.get().expect("cannot get receiver.").1.clone()
}

pub fn init_channels() {
    let (sender, receiver): (Sender<Message>, Receiver<Message>) = mpsc::channel();
    if let Err(e) = CHANNEL.set((Arc::new(Mutex::new(sender)), Arc::new(Mutex::new(receiver)))) {
        error!("init channels failed: {:?}", e);
        process::exit(1);
    }
}