discours.io/quoter
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix-stream-once

Browse Source
This commit is contained in:
Untone 2023-10-16 23:32:28 +03:00
parent a84bf61e4e
commit 64fcd55305
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main.rs
View File

@ -115,8 +115,13 @@ async fn connect_handler(
ServerError("Internal Server Error")
})?;
let server_event_stream =
futures::stream::once(async move { Ok::<_, actix_web::Error>(Bytes::from(server_event)) });
let server_event_stream = futures::stream::unfold(rx, |mut rx| async {
let result = rx.recv().await;
match result {
Ok(server_event) => Some((Ok::<_, actix_web::Error>(Bytes::from(server_event)), rx)),
Err(_) => None,
}
});
Ok(HttpResponse::Ok()
.append_header(("content-type", "text/event-stream"))