Skip to content

Commit 76ab5e1

Browse files
committed
Attempt to make tests more robust: explicitly send WS close when tarmination flag is set in API server.
1 parent 897e160 commit 76ab5e1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

server/src/api_server/mod.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -215,6 +215,10 @@ async fn handle_ws_session(
215215
_ = sleep(Duration::from_millis(100)) => {
216216
if server.terminate_flag.load(Relaxed) {
217217
tracing::debug!("Termination flag set. Closing session.");
218+
// Send WebSocket close frame for proper close handshake
219+
if let Err(e) = ws_tx.send(Message::close()).await {
220+
tracing::debug!(details=%e, "Failed to send close frame (client likely disconnected).");
221+
}
218222
break;
219223
}},
220224

0 commit comments

Comments
 (0)