clone/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-04-27 08:05:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Ahmed Ibrahim
2026-02-17 15:15:05 -08:00
parent 2c1d6af908
commit f26c001568
2 changed files with 116 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
codex-rs/codex-api/src/endpoint/realtime_websocket/methods.rs
View File

@@ -320,30 +320,38 @@ impl RealtimeWebsocketEvents {
return Ok(None);
}
let msg = match self.rx_message.lock().await.recv().await {
Some(Ok(msg)) => msg,
Some(Err(err)) => {
self.is_closed.store(true, Ordering::SeqCst);
return Err(ApiError::Stream(format!(
"failed to read websocket message: {err}"
)));
}
None => {
self.is_closed.store(true, Ordering::SeqCst);
return Ok(None);
}
};
loop {
let msg = match self.rx_message.lock().await.recv().await {
Some(Ok(msg)) => msg,
Some(Err(err)) => {
self.is_closed.store(true, Ordering::SeqCst);
return Err(ApiError::Stream(format!(
"failed to read websocket message: {err}"
)));
}
None => {
self.is_closed.store(true, Ordering::SeqCst);
return Ok(None);
}
};
match msg {
Message::Text(text) => Ok(parse_realtime_event(&text)),
Message::Close(_) => {
self.is_closed.store(true, Ordering::SeqCst);
Ok(None)
match msg {
Message::Text(text) => {
if let Some(event) = parse_realtime_event(&text) {
return Ok(Some(event));
}
}
Message::Close(_) => {
self.is_closed.store(true, Ordering::SeqCst);
return Ok(None);
}
Message::Binary(_) => {
return Ok(Some(RealtimeEvent::Error(
"unexpected binary realtime websocket event".to_string(),
)));
}
Message::Frame(_) | Message::Ping(_) | Message::Pong(_) => {}
}
Message::Binary(_) => Ok(Some(RealtimeEvent::Error(
"unexpected binary realtime websocket event".to_string(),
))),
Message::Frame(_) | Message::Ping(_) | Message::Pong(_) => Ok(None),
}
}
}