Updates for latest version of tungstenite

entropyxyz · Dec 19, 2024 · 7857741 · 7857741
1 parent d43059d
commit 7857741
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/crates/protocol/src/protocol_transport/mod.rs b/crates/protocol/src/protocol_transport/mod.rs
@@ -106,14 +106,14 @@ impl WsConnection for tokio_tungstenite::WebSocketStream<MaybeTlsStream<tokio::n
             .ok_or(WsError::ConnectionClosed)?
             .map_err(|e| WsError::ConnectionError(e.to_string()))?
         {
-            Ok(msg)
+            Ok(msg.to_vec())
         } else {
             Err(WsError::UnexpectedMessageType)
         }
     }
 
     async fn send(&mut self, msg: Vec<u8>) -> Result<(), WsError> {
-        SinkExt::send(&mut self, tungstenite::Message::Binary(msg))
+        SinkExt::send(&mut self, tungstenite::Message::Binary(msg.into()))
             .await
             .map_err(|_| WsError::ConnectionClosed)
     }
@@ -130,14 +130,14 @@ impl WsConnection for tokio_tungstenite::WebSocketStream<tokio::net::TcpStream>
             .ok_or(WsError::ConnectionClosed)?
             .map_err(|e| WsError::ConnectionError(e.to_string()))?
         {
-            Ok(msg)
+            Ok(msg.to_vec())
         } else {
             Err(WsError::UnexpectedMessageType)
         }
     }
 
     async fn send(&mut self, msg: Vec<u8>) -> Result<(), WsError> {
-        SinkExt::send(&mut self, tungstenite::Message::Binary(msg))
+        SinkExt::send(&mut self, tungstenite::Message::Binary(msg.into()))
             .await
             .map_err(|_| WsError::ConnectionClosed)
     }