mirrors/magic-nix-cache
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Don't run server in tokio::spawn

Browse source
This commit is contained in:
Luc Perkins 2024-05-17 15:41:45 -03:00
parent 2e05bd5fff
commit d1c5d5203b
No known key found for this signature in database
GPG key ID: 16DB1108FB591835
1 changed files with 9 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
magic-nix-cache/src/main.rs
View file

@ -371,21 +371,6 @@ async fn main_cli() -> Result<()> {
tracing::info!("Listening on {}", args.listen); tracing::info!("Listening on {}", args.listen);
let server = axum::Server::bind(&args.listen)
.serve(app.into_make_service())
.with_graceful_shutdown(async move {
shutdown_receiver.await.ok();
tracing::info!("Shutting down");
});
// Spawn here so that post-startup tasks can proceed
tokio::spawn(async move {
if let Err(e) = server.await {
tracing::error!("failed to start up daemon: {e}");
exit(1);
}
});
// Notify of startup via HTTP // Notify of startup via HTTP
if let Some(startup_notification_url) = args.startup_notification_url { if let Some(startup_notification_url) = args.startup_notification_url {
tracing::debug!("Startup notification via HTTP POST to {startup_notification_url}"); tracing::debug!("Startup notification via HTTP POST to {startup_notification_url}");
@ -427,11 +412,20 @@ async fn main_cli() -> Result<()> {
tracing::debug!("Created startup notification file at {startup_notification_file_path:?}"); tracing::debug!("Created startup notification file at {startup_notification_file_path:?}");
} }
let ret = axum::Server::bind(&args.listen)
.serve(app.into_make_service())
.with_graceful_shutdown(async move {
shutdown_receiver.await.ok();
tracing::info!("Shutting down");
})?;
// Notify diagnostics endpoint // Notify diagnostics endpoint
if let Some(diagnostic_endpoint) = diagnostic_endpoint { if let Some(diagnostic_endpoint) = diagnostic_endpoint {
state.metrics.send(diagnostic_endpoint).await; state.metrics.send(diagnostic_endpoint).await;
} }
ret?;
Ok(()) Ok(())
} }