44 lines
1 KiB
Rust
44 lines
1 KiB
Rust
use std::{
|
|
net::SocketAddr,
|
|
sync::{Arc, atomic::Ordering},
|
|
};
|
|
|
|
use axum::Router;
|
|
use axum_server::{Handle as ServerHandle, bind};
|
|
use conduwuit::{Result, Server, debug_info, info};
|
|
use tokio::task::JoinSet;
|
|
|
|
pub(super) async fn serve(
|
|
server: &Arc<Server>,
|
|
app: Router,
|
|
handle: ServerHandle,
|
|
addrs: Vec<SocketAddr>,
|
|
) -> Result<()> {
|
|
let app = app.into_make_service_with_connect_info::<SocketAddr>();
|
|
let mut join_set = JoinSet::new();
|
|
for addr in &addrs {
|
|
join_set
|
|
.spawn_on(bind(*addr).handle(handle.clone()).serve(app.clone()), server.runtime());
|
|
}
|
|
|
|
info!("Listening on {addrs:?}");
|
|
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(())
|
|
}
|