From fc590c14920ea8e8755eb19ccc4c2ceb756ae1d5 Mon Sep 17 00:00:00 2001
From: naskya <m@naskya.net>
Date: Thu, 6 Jun 2024 17:10:38 +0900
Subject: [PATCH] chore (backend-rs): use ws or wss for WebSocket

---
 packages/backend-rs/src/config/server.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/backend-rs/src/config/server.rs b/packages/backend-rs/src/config/server.rs
index 41ada46a4e..93169fe14c 100644
--- a/packages/backend-rs/src/config/server.rs
+++ b/packages/backend-rs/src/config/server.rs
@@ -278,7 +278,10 @@ pub fn load_config() -> Config {
         None => hostname.clone(),
     };
     let scheme = url.scheme().to_owned();
-    let ws_scheme = scheme.replace("http", "ws");
+    let ws_scheme = match scheme.as_str() {
+        "http" => "ws",
+        _ => "wss",
+    };
 
     let cluster_limits = match server_config.cluster_limits {
         Some(cl) => WorkerConfig {
@@ -344,7 +347,7 @@ pub fn load_config() -> Config {
         hostname,
         redis_key_prefix,
         scheme,
-        ws_scheme,
+        ws_scheme: ws_scheme.to_string(),
     }
 }