cubixle/radius-rs
1
0
Fork 0
mirror of https://github.com/cubixle/radius-rs.git synced 2026-04-24 22:34:42 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use tokio::select

Browse Source
This commit is contained in:
moznion
2020-11-23 11:40:22 +09:00
parent 6164f7b597
commit 1d8f3b5daf
1 changed files with 28 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
src/server.rs
View File

@@ -39,31 +39,36 @@ impl<T: RequestHandler, U: SecretProvider> Server<T, U> {
loop {
let conn = conn_arc.clone();
let (size, remote_addr) = conn.recv_from(&mut buf).await?;
let request_data = buf[..size].to_vec();
tokio::select! {
received = conn.recv_from(&mut buf) => {
let (size, remote_addr) = received?;
let local_addr = match conn.local_addr() {
Ok(addr) => addr,
Err(e) => {
error!("failed to get a local address from from a connection; {}", e);
continue;
let request_data = buf[..size].to_vec();
let local_addr = match conn.local_addr() {
Ok(addr) => addr,
Err(e) => {
error!("failed to get a local address from from a connection; {}", e);
continue;
}
};
let undergoing_requests_lock = undergoing_requests_lock_arc.clone();
tokio::spawn(async move {
Self::process_request(
conn,
&request_data,
local_addr,
remote_addr,
undergoing_requests_lock,
self.request_handler,
self.secret_provider,
self.skip_authenticity_validation,
).await;
});
}
};
let undergoing_requests_lock = undergoing_requests_lock_arc.clone();
tokio::spawn(async move {
Self::process_request(
conn,
&request_data,
local_addr,
remote_addr,
undergoing_requests_lock,
self.request_handler,
self.secret_provider,
self.skip_authenticity_validation,
).await;
});
}
}
}