Add support for wss

Eluminatis-of-LU · Dec 7, 2023 · 5b49bf0 · 5b49bf0
1 parent 3e7649c
commit 5b49bf0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sojj/Services/JudgeService.cs b/sojj/Services/JudgeService.cs
@@ -41,7 +41,8 @@ public JudgeService(ILogger<JudgeService> logger, IConfiguration configuration)
 
         httpClient = new HttpClient(pollyHandler);
         httpClient.BaseAddress = baseUrl;
-        wsUrl = new Uri("ws://" + httpClient.BaseAddress.Host + ":" + httpClient.BaseAddress.Port + "/judge/consume-conn/websocket");
+        var wsScheme = httpClient.BaseAddress.Scheme.Equals("https", StringComparison.OrdinalIgnoreCase) ? "wss://" : "ws://";
+        wsUrl = new Uri(wsScheme + httpClient.BaseAddress.Host + ":" + httpClient.BaseAddress.Port + "/judge/consume-conn/websocket");
         ws = new ClientWebSocket();
         ws.Options.Cookies = socketHandler.CookieContainer;
     }