YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-01 18:25:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

udp: add config key worker_channel_size to enable bounded channels

Browse source
This commit is contained in:
Joakim Frostegård 2021-11-28 21:59:11 +01:00
parent abea88dbf8
commit 2fe57b9f67
3 changed files with 33 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

22
aquatic_udp/src/common.rs
View file

@ -5,7 +5,7 @@ use std::sync::atomic::AtomicUsize;
use std::sync::Arc;
use std::time::Instant;
use crossbeam_channel::Sender;
use crossbeam_channel::{Sender, TrySendError};
use aquatic_common::access_list::{create_access_list_cache, AccessListArcSwap};
use aquatic_common::AHashIndexMap;
@ -88,8 +88,14 @@ impl ConnectedRequestSender {
request: ConnectedRequest,
addr: SocketAddr,
) {
if let Err(err) = self.senders[index.0].try_send((self.index, request, addr)) {
::log::warn!("request_sender.try_send failed: {:?}", err)
match self.senders[index.0].try_send((self.index, request, addr)) {
Ok(()) => {},
Err(TrySendError::Full(_)) => {
::log::error!("Request channel {} is full, dropping request. Try increasing number of request workers or raising config.worker_channel_size.", index.0)
}
Err(TrySendError::Disconnected(_)) => {
panic!("Request channel {} is disconnected", index.0);
}
}
}
}
@ -109,8 +115,14 @@ impl ConnectedResponseSender {
response: ConnectedResponse,
addr: SocketAddr,
) {
if let Err(err) = self.senders[index.0].try_send((response, addr)) {
::log::warn!("request_sender.try_send failed: {:?}", err)
match self.senders[index.0].try_send((response, addr)) {
Ok(()) => {},
Err(TrySendError::Full(_)) => {
::log::error!("Response channel {} is full, dropping response. Try increasing number of socket workers or raising config.worker_channel_size.", index.0)
}
Err(TrySendError::Disconnected(_)) => {
panic!("Response channel {} is disconnected", index.0);
}
}
}
}