hongshaorou/rustdesk
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
open-trade 2022-01-04 17:45:52 +08:00
parent fb392b81cb
commit 533efd04d7
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/rendezvous_mediator.rs
View File

@ -30,7 +30,7 @@ type Message = RendezvousMessage;
lazy_static::lazy_static! { lazy_static::lazy_static! {
pub static ref SOLVING_PK_MISMATCH: Arc<Mutex<String>> = Default::default(); pub static ref SOLVING_PK_MISMATCH: Arc<Mutex<String>> = Default::default();
} }
static EXITED: AtomicBool = AtomicBool::new(false); static SHOULD_EXIT: AtomicBool = AtomicBool::new(false);
#[derive(Clone)] #[derive(Clone)]
pub struct RendezvousMediator { pub struct RendezvousMediator {
@ -59,14 +59,14 @@ impl RendezvousMediator {
} }
let mut futs = Vec::new(); let mut futs = Vec::new();
let servers = Config::get_rendezvous_servers(); let servers = Config::get_rendezvous_servers();
EXITED.store(false, Ordering::SeqCst); SHOULD_EXIT.store(false, Ordering::SeqCst);
for host in servers.clone() { for host in servers.clone() {
let server = server.clone(); let server = server.clone();
let servers = servers.clone(); let servers = servers.clone();
futs.push(tokio::spawn(async move { futs.push(tokio::spawn(async move {
allow_err!(Self::start(server, host, servers).await); allow_err!(Self::start(server, host, servers).await);
// EXITED here is to ensure once one exits, the others also exit. // SHOULD_EXIT here is to ensure once one exits, the others also exit.
EXITED.store(true, Ordering::SeqCst); SHOULD_EXIT.store(true, Ordering::SeqCst);
})); }));
} }
join_all(futs).await; join_all(futs).await;
@ -224,7 +224,7 @@ impl RendezvousMediator {
if !Config::get_option("stop-service").is_empty() { if !Config::get_option("stop-service").is_empty() {
break; break;
} }
if EXITED.load(Ordering::SeqCst) { if SHOULD_EXIT.load(Ordering::SeqCst) {
break; break;
} }
if rz.addr.port() == 0 { if rz.addr.port() == 0 {