wait_for_arm: properly retry without timeout

LiberTEM · Jul 24, 2024 · 6bfe1fb · 6bfe1fb
1 parent 0f820ee
commit 6bfe1fb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/common/src/generic_connection.rs b/common/src/generic_connection.rs
@@ -253,10 +253,10 @@ where
             // wait indefinitely:
             loop {
                 let res = self.wait_for_arm_inner(Duration::from_millis(100), &periodic_callback);
-                if let Err(ConnectionError::Timeout) = &res {
-                    continue;
-                } else {
-                    break res;
+                match res {
+                    Err(ConnectionError::Timeout) => continue,
+                    Ok(None) => continue,
+                    other => break other, // error, or Some(result)
                 }
             }
         }