Make client join shards and return

Instead of running a loop forever, have the client join threads when
they end.

This is currently a poor way of having the client eventually end, but
for now it's a duck-tape solution, as when one shard completely fails to
reboot it's an indicator of a larger issue (invalidated authentication,
intermittent network issues, etc.)

1 files changed, 8 insertions, 4 deletions

diff --git a/src/client/mod.rs b/src/client/mod.rs
index ec2af1e..ff5e4bd 100644
--- a/src/client/mod.rs
+++ b/src/client/mod.rs
@@ -937,6 +937,8 @@ impl Client {
         let shards_index = shard_data.map_or(0, |x| x[0]);
         let shards_total = shard_data.map_or(1, |x| x[1] + 1);
 
+        let mut threads = vec![];
+
         for shard_number in shards_index..shards_total {
             let shard_info = shard_data.map(|s| [shard_number, s[2]]);
 
@@ -993,9 +995,9 @@ impl Client {
                         }
                     }};
 
-                    thread::spawn(move || {
+                    threads.push(thread::spawn(move || {
                         monitor_shard(monitor_info);
-                    });
+                    }));
                 },
                 Err(why) => warn!("Error starting shard {:?}: {:?}", shard_info, why),
             }
@@ -1006,9 +1008,11 @@ impl Client {
             thread::sleep(Duration::from_secs(5));
         }
 
-        loop {
-            thread::sleep(Duration::from_secs(1));
+        for thread in threads {
+            let _ = thread.join();
         }
+
+        Ok(())
     }
 }