close Ws if event stream disconnects

drift-labs · Feb 13, 2025 · 8f45f41 · 8f45f41
1 parent d6b0527
commit 8f45f41
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/websocket.rs b/src/websocket.rs
@@ -153,10 +153,8 @@ async fn accept_connection(
                                     warn!(target: LOG_TARGET, "event stream finished: {sub_account_id:?}");
                                     subscription_map.lock().await.remove(&sub_account_id);
                                     // the event subscription task has failed
-                                    // notify the caller to handle reconnecting
-                                    let _ = message_tx.try_send(
-                                        Message::Close(None)
-                                    );
+                                    // close the Ws so client can handle resubscription
+                                    let _ = message_tx.try_send(Message::Close(None));
                                 }
                             });
                             subscription_map.insert(request.sub_account_id, join_handle);