main: Use new server spawner.

chainwayxyz · Aug 26, 2024 · 95aa1fc · 95aa1fc
1 parent a517786
commit 95aa1fc
Showing 1 changed file with 6 additions and 11 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -1,25 +1,20 @@
 //! # RPC Server Starter
 
 use bitcoin_mock_rpc::rpc::spawn_rpc_server;
-use std::{env, time::Duration};
+use std::env;
 
-#[tokio::main]
-async fn main() {
+fn main() {
     println!("Bitcoin Mock Rpc (C) Chainway, 2024");
     println!(
         "Usage: {} [HOST] [PORT]",
         env::args().collect::<Vec<String>>().first().unwrap()
     );
 
     let server_info = handle_args();
-    let address = spawn_rpc_server(server_info.0.as_deref(), server_info.1)
-        .await
-        .unwrap();
-    println!("Server started at {}", address);
-
-    loop {
-        std::thread::sleep(Duration::from_secs(100));
-    }
+    let server = spawn_rpc_server(server_info.0.as_deref(), server_info.1).unwrap();
+    println!("Server started at {}", server.0);
+
+    server.1.join().unwrap()
 }
 
 fn handle_args() -> (Option<String>, Option<u16>) {