Added cli options

Author: r2dev2

Cargo.toml

@@ -11,6 +11,7 @@ env_logger = "0.7"
 futures = "0.3.1"
 tokio = { version="0.2", features=["full"] }
 actix-cors="0.2.0"
+clap = "2.33.3"
 
 [dependencies.serde]
 version = "1.0"

src/main.rs

@@ -16,6 +16,7 @@ use std::task::{Context, Poll};
 use actix_cors::Cors;
 use actix_web::web::{Bytes, Data, Path};
 use actix_web::{get, post, web, App, Error, HttpResponse, HttpServer, Responder};
+use clap::{crate_version, clap_app};
 use futures::Stream;
 use serde::{Deserialize, Serialize};
 use tokio::sync::broadcast::{channel, Receiver, Sender};
@@ -25,8 +26,17 @@ use tokio::sync::broadcast::{channel, Receiver, Sender};
 async fn main() -> std::io::Result<()> {
     env_logger::init();
 
+    let matches = clap_app!(myapp =>
+        (version: crate_version!())
+        (author: "Ronak B.")
+        (about: "A sse-based pubsub with different channels")
+        (@arg HOST: --host +takes_value "Address to host on")
+        (@arg PORT: --port +takes_value "Port to host on")
+    ).get_matches();
+
     let data = BroadcasterMap::create();
 
+
     HttpServer::new(move || {
         App::new()
             .wrap(
@@ -39,7 +49,9 @@ async fn main() -> std::io::Result<()> {
             .service(new_client)
             .service(broadcast)
     })
-    .bind("127.0.0.1:8080")?
+    .bind(format!("{}:{}",
+            matches.value_of("HOST").unwrap_or("127.0.0.1"),
+            matches.value_of("PORT").unwrap_or("8080")))?
     .run()
     .await
 }