feat(volo-http): add sse client support

Author: Joshuahoky

Cargo.lock

@@ -123,6 +123,12 @@ required-features = ["__tls"]
 name = "http-tls-client"
 path = "src/http/http-tls-client.rs"
 required-features = ["__tls"]
+[[bin]]
+name = "http-sse-server"
+path = "src/http/sse/server.rs"
+[[bin]]
+name = "http-sse-client"
+path = "src/http/sse/client.rs"
 
 # shmipc
 [[bin]]
@@ -146,6 +152,7 @@ http-body-util.workspace = true
 hyper.workspace = true
 hyper-util.workspace = true
 metainfo.workspace = true
+mime.workspace = true
 motore.workspace = true
 serde.workspace = true
 tokio = { workspace = true, features = ["full"] }

examples/Cargo.toml

@@ -0,0 +1,32 @@
+use volo_http::{
+    client::{Client, sse::SseExt},
+    error::BoxError,
+};
+
+#[volo::main]
+async fn main() -> Result<(), BoxError> {
+    let client = Client::builder().build()?;
+
+    let mut reader = client
+        .get("http://127.0.0.1:8080/sse")
+        .header(http::header::ACCEPT, mime::TEXT_EVENT_STREAM.essence_str())
+        .send()
+        .await?
+        .into_sse()?;
+
+    while let Some(event) = reader.read().await? {
+        println!("event: {}", event.event());
+        if let Some(data) = event.data() {
+            println!("data: {}", data);
+        }
+        if let Some(id) = event.id() {
+            println!("id: {}", id);
+        }
+        if let Some(retry) = event.retry() {
+            println!("retry: {}", retry.as_millis());
+        }
+        println!();
+    }
+
+    Ok(())
+}

examples/src/http/sse/client.rs

@@ -0,0 +1,34 @@
+use std::{convert::Infallible, net::SocketAddr};
+
+use async_stream::stream;
+use futures::Stream;
+use tokio::time::Duration;
+use volo::net::Address;
+use volo_http::server::{
+    Server,
+    response::sse::{Event, Sse},
+    route::{Router, get},
+};
+
+async fn sse_handler() -> Sse<impl Stream<Item = Result<Event, Infallible>>> {
+    let stream = stream! {
+        loop {
+            yield Ok(Event::new().event("ping").data("hello"));
+            tokio::time::sleep(Duration::from_secs(1)).await;
+        }
+    };
+
+    Sse::new(stream)
+}
+
+#[volo::main]
+async fn main() {
+    let app = Router::new().route("/sse", get(sse_handler));
+
+    let addr: SocketAddr = "127.0.0.1:8080".parse().unwrap();
+    let addr = Address::from(addr);
+
+    println!("Server running on {}", addr);
+
+    Server::new(app).run(addr).await.unwrap();
+}

examples/src/http/sse/server.rs

@@ -59,6 +59,7 @@ pub mod dns;
 pub mod layer;
 pub mod loadbalance;
 mod request_builder;
+pub mod sse;
 pub mod target;
 #[cfg(test)]
 pub mod test_helpers;