trifid/tfclient/src/daemon.rs

use std::sync::mpsc;
use std::sync::mpsc::Sender;
use std::thread;
use log::{error, info};

use crate::apiworker::{apiworker_main, APIWorkerMessage};
use crate::config::load_config;
use crate::nebulaworker::{nebulaworker_main, NebulaWorkerMessage};
use crate::socketworker::{socketworker_main, SocketWorkerMessage};
use crate::util::check_server_url;

pub fn daemon_main(name: String, server: String) {
    // Validate the `server`
    check_server_url(&server);

    info!("Loading config...");
    let config = match load_config(&name) {
        Ok(cfg) => cfg,
        Err(e) => {
            error!("Error loading configuration: {}", e);
            std::process::exit(1);
        }
    };

    info!("Starting API thread...");

    let (tx_api, rx_api) = mpsc::channel::<APIWorkerMessage>();
    let (tx_socket, rx_socket) = mpsc::channel::<SocketWorkerMessage>();
    let (tx_nebula, rx_nebula) = mpsc::channel::<NebulaWorkerMessage>();

    let transmitter = ThreadMessageSender {
        socket_thread: tx_socket,
        api_thread: tx_api,
        nebula_thread: tx_nebula
    };

    let config_api = config.clone();
    let transmitter_api = transmitter.clone();
    let api_thread = thread::spawn(move || {
        apiworker_main(config_api, transmitter_api, rx_api);
    });

    info!("Starting Nebula thread...");
    let config_nebula = config.clone();
    let transmitter_nebula = transmitter.clone();
    let nebula_thread = thread::spawn(move || {
        nebulaworker_main(config_nebula, transmitter_nebula, rx_nebula);
    });

    info!("Starting socket worker thread...");
    let socket_thread = thread::spawn(move || {
        socketworker_main(config, transmitter, rx_socket);
    });

    info!("Waiting for socket thread to exit...");
    match socket_thread.join() {
        Ok(_) => (),
        Err(_) => {
            error!("Error waiting for socket thread to exit.");
            std::process::exit(1);
        }
    }

    info!("Waiting for API thread to exit...");
    match api_thread.join() {
        Ok(_) => (),
        Err(_) => {
            error!("Error waiting for api thread to exit.");
            std::process::exit(1);
        }
    }

    info!("Waiting for Nebula thread to exit...");
    match nebula_thread.join() {
        Ok(_) => (),
        Err(_) => {
            error!("Error waiting for nebula thread to exit.");
            std::process::exit(1);
        }
    }

    info!("All threads exited");
}

#[derive(Clone)]
pub struct ThreadMessageSender {
    socket_thread: Sender<SocketWorkerMessage>,
    api_thread: Sender<APIWorkerMessage>,
    nebula_thread: Sender<NebulaWorkerMessage>
}