Upgrade main.rs according to the changes

Author: silvestrpredko

Cargo.lock

@@ -16,6 +16,7 @@ tokio = { version = "1.36", features = ["rt", "time", "net", "io-util", "macros"
 serde = { version = "1.0.196", features = ["derive"] }
 serde_json = "1.0.113"
 bytes = { version = "1", features = ["serde"] }
+ctrlc = "3.4"
 flume = { version = "0.11.0", default-features = false, features = ["async"]}
 slab = "0.4.9"
 thiserror = "1.0.57"

rumqttd/Cargo.toml

@@ -84,10 +84,15 @@ fn main() {
 
     validate_config(&configs);
 
-    // println!("{:#?}", configs);
+    let _guard = Broker::new(configs).start().unwrap().drop_guard();
 
-    let mut broker = Broker::new(configs);
-    broker.start().unwrap();
+    let (tx, rx) = flume::bounded::<()>(1);
+    ctrlc::set_handler(move || {
+        let _ = tx.send(());
+    })
+    .expect("Error setting Ctrl-C handler");
+
+    rx.recv().expect("Could not receive Ctrl-C signal");
 }
 
 // Do any extra validation that needs to be done before starting the broker here.