enterprise scale rust

hptp-recv/src/main.rs

@@ -1,3 +1,5 @@
+use std::str::FromStr;
+
 extern crate hptp;
 
 #[derive(Debug)]
@@ -6,8 +8,17 @@ struct Hostname {
     port: u16,
 }
 
-fn go(hn: Hostname) {
+struct InvalidHostnameError;
-    println!("ip={}, port={}", hn.ip, hn.port)
+
+impl FromStr for Hostname {
+    type Err = InvalidHostnameError;
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        let i = s.find(':').ok_or(InvalidHostnameError)?;
+        Ok(Hostname {
+            ip: s[..i].into(),
+            port: s[i + 1..].parse().map_err(|_| InvalidHostnameError)?,
+        })
+    }
 }
 
 fn usage() {
@@ -15,20 +26,16 @@ fn usage() {
 }
 
 fn parse_args() -> Result<Hostname, ()> {
-    let mut args = std::env::args();
+    std::env::args().nth(1).ok_or(())?.parse().map_err(|_| ())
-    let arg = args.nth(1).ok_or(())?;
-    let i = arg.find(':').ok_or(())?;
-    let ip = &arg[..i];
-    let port: u16 = arg[i + 1..].parse().map_err(|_| ())?;
-    Ok(Hostname {
-        ip: ip.into(),
-        port,
-    })
 }
 
 fn main() {
     match parse_args() {
-        Ok(x) => go(x),
+        Ok(x) => start(x),
         Err(_) => usage(),
     }
 }
+
+fn start(hn: Hostname) {
+    println!("ip={}, port={}", hn.ip, hn.port)
+}