handle ctrlc

src/bin/server/main.rs

@@ -17,7 +17,7 @@ use std::{
         atomic::{AtomicU64, Ordering},
     },
 };
-use tokio::sync::RwLock;
+use tokio::{signal::ctrl_c, sync::RwLock};
 
 #[derive(Parser, Debug)]
 #[command(version, about, long_about = None)]
@@ -42,9 +42,18 @@ pub async fn run_server(port: u16) {
         msgs: db.open_tree("msgs").unwrap(),
         senders: Default::default(),
         count: 0.into(),
-        db,
+        db: db.clone(),
     };
-    listen(port, path, handler).await;
+    let (endpoint, handle) = listen(port, path, handler);
+    let _ = ctrl_c().await;
+    println!("stopping server");
+    println!("closing connections...");
+    endpoint.close(0u32.into(), &[]);
+    let _ = handle.await;
+    endpoint.wait_idle().await;
+    println!("saving...");
+    db.flush_async().await.unwrap();
+    println!("saved");
 }
 
 type ClientId = u64;