tyler/proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

proxy: log if there are too many open connections

Browse Source
This commit is contained in:
Dietmar Maurer 2020-11-04 08:49:35 +01:00
parent 0081903f7c
commit ea93bea7bf
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/bin/proxmox-backup-proxy.rs
View File

@ -167,12 +167,12 @@ fn accept_connections(
debug: bool, debug: bool,
) -> tokio::sync::mpsc::Receiver<Result<tokio_openssl::SslStream<tokio::net::TcpStream>, Error>> { ) -> tokio::sync::mpsc::Receiver<Result<tokio_openssl::SslStream<tokio::net::TcpStream>, Error>> {
let (sender, receiver) = tokio::sync::mpsc::channel(1024); const MAX_PENDING_ACCEPTS: usize = 1024;
let (sender, receiver) = tokio::sync::mpsc::channel(MAX_PENDING_ACCEPTS);
let accept_counter = Arc::new(()); let accept_counter = Arc::new(());
const MAX_PENDING_ACCEPTS: usize = 1024;
tokio::spawn(async move { tokio::spawn(async move {
loop { loop {
match listener.accept().await { match listener.accept().await {
@ -186,9 +186,7 @@ fn accept_connections(
let mut sender = sender.clone(); let mut sender = sender.clone();
if Arc::strong_count(&accept_counter) > MAX_PENDING_ACCEPTS { if Arc::strong_count(&accept_counter) > MAX_PENDING_ACCEPTS {
if debug {
eprintln!("connection rejected - to many open connections"); eprintln!("connection rejected - to many open connections");
}
continue; continue;
} }