box enum variant

rkrishn7 · Apr 2, 2024 · 266cc9e · 266cc9e
1 parent 3a4cd94
commit 266cc9e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/kiwi/src/tls.rs b/src/kiwi/src/tls.rs
@@ -34,7 +34,7 @@ pub fn tls_acceptor(cert: impl AsRef<Path>, key: impl AsRef<Path>) -> anyhow::Re
 
 pub enum MaybeTlsStream<S> {
     Plain(S),
-    Tls(TlsStream<S>),
+    Tls(Box<TlsStream<S>>),
 }
 
 impl<S: AsyncRead + AsyncWrite + Unpin> AsyncRead for MaybeTlsStream<S> {

diff --git a/src/kiwi/src/ws.rs b/src/kiwi/src/ws.rs
@@ -57,7 +57,9 @@ where
         tokio::spawn(async move {
             let io = if let Some(acceptor) = acceptor {
                 match acceptor.accept(stream).await {
-                    Ok(stream) => hyper_util::rt::TokioIo::new(MaybeTlsStream::Tls(stream)),
+                    Ok(stream) => {
+                        hyper_util::rt::TokioIo::new(MaybeTlsStream::Tls(Box::new(stream)))
+                    }
                     Err(e) => {
                         tracing::error!(addr = ?addr, "Failed to accept TLS connection: {}", e);
                         return;