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 +}