magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

move unix socket unlink from services to router

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-05-31 00:43:30 +00:00 • committed by June 🍓🦴
parent 2e45cb281a
commit 6d1144bb69
3 changed files with 24 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

29
src/router/serve/unix.rs
View file

@ -11,6 +11,7 @@ use hyper_util::{
};
use tokio::{
fs,
net::{unix::SocketAddr, UnixListener, UnixStream},
sync::broadcast::{self},
task::JoinSet,
};
@ -37,16 +38,14 @@ pub(super) async fn serve(
}
}
drop(listener);
tasks.shutdown().await;
fini(listener, tasks).await;
Ok(())
}
async fn accept(
server: &Arc<Server>, listener: &tokio::net::UnixListener, tasks: &mut JoinSet<()>,
mut app: IntoMakeService<Router>, builder: server::conn::auto::Builder<TokioExecutor>,
conn: (tokio::net::UnixStream, tokio::net::unix::SocketAddr),
server: &Arc<Server>, listener: &UnixListener, tasks: &mut JoinSet<()>, mut app: IntoMakeService<Router>,
builder: server::conn::auto::Builder<TokioExecutor>, conn: (UnixStream, SocketAddr),
) {
let (socket, remote) = conn;
let socket = TokioIo::new(socket);
@ -67,7 +66,7 @@ async fn accept(
while tasks.try_join_next().is_some() {}
}
async fn init(server: &Arc<Server>) -> Result<tokio::net::UnixListener> {
async fn init(server: &Arc<Server>) -> Result<UnixListener> {
use std::os::unix::fs::PermissionsExt;
let config = &server.config;
@ -89,7 +88,7 @@ async fn init(server: &Arc<Server>) -> Result<tokio::net::UnixListener> {
return Err(Error::Err(format!("Failed to create {dir:?} for socket {path:?}: {e}")));
}
let listener = tokio::net::UnixListener::bind(path);
let listener = UnixListener::bind(path);
if let Err(e) = listener {
return Err(Error::Err(format!("Failed to bind listener {path:?}: {e}")));
}
@ -105,3 +104,19 @@ async fn init(server: &Arc<Server>) -> Result<tokio::net::UnixListener> {
Ok(listener.unwrap())
}
async fn fini(listener: UnixListener, mut tasks: JoinSet<()>) {
let local = listener.local_addr();
drop(listener);
tasks.shutdown().await;
if let Ok(local) = local {
if let Some(path) = local.as_pathname() {
debug!(?path, "Removing unix socket file.");
if let Err(e) = fs::remove_file(path).await {
warn!(?path, "Failed to remove UNIX socket file: {e}");
}
}
}
}