retyrn on broken pipe

risingwavelabs · Feb 22, 2024 · dfbd4dd · dfbd4dd
1 parent 00bd55c
commit dfbd4dd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/tests/simulation/src/slt.rs b/src/tests/simulation/src/slt.rs
@@ -274,10 +274,13 @@ pub async fn run_slt_task(
                         })
                         .await
                     {
+                        let err_string = err.to_string();
                         // cluster could be still under recovering if killed before, retry if
                         // meets `no reader for dml in table with id {}`.
-                        let should_retry =
-                            err.to_string().contains("no reader for dml in table") && i < MAX_RETRY;
+                        let should_retry = (err_string.contains("no reader for dml in table")
+                            || err_string
+                                .contains("error reading a body from connection: broken pipe"))
+                            && i < MAX_RETRY;
                         if !should_retry {
                             panic!("{}", err);
                         }