ws_coin: better error handling

WestXu · May 7, 2024 · b60ddc5 · b60ddc5
1 parent 1517b38
commit b60ddc5
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/src/ws_coin/mod.rs b/src/ws_coin/mod.rs
@@ -80,7 +80,7 @@ impl WsCoin {
                     break;
                 }
                 Err(error) => {
-                    println!("{}", error);
+                    println!("error happened during connection: {error}, retrying...");
                 }
             }
         }
@@ -145,12 +145,7 @@ impl WsCoin {
                     Ok(price) => yield price,
                     Err(error) => {
                         println!("{}\n\n\n\n\n\n\n\n", error);
-                        match error {
-                            RecvError::RecevingError(_) => {
-                                self.reconnect().await;
-                            }
-                            RecvError::UnexpectedMsgError(_) | RecvError::ParsingError(_) => (),
-                        }
+                        self.reconnect().await
                     }
                 }
             }