SSE
提供对 SSE
支持的中间件.
配置 Cargo.toml
salvo = { version = "*", features = ["sse"] }
示例代码
use std::convert::Infallible;
use std::time::Duration;
use futures_util::StreamExt;
use salvo::sse::{self, SseEvent};
use salvo::prelude::*;
use tokio::time::interval;
use tokio_stream::wrappers::IntervalStream;
// create server-sent event
fn sse_counter(counter: u64) -> Result<SseEvent, Infallible> {
Ok(SseEvent::default().data(counter.to_string()))
}
#[handler]
async fn handle_tick(res: &mut Response) {
let event_stream = {
let mut counter: u64 = 0;
let interval = interval(Duration::from_secs(1));
let stream = IntervalStream::new(interval);
stream.map(move |_| {
counter += 1;
sse_counter(counter)
})
};
sse::streaming(res, event_stream).ok();
}
#[tokio::main]
async fn main() {
tracing_subscriber::fmt().init();
let router = Router::with_path("ticks").get(handle_tick);
tracing::info!("Listening on http://127.0.0.1:7878");
Server::new(TcpListener::bind("127.0.0.1:7878")).serve(router).await;
}