fix: dont panic when in loop

BUAA-SE-Compiling · Oct 5, 2020 · 566f983 · 566f983
1 parent cef0beb
commit 566f983
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/judger/src/main.rs b/judger/src/main.rs
@@ -147,9 +147,12 @@ async fn client(cmd: opt::ConnectSubCmd) {
     ABORT_HANDLE.set(handle).unwrap();
 
     let sink = loop {
-        let (sink, stream) = connect_to_coordinator(&client_config)
-            .await
-            .expect("Failed to connect");
+        let (sink, stream) = match connect_to_coordinator(&client_config).await {
+            Ok(e) => e,
+            Err(e) => {
+                continue;
+            }
+        };
         let sink = client_loop(stream, sink, client_config.clone()).await;
         if client_config.cancel_handle.is_cancelled() {
             break sink;