romenskiy2012/tuwunel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Add active request assert on tls listener shutdown

Browse source
This commit is contained in:
dasha_uwu 2025-08-28 23:11:09 +05:00 • committed by Jason Volk
parent 33a16f4831
commit 54ce221e88
1 changed files with 24 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/router/serve/tls.rs
View file

@ -1,4 +1,7 @@
use std::{net::SocketAddr, sync::Arc};
use std::{
net::SocketAddr,
sync::{Arc, atomic::Ordering},
};
use axum::Router;
use axum_server::Handle as ServerHandle;
@ -7,7 +10,7 @@ use axum_server_dual_protocol::{
axum_server::{bind_rustls, tls_rustls::RustlsConfig},
};
use tokio::task::JoinSet;
use tuwunel_core::{Result, Server, debug, err, info, warn};
use tuwunel_core::{Result, Server, debug, debug_info, err, info, warn};
pub(super) async fn serve(
server: &Arc<Server>,
@ -73,5 +76,24 @@ pub(super) async fn serve(
while join_set.join_next().await.is_some() {}
let handle_active = server
.metrics
.requests_handle_active
.load(Ordering::Relaxed);
debug_info!(
handle_finished = server
.metrics
.requests_handle_finished
.load(Ordering::Relaxed),
panics = server
.metrics
.requests_panic
.load(Ordering::Relaxed),
handle_active,
"Stopped listening on {addrs:?}",
);
debug_assert!(handle_active == 0, "active request handles still pending");
Ok(())
}