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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
mod config;
mod system;
mod listener;
use std::{fs, thread};
use tokio::{runtime, task::JoinSet};
fn main() {
let initial_config = fs::read_to_string("./config.toml").expect("Could not find config file");
let config = config::Config::load(initial_config.to_string());
let waiter_pool = runtime::Builder::new_multi_thread()
.worker_threads(config.systems.len()).build().unwrap();
let mut waiters = JoinSet::new();
for (system_name, system_config) in config.systems.into_iter() {
let system_handle = spawn_system(&system_name, system_config);
waiters.spawn_blocking_on(move || -> String {
system_handle.join();
system_name
}, waiter_pool.handle());
}
runtime::Builder::new_current_thread().build().unwrap().block_on(async {
while let Some(system_join) = waiters.join_next().await {
if let Ok(system_name) = system_join {
println!("Thread joined for system: {}. Updating config and restarting.", system_name);
let config_file = if let Ok(config_file) = fs::read_to_string("./config.toml") {
config_file
} else {
println!("Could not open config file, continuing with initial config");
initial_config.clone()
};
let updated_config = config::Config::load(config_file);
if let Some((_, system_config)) = updated_config.systems.into_iter().find(|(name, _)| name.eq(&system_name)) {
let system_handle = spawn_system(&system_name, system_config.clone());
waiters.spawn_blocking_on(move || -> String {
system_handle.join();
system_name
}, waiter_pool.handle());
} else {
println!("New config file but this system no longer exists, exiting.");
}
}
}
})
}
fn spawn_system(system_name : &String, system_config: config::System) -> thread::JoinHandle<()> {
let name = system_name.clone();
thread::spawn(move || {
let thread_local_runtime = runtime::Builder::new_current_thread().enable_all().build().unwrap();
thread_local_runtime.block_on(async {
let mut system = system::System::new(name, system_config);
system.start_clients().await;
});
})
}
|
