Update src/sse/stream.rs

src/sse/stream.rs

@@ -0,0 +1,18 @@
+use axum::{extract::State, response::sse::{Event, Sse}};
+use futures_util::stream::Stream;
+use std::{convert::Infallible, sync::Arc};
+use tokio_stream::{wrappers::BroadcastStream, StreamExt};
+use crate::AppState;
+
+pub async fn sse_handler(
+    State(state): State<Arc<AppState>>,
+) -> Sse<impl Stream<Item = Result<Event, Infallible>>> {
+    let rx = state.tx.subscribe();
+    let stream = BroadcastStream::new(rx).map(|msg| {
+        match msg {
+            Ok(data) => Ok(Event::default().data(data)),
+            Err(_) => Ok(Event::default().data("error")),
+        }
+    });
+    Sse::new(stream)
+}