Merge pull request PacktPublishing#20 from vbauerster/env-arg-nth

less verbose way to get base_url from env::args

Author: cfsamson

ch04/a-epoll/src/main.rs

@@ -84,13 +84,10 @@ fn main() -> Result<()> {
 
     let mut streams = vec![];
 
-    let args: Vec<String> = env::args().collect();
-    let base_url;
-    if args.len() > 1 {
-        base_url = args[1].clone();
-    } else {
-        base_url = String::from("localhost");
-    }
+    let base_url = env::args()
+        .nth(1)
+        .unwrap_or_else(|| String::from("localhost"));
+
     let addr = format!("{}:8080", &base_url);
 
     for i in 0..n_events {

ch04/b-epoll-mio/src/main.rs

@@ -91,13 +91,10 @@ fn main() -> Result<()> {
 
     let mut streams = vec![];
 
-    let args: Vec<String> = env::args().collect();
-    let base_url;
-    if args.len() > 1 {
-        base_url = args[1].clone();
-    } else {
-        base_url = String::from("localhost");
-    }
+    let base_url = env::args()
+        .nth(1)
+        .unwrap_or_else(|| String::from("localhost"));
+
     let addr = format!("{}:8080", &base_url);
 
     for i in 0..n_events {