feat: add graceful shutdown of base node

tari-project · Jan 11, 2023 · c9797c5 · c9797c5
1 parent b3380c7
commit c9797c5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/applications/tari_base_node/src/commands/cli_loop.rs b/applications/tari_base_node/src/commands/cli_loop.rs
@@ -177,7 +177,7 @@ impl CliLoop {
             if let Err(err) = self.context.handle_command_str(line).await {
                 println!("Wrong command to watch `{}`. Failed with: {}", line, err);
             } else {
-                loop {
+                while !self.done {
                     let interval = time::sleep(interval);
                     tokio::select! {
                         _ = interval => {
@@ -188,6 +188,9 @@ impl CliLoop {
                         },
                         _ = &mut interrupt => {
                             break;
+                        },
+                        _ = self.shutdown_signal.wait() => {
+                            self.done = true;
                         }
                     }
                 }