utils rework
prplV
parent
1c6729daab
commit
b6ecb10a77
|
|
@ -117,6 +117,7 @@ impl PrebootParams {
|
||||||
eprintln!("Local config file {} doesn't exist", &self.config.display());
|
eprintln!("Local config file {} doesn't exist", &self.config.display());
|
||||||
return Err(Error::msg("Local Config Not Found. Cannot start"));
|
return Err(Error::msg("Local Config Not Found. Cannot start"));
|
||||||
}
|
}
|
||||||
|
// redis server check
|
||||||
Ok(self)
|
Ok(self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -4,8 +4,9 @@ pub mod metrics;
|
||||||
pub mod prcs;
|
pub mod prcs;
|
||||||
pub mod services;
|
pub mod services;
|
||||||
|
|
||||||
//
|
// TODO : saving current flags state
|
||||||
|
|
||||||
|
use crate::options::structs::CustomError;
|
||||||
use crate::options::structs::TrackingProcess;
|
use crate::options::structs::TrackingProcess;
|
||||||
use files::create_watcher;
|
use files::create_watcher;
|
||||||
use files::file_handler;
|
use files::file_handler;
|
||||||
|
|
@ -60,125 +61,130 @@ pub async fn run_daemons(
|
||||||
loop {
|
loop {
|
||||||
let run_hand = running_handler(proc.clone(), tx.clone(), watchers_clone.clone());
|
let run_hand = running_handler(proc.clone(), tx.clone(), watchers_clone.clone());
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
_ = run_hand => {},
|
_ = run_hand => continue,
|
||||||
_val = rx.recv() => {
|
_val = rx.recv() => {
|
||||||
match _val.unwrap() {
|
if process_protocol_symbol(proc.clone(), _val.unwrap()).await.is_err() {
|
||||||
// 1 - File-dependency handling error -> terminating (after waiting)
|
return;
|
||||||
1 => {
|
}
|
||||||
if is_active(&proc.name).await {
|
|
||||||
error!("File-dependency handling error: Terminating {} process ..." , &proc.name);
|
|
||||||
terminate_process(&proc.name).await;
|
|
||||||
tokio::time::sleep(Duration::from_millis(100)).await;
|
|
||||||
}
|
|
||||||
return;
|
|
||||||
},
|
|
||||||
// 2 - File-dependency handling error -> holding (after waiting)
|
|
||||||
2 => {
|
|
||||||
if !is_frozen(&proc.name).await {
|
|
||||||
error!("File-dependency handling error: Freezing {} process ..." , &proc.name);
|
|
||||||
freeze_process(&proc.name).await;
|
|
||||||
tokio::time::sleep(Duration::from_millis(100)).await;
|
|
||||||
}
|
|
||||||
},
|
|
||||||
// 3 - Running process error
|
|
||||||
3 => {
|
|
||||||
error!("Error due to starting {} process", &proc.name);
|
|
||||||
break;
|
|
||||||
},
|
|
||||||
// 4 - Timeout of waiting service-dependency -> staying (after waiting)
|
|
||||||
4 => {
|
|
||||||
// warn!("Timeout of waiting service-dependency: Ignoring on {} process ..." , &proc.name);
|
|
||||||
tokio::time::sleep(Duration::from_millis(100)).await;
|
|
||||||
},
|
|
||||||
// 5 - Timeout of waiting service-dependency -> terminating (after waiting)
|
|
||||||
5 => {
|
|
||||||
if is_active(&proc.name).await {
|
|
||||||
error!("Timeout of waiting service-dependency: Terminating {} process ..." , &proc.name);
|
|
||||||
terminate_process(&proc.name).await;
|
|
||||||
tokio::time::sleep(Duration::from_millis(1000)).await;
|
|
||||||
}
|
|
||||||
},
|
|
||||||
// 6 - Timeout of waiting service-dependency -> holding (after waiting)
|
|
||||||
6 => {
|
|
||||||
// println!("holding {}-{}", proc.name, is_active(&proc.name).await);
|
|
||||||
if !is_frozen(&proc.name).await {
|
|
||||||
error!("Timeout of waiting service-dependency: Freezing {} process ..." , &proc.name);
|
|
||||||
freeze_process(&proc.name).await;
|
|
||||||
tokio::time::sleep(Duration::from_secs(1)).await;
|
|
||||||
}
|
|
||||||
},
|
|
||||||
// // 7 - File-dependency change -> terminating (after check)
|
|
||||||
7 => {
|
|
||||||
error!("File-dependency warning (file changed). Terminating {} process...", &proc.name);
|
|
||||||
terminate_process(&proc.name).await;
|
|
||||||
tokio::time::sleep(Duration::from_millis(100)).await;
|
|
||||||
return;
|
|
||||||
},
|
|
||||||
// // 8 - File-dependency change -> restarting (after check)
|
|
||||||
8 => {
|
|
||||||
warn!("File-dependency warning (file changed). Restarting {} process...", &proc.name);
|
|
||||||
let _ = restart_process(&proc.name, &proc.path).await;
|
|
||||||
tokio::time::sleep(Duration::from_millis(100)).await;
|
|
||||||
},
|
|
||||||
// // 9 - File-dependency change -> staying (after check)
|
|
||||||
9 => {
|
|
||||||
// no need to trash logs
|
|
||||||
warn!("File-dependency warning (file changed). Ignoring event on {} process...", &proc.name);
|
|
||||||
tokio::time::sleep(Duration::from_millis(100)).await;
|
|
||||||
},
|
|
||||||
|
|
||||||
// 10 - Process unfreaze call via file handler (or service handler)
|
|
||||||
10 | 11 => {
|
|
||||||
if is_frozen(&proc.name).await {
|
|
||||||
warn!("Unfreezing process {} call...", &proc.name);
|
|
||||||
unfreeze_process(&proc.name).await;
|
|
||||||
}
|
|
||||||
tokio::time::sleep(Duration::from_millis(100)).await;
|
|
||||||
},
|
|
||||||
// 11 - Process unfreaze call via service handler
|
|
||||||
// 11 => {
|
|
||||||
// if is_frozen(&proc.name).await {
|
|
||||||
// warn!("Unfreezing process {} call...", &proc.name);
|
|
||||||
// unfreeze_process(&proc.name).await;
|
|
||||||
// }
|
|
||||||
// tokio::time::sleep(Duration::from_millis(100)).await;
|
|
||||||
// },
|
|
||||||
// 101 - Impermissible trigger values in JSON
|
|
||||||
101 => {
|
|
||||||
error!("Impermissible trigger values in JSON in {}'s block. Killing thread...", proc.name);
|
|
||||||
if is_active(&proc.name).await {
|
|
||||||
terminate_process(&proc.name).await;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
},
|
|
||||||
//
|
|
||||||
// 121 - Cannot create valid watcher for file dependency
|
|
||||||
121 => {
|
|
||||||
error!("Cannot create valid watcher for {}'s file dependency. Terminating thread...", proc.name);
|
|
||||||
let _ = terminate_process("runner-rs").await;
|
|
||||||
break;
|
|
||||||
},
|
|
||||||
// 111 - global thread termination with killing current child in a face
|
|
||||||
// of a current process
|
|
||||||
111 => {
|
|
||||||
warn!("Terminating {}'s child processes...", &proc.name);
|
|
||||||
match is_active(&proc.name).await {
|
|
||||||
true => {
|
|
||||||
terminate_process(&proc.name).await;
|
|
||||||
},
|
|
||||||
false => {
|
|
||||||
log::info!("Process {} is already terminated!", proc.name);
|
|
||||||
},
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
},
|
|
||||||
_ => {},
|
|
||||||
}
|
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
tokio::task::yield_now().await;
|
tokio::task::yield_now().await;
|
||||||
}
|
}
|
||||||
tokio::task::yield_now().await;
|
}
|
||||||
|
|
||||||
|
async fn process_protocol_symbol(proc: Arc<TrackingProcess>, val: u8) -> Result<(), CustomError>{
|
||||||
|
match val {
|
||||||
|
// 1 - File-dependency handling error -> terminating (after waiting)
|
||||||
|
1 => {
|
||||||
|
if is_active(&proc.name).await {
|
||||||
|
error!("File-dependency handling error: Terminating {} process ..." , &proc.name);
|
||||||
|
terminate_process(&proc.name).await;
|
||||||
|
tokio::time::sleep(Duration::from_millis(500)).await;
|
||||||
|
}
|
||||||
|
// return;
|
||||||
|
},
|
||||||
|
// 2 - File-dependency handling error -> holding (after waiting)
|
||||||
|
2 => {
|
||||||
|
if !is_frozen(&proc.name).await {
|
||||||
|
error!("File-dependency handling error: Freezing {} process ..." , &proc.name);
|
||||||
|
freeze_process(&proc.name).await;
|
||||||
|
tokio::time::sleep(Duration::from_millis(100)).await;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
// 3 - Running process error
|
||||||
|
3 => {
|
||||||
|
error!("Error due to starting {} process", &proc.name);
|
||||||
|
return Err(CustomError::Fatal)
|
||||||
|
},
|
||||||
|
// 4 - Timeout of waiting service-dependency -> staying (after waiting)
|
||||||
|
4 => {
|
||||||
|
// warn!("Timeout of waiting service-dependency: Ignoring on {} process ..." , &proc.name);
|
||||||
|
tokio::time::sleep(Duration::from_millis(100)).await;
|
||||||
|
},
|
||||||
|
// 5 - Timeout of waiting service-dependency -> terminating (after waiting)
|
||||||
|
5 => {
|
||||||
|
if is_active(&proc.name).await {
|
||||||
|
error!("Timeout of waiting service-dependency: Terminating {} process ..." , &proc.name);
|
||||||
|
terminate_process(&proc.name).await;
|
||||||
|
tokio::time::sleep(Duration::from_millis(500)).await;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
// 6 - Timeout of waiting service-dependency -> holding (after waiting)
|
||||||
|
6 => {
|
||||||
|
// println!("holding {}-{}", proc.name, is_active(&proc.name).await);
|
||||||
|
if !is_frozen(&proc.name).await {
|
||||||
|
error!("Timeout of waiting service-dependency: Freezing {} process ..." , &proc.name);
|
||||||
|
freeze_process(&proc.name).await;
|
||||||
|
tokio::time::sleep(Duration::from_secs(1)).await;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
// // 7 - File-dependency change -> terminating (after check)
|
||||||
|
7 => {
|
||||||
|
error!("File-dependency warning (file changed). Terminating {} process...", &proc.name);
|
||||||
|
terminate_process(&proc.name).await;
|
||||||
|
tokio::time::sleep(Duration::from_millis(100)).await;
|
||||||
|
return Err(CustomError::Fatal)
|
||||||
|
},
|
||||||
|
// // 8 - File-dependency change -> restarting (after check)
|
||||||
|
8 => {
|
||||||
|
warn!("File-dependency warning (file changed). Restarting {} process...", &proc.name);
|
||||||
|
let _ = restart_process(&proc.name, &proc.path).await;
|
||||||
|
tokio::time::sleep(Duration::from_millis(100)).await;
|
||||||
|
},
|
||||||
|
// // 9 - File-dependency change -> staying (after check)
|
||||||
|
9 => {
|
||||||
|
// no need to trash logs
|
||||||
|
warn!("File-dependency warning (file changed). Ignoring event on {} process...", &proc.name);
|
||||||
|
tokio::time::sleep(Duration::from_millis(100)).await;
|
||||||
|
},
|
||||||
|
|
||||||
|
// 10 - Process unfreaze call via file handler (or service handler)
|
||||||
|
10 | 11 => {
|
||||||
|
if is_frozen(&proc.name).await {
|
||||||
|
warn!("Unfreezing process {} call...", &proc.name);
|
||||||
|
unfreeze_process(&proc.name).await;
|
||||||
|
}
|
||||||
|
tokio::time::sleep(Duration::from_millis(100)).await;
|
||||||
|
},
|
||||||
|
// 11 - Process unfreaze call via service handler
|
||||||
|
// 11 => {
|
||||||
|
// if is_frozen(&proc.name).await {
|
||||||
|
// warn!("Unfreezing process {} call...", &proc.name);
|
||||||
|
// unfreeze_process(&proc.name).await;
|
||||||
|
// }
|
||||||
|
// tokio::time::sleep(Duration::from_millis(100)).await;
|
||||||
|
// },
|
||||||
|
// 101 - Impermissible trigger values in JSON
|
||||||
|
101 => {
|
||||||
|
error!("Impermissible trigger values in JSON in {}'s block. Killing thread...", proc.name);
|
||||||
|
if is_active(&proc.name).await {
|
||||||
|
terminate_process(&proc.name).await;
|
||||||
|
}
|
||||||
|
return Err(CustomError::Fatal)
|
||||||
|
},
|
||||||
|
//
|
||||||
|
// 121 - Cannot create valid watcher for file dependency
|
||||||
|
121 => {
|
||||||
|
error!("Cannot create valid watcher for {}'s file dependency. Terminating thread...", proc.name);
|
||||||
|
let _ = terminate_process("runner-rs").await;
|
||||||
|
return Err(CustomError::Fatal)
|
||||||
|
},
|
||||||
|
// 111 - global thread termination with killing current child in a face
|
||||||
|
// of a current process
|
||||||
|
111 => {
|
||||||
|
warn!("Terminating {}'s child processes...", &proc.name);
|
||||||
|
match is_active(&proc.name).await {
|
||||||
|
true => {
|
||||||
|
terminate_process(&proc.name).await;
|
||||||
|
},
|
||||||
|
false => {
|
||||||
|
log::info!("Process {} is already terminated!", proc.name);
|
||||||
|
},
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => {},
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
// check process status daemon
|
// check process status daemon
|
||||||
/// # Fn `run_daemons`
|
/// # Fn `run_daemons`
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue