feat:add ctrl-c to server

SkyUOI · Jun 22, 2024 · 223c12f · 223c12f
1 parent dd4b534
commit 223c12f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/gdrust/server/src/lib.rs b/gdrust/server/src/lib.rs
@@ -110,6 +110,18 @@ pub async fn lib_main() -> Result<(), Box<dyn Error>> {
             }
         }
     });
+    tokio::spawn(async {
+        match tokio::signal::ctrl_c().await {
+            Ok(()) => {
+                eprintln!("Shutdowning...");
+                exit(0);
+            }
+            Err(err) => {
+                eprintln!("Unable to listen for shutdown signal: {}", err);
+                // we also shut down in case of error
+            }
+        }
+    });
     loop {
         print!(">>>");
         std::io::stdout().flush().unwrap();