trifid/tfclient/src/nebulaworker.rs

// Code to handle the nebula worker

use std::sync::mpsc::{Receiver, TryRecvError};
use log::{error, info};
use crate::config::{load_cdata, save_cdata, TFClientConfig};
use crate::daemon::ThreadMessageSender;

pub enum NebulaWorkerMessage {
    Shutdown,
    ConfigUpdated
}

pub fn nebulaworker_main(_config: TFClientConfig, instance: String, _transmitter: ThreadMessageSender, rx: Receiver<NebulaWorkerMessage>) {
    let cdata = match load_cdata(&instance) {
        Ok(data) => data,
        Err(e) => {
            error!("unable to load cdata: {}", e);
            error!("nebula thread exiting with error");
            return;
        }
    };

    // dont need to save it, because we do not, in any circumstance, write to it
    loop {
        match rx.try_recv() {
            Ok(msg) => {
                match msg {
                    NebulaWorkerMessage::Shutdown => {
                        info!("recv on command socket: shutdown, stopping");
                        break;
                    },
                    NebulaWorkerMessage::ConfigUpdated => {
                        info!("our configuration has been updated - reloading");
                    }
                }
            },
            Err(e) => {
                match e {
                    TryRecvError::Empty => {}
                    TryRecvError::Disconnected => {
                        error!("nebulaworker command socket disconnected, shutting down to prevent orphaning");
                        break;
                    }
                }
            }
        }
    }
}