Split logic in multiple files!

1 files changed, 2 insertions, 3 deletions

diff --git a/src/main.rs b/src/main.rs
index 471573b..91074d1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,17 +3,17 @@ use std::net::SocketAddrV4;
 
 use log::*;
 
-use igd::aio::search_gateway;
 use igd::PortMappingProtocol;
 
 mod diplonat;
 mod config;
+mod gw;
 
 #[tokio::main]
 async fn main() {
     pretty_env_logger::init();
 
-    let ctx = diplonat::setup().expect("Setup failed");
+    let ctx = diplonat::setup().await.expect("Setup failed");
     diplonat::listen();
 /*
     let url = format!("http://127.0.0.1:8500/v1/catalog/node/{}", config.consul_node_name);
@@ -28,7 +28,6 @@ async fn main() {
         Ok(g) => g,
         Err(err) => return println!("Faild to find IGD: {}", err),
     };
-    info!("Gateway: {}", gateway);
 
     let service = format!("{}:{}", config.private_ip, 1234);
     let service: SocketAddrV4 = service.parse().expect("Invalid socket address");