feat(viz): export serve_with_upgrades

viz-rs · Nov 20, 2023 · e508200 · e508200
1 parent 5394d3b
commit e508200
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/websocket-chat/src/main.rs b/examples/websocket-chat/src/main.rs
@@ -6,7 +6,7 @@ use std::{net::SocketAddr, sync::Arc};
 use tokio::net::TcpListener;
 use tokio::sync::broadcast::{channel, Sender};
 use viz::{
-    get, serve,
+    get, serve_with_upgrades,
     types::{Message, Params, State, WebSocket},
     HandlerExt, IntoHandler, IntoResponse, Request, RequestExt, Response, ResponseExt, Result,
     Router, Tree,
@@ -67,7 +67,7 @@ async fn main() -> Result<()> {
         let (stream, addr) = listener.accept().await?;
         let tree = tree.clone();
         tokio::task::spawn(async move {
-            if let Err(err) = serve(stream, tree, Some(addr)).await {
+            if let Err(err) = serve_with_upgrades(stream, tree, Some(addr)).await {
                 eprintln!("Error while serving HTTP connection: {err}");
             }
         });

diff --git a/viz/src/lib.rs b/viz/src/lib.rs
@@ -538,7 +538,7 @@ pub use responder::Responder;
 #[cfg(any(feature = "http1", feature = "http2"))]
 mod serve;
 #[cfg(any(feature = "http1", feature = "http2"))]
-pub use serve::serve;
+pub use serve::{serve, serve_with_upgrades};
 
 /// TLS
 pub mod tls;