Mirrors/rustdesk-server
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk-server.git synced 2026-03-10 14:03:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor stop

Browse Source
This commit is contained in:
open-trade
2021-03-25 17:55:44 +08:00
parent 3d1260c8ad
commit 9132d85f74
2 changed files with 47 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/relay_server.rs
View File

@@ -6,6 +6,7 @@ use hbb_common::{
tcp::{new_listener, FramedStream},
tokio::{
self,
net::TcpListener,
time::{interval, Duration},
},
ResultType,
@@ -26,8 +27,19 @@ pub const DEFAULT_PORT: &'static str = "21117";
pub async fn start(port: &str, license: &str, stop: Arc<Mutex<bool>>) -> ResultType<()> {
let addr = format!("0.0.0.0:{}", port);
log::info!("Listening on tcp {}", addr);
let mut timer = interval(Duration::from_millis(300));
let mut listener = new_listener(addr, false).await?;
loop {
if *stop.lock().unwrap() {
sleep(0.1).await;
continue;
}
log::info!("Start");
io_loop(&mut listener, license, stop.clone()).await;
}
}
async fn io_loop(listener: &mut TcpListener, license: &str, stop: Arc<Mutex<bool>>) {
let mut timer = interval(Duration::from_millis(100));
loop {
tokio::select! {
Ok((stream, addr)) = listener.accept() => {
@@ -44,7 +56,6 @@ pub async fn start(port: &str, license: &str, stop: Arc<Mutex<bool>>) -> ResultT
}
}
}
Ok(())
}
async fn make_pair(stream: FramedStream, addr: SocketAddr, license: &str) -> ResultType<()> {