fix it

Signed-off-by: Bugen Zhao <[email protected]>
risingwavelabs · Jul 25, 2024 · c36de8a · c36de8a
1 parent 5a55937
commit c36de8a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/utils/runtime/src/lib.rs b/src/utils/runtime/src/lib.rs
@@ -97,13 +97,13 @@ where
         spawn_prof_thread(profile_path);
     }
 
-    let mut sigint = tokio::signal::unix::signal(SignalKind::interrupt()).unwrap();
-    let mut sigterm = tokio::signal::unix::signal(SignalKind::terminate()).unwrap();
-
     let future_with_shutdown = async move {
         let shutdown = CancellationToken::new();
         let mut fut = pin!(f(shutdown.clone()));
 
+        let mut sigint = tokio::signal::unix::signal(SignalKind::interrupt()).unwrap();
+        let mut sigterm = tokio::signal::unix::signal(SignalKind::terminate()).unwrap();
+
         tokio::select! {
             biased;