impl Address for http::Uri

actix-connect/src/connect.rs

@@ -61,7 +61,9 @@ impl<T: Address> Connect<T> {
         }
     }
 
-    /// Set port
+    /// Use port if address does not provide one.
+    ///
+    /// By default it set to 0
     pub fn set_port(mut self, port: u16) -> Self {
         self.port = port;
         self

actix-connect/src/lib.rs

@@ -14,6 +14,9 @@ mod error;
 mod resolver;
 pub mod ssl;
 
+#[cfg(feature = "uri")]
+mod uri;
+
 pub use trust_dns_resolver::{error::ResolveError, AsyncResolver};
 
 pub use self::connect::{Address, Connect, Connection};

actix-connect/src/uri.rs

@@ -0,0 +1,35 @@
+use http::Uri;
+
+use crate::Address;
+
+impl Address for Uri {
+    fn host(&self) -> &str {
+        self.host().unwrap_or("")
+    }
+
+    fn port(&self) -> Option<u16> {
+        if let Some(port) = self.port_u16() {
+            Some(port)
+        } else {
+            port(self.scheme_str())
+        }
+    }
+}
+
+// TODO: load data from file
+fn port(scheme: Option<&str>) -> Option<u16> {
+    if let Some(scheme) = scheme {
+        match scheme {
+            "http" => Some(80),
+            "https" => Some(443),
+            "ws" => Some(80),
+            "wss" => Some(443),
+            "amqp" => Some(5672),
+            "mqtt" => Some(1883),
+            "mqtts" => Some(8883),
+            _ => None,
+        }
+    } else {
+        None
+    }
+}