Put the listener / ret bit back to where it was

Author: grahamc

magic-nix-cache/src/main.rs

@@ -484,11 +484,6 @@ async fn main_cli(args: Args, recorder: detsys_ids_client::Recorder) -> Result<(
 
     let app = app.layer(Extension(state.clone()));
 
-    let serve = axum::serve(listener, app.into_make_service()).with_graceful_shutdown(async move {
-        shutdown_token.cancelled_owned().await;
-        tracing::info!("Shutting down");
-    });
-
     let startup_blob = serde_json::json!({
         "address": listener_addr
     });
@@ -557,10 +552,16 @@ async fn main_cli(args: Args, recorder: detsys_ids_client::Recorder) -> Result<(
         tracing::debug!("Created startup notification file at {startup_notification_file_path:?}");
     }
 
+    let listener = tokio::net::TcpListener::bind(&args.listen).await?;
+    let ret = axum::serve(listener, app.into_make_service()).with_graceful_shutdown(async move {
+        shutdown_token.cancelled_owned().await;
+        tracing::info!("Shutting down");
+    });
+
     // Notify diagnostics endpoint
     state.metrics.send().await;
 
-    serve.await?;
+    ret.await?;
 
     Ok(())
 }