From: MegaBrutal <megabrutal+github@megabrutal.com>
Date: Sat, 18 Feb 2023 00:40:00 +0000 (+0100)
Subject: Query multiple addresses from command line
X-Git-Tag: v0.2.0~10
X-Git-Url: http://git.megabrutal.com/?p=hlquery.git;a=commitdiff_plain;h=1377ceefd2999d268d1ae58065cd9136d2bb02ab

Query multiple addresses from command line
---

diff --git a/src/main.rs b/src/main.rs
index 0255860..4bd31f3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -4,13 +4,13 @@ use a2s::A2SClient;
 
 fn main() {
     let client = A2SClient::new().unwrap();
-    let addresses = args().nth(1)
-        .unwrap_or_else(|| String::from("localhost:27015"))
-        .to_socket_addrs().unwrap()
-        .flat_map(|sa| match sa {
-            SocketAddr::V4(sa4) => Some(sa4),
-            _ => None
-        });
+    let addresses = args().skip(1)
+        .flat_map(|arg| arg.to_socket_addrs())
+        .flat_map(|iter_addr| iter_addr.flat_map(|sa| match sa {
+                SocketAddr::V4(sa4) => Some(sa4),
+                _ => None
+            })
+        );
 
     for address in addresses {