feat(viz): add serve_with_upgrades

viz/src/serve.rs

@@ -14,6 +14,25 @@ use crate::Responder;
 ///
 /// Will return `Err` if the connection does not be served.
 pub async fn serve<I>(stream: I, tree: Arc<Tree>, addr: Option<SocketAddr>) -> Result<()>
+where
+    I: AsyncRead + AsyncWrite + Unpin + Send + 'static,
+{
+    Builder::new(TokioExecutor::new())
+        .serve_connection(Io::new(stream), Responder::new(tree, addr))
+        .await
+        .map_err(Into::into)
+}
+
+/// Serve the connections with upgrades.
+///
+/// # Errors
+///
+/// Will return `Err` if the connection does not be served.
+pub async fn serve_with_upgrades<I>(
+    stream: I,
+    tree: Arc<Tree>,
+    addr: Option<SocketAddr>,
+) -> Result<()>
 where
     I: AsyncRead + AsyncWrite + Unpin + Send + 'static,
 {