FIX ws upgrade

AnubisLMS · May 6, 2024 · 9896108 · 9896108
1 parent 81a6a02
commit 9896108
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/theia/proxy-rs/src/main.rs b/theia/proxy-rs/src/main.rs
@@ -212,7 +212,7 @@ async fn handle(
 
     let host = format!("ws://{}:{}", cluster_address, port);
 
-    ws.on_upgrade(move |socket| ws::forward(&host, ws));
+    ws.on_upgrade(move |socket| ws::forward(host, socket));
     (StatusCode::OK, "authorized".to_string())
 }
 

diff --git a/theia/proxy-rs/src/ws.rs b/theia/proxy-rs/src/ws.rs
@@ -1,6 +1,4 @@
-use anyhow::Result;
 use axum::extract::ws::{CloseFrame, Message as AxumMessage, WebSocket};
-use axum::extract::WebSocketUpgrade;
 use futures_util::{SinkExt, StreamExt};
 use tokio_tungstenite::tungstenite;
 use tokio_tungstenite::{connect_async, tungstenite::protocol::Message as TsMessage};
@@ -68,7 +66,7 @@ impl WebSocketMessage {
     }
 }
 
-pub async fn forward(url: &str, client_ws: WebSocketUpgrade) {
+pub async fn forward(url: String, client_ws: WebSocket) {
     let server_ws = match connect_async(url).await {
         Ok((ws, _)) => ws,
         Err(e) => {