ws vs wss

golemcloud · Sep 6, 2023 · bda43bf · bda43bf
1 parent 54125c5
commit bda43bf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/clients/worker.rs b/src/clients/worker.rs
@@ -272,7 +272,10 @@ impl<C: golem_client::instance::Instance + Send + Sync> WorkerClient for WorkerC
         auth: &CloudAuthentication,
     ) -> Result<(), GolemError> {
         let mut url = self.base_url.clone();
-        url.set_scheme("wss")
+
+        let ws_schema = if url.scheme() == "http" { "ws" } else { "wss" };
+
+        url.set_scheme(ws_schema)
             .map_err(|_| GolemError("Can't set schema.".to_string()))?;
         url.path_segments_mut()
             .map_err(|_| GolemError("Can't get path.".to_string()))?