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

refactor main task stack through service mgr

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-07-13 06:07:09 +00:00
parent b8baa1223d
commit 5be679e17b
4 changed files with 93 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/router/serve/mod.rs
View file

@ -4,23 +4,23 @@ mod unix;
use std::sync::Arc;
use axum::Router;
use axum_server::Handle as ServerHandle;
use conduit::{Error, Result, Server};
use conduit::{Result, Server};
use tokio::sync::broadcast;
use crate::layers;
/// Serve clients
pub(super) async fn serve(
server: &Arc<Server>, app: Router, handle: ServerHandle, shutdown: broadcast::Receiver<()>,
) -> Result<(), Error> {
pub(super) async fn serve(server: Arc<Server>, handle: ServerHandle, shutdown: broadcast::Receiver<()>) -> Result<()> {
let config = &server.config;
let addrs = config.get_bind_addrs();
let app = layers::build(&server)?;
if cfg!(unix) && config.unix_socket_path.is_some() {
unix::serve(server, app, shutdown).await
unix::serve(&server, app, shutdown).await
} else if config.tls.is_some() {
tls::serve(server, app, handle, addrs).await
tls::serve(&server, app, handle, addrs).await
} else {
plain::serve(server, app, handle, addrs).await
plain::serve(&server, app, handle, addrs).await
}
}