add start/stop for external use

This commit is contained in:
open-trade 2021-03-19 17:29:08 +08:00
parent 5277ace042
commit cd241958a2

43
mod.rs
View File

@ -1,2 +1,45 @@
mod src;
pub use src::*;
use hbb_common::{allow_err, log};
use std::sync::{Arc, Mutex};
lazy_static::lazy_static! {
static ref STOP: Arc<Mutex<bool>> = Arc::new(Mutex::new(true));
}
fn is_running() -> bool {
!*STOP.lock().unwrap()
}
pub fn start() {
if is_running() {
return;
}
*STOP.lock().unwrap() = false;
let port = rendezvous_server::DEFAULT_PORT;
let addr = format!("0.0.0.0:{}", port);
let addr2 = format!("0.0.0.0:{}", port.parse::<i32>().unwrap_or(0) - 1);
let relay_servers: Vec<String> = Default::default();
std::thread::spawn(move || {
allow_err!(rendezvous_server::RendezvousServer::start(
&addr,
&addr2,
relay_servers,
0,
Default::default(),
Default::default(),
STOP.clone(),
));
});
std::thread::spawn(|| {
allow_err!(relay_server::start(
relay_server::DEFAULT_PORT,
STOP.clone()
));
});
}
pub fn stop() {
*STOP.lock().unwrap() = true;
}