diff --git a/src/bin/relay.rs b/src/bin/relay.rs
index 764e4f1..972928f 100644
--- a/src/bin/relay.rs
+++ b/src/bin/relay.rs
@@ -244,8 +244,7 @@ async fn handle_all (req: Request
, state: Arc )
}
}
-#[tokio::main]
-async fn main () -> Result <(), Box > {
+async fn relay_main () -> Result <(), Box > {
let addr = SocketAddr::from(([0, 0, 0, 0], 4000));
let state = Arc::new (ServerState::default ());
@@ -268,3 +267,8 @@ async fn main () -> Result <(), Box > {
Ok (())
}
+
+#[tokio::main]
+async fn main () -> Result <(), Box > {
+ relay_main ().await
+}
diff --git a/src/bin/server.rs b/src/bin/server.rs
index 6a23999..5ab7391 100644
--- a/src/bin/server.rs
+++ b/src/bin/server.rs
@@ -55,8 +55,7 @@ async fn handle_req_resp (
Ok (())
}
-#[tokio::main]
-async fn main () -> Result <(), Box > {
+async fn server_main () -> Result <(), Box > {
let client = Arc::new (Client::new ());
let mut backoff_delay = 0;
@@ -91,3 +90,8 @@ async fn main () -> Result <(), Box > {
});
}
}
+
+#[tokio::main]
+async fn main () -> Result <(), Box > {
+ server_main ().await
+}