Update src/sse/stream.rs

src/sse/stream.rs

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