From fbb09a170c4a87e6fd81e99288c39128fde56564 Mon Sep 17 00:00:00 2001
From: fengshuonan <sn93@qq.com>
Date: Wed, 29 Mar 2023 11:50:25 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=907.3.4=E3=80=91=E4=BF=AE=E5=A4=8D?=
 =?UTF-8?q?=E4=BC=9A=E8=AF=9D=E5=88=B7=E6=96=B0=E7=9A=84wsurl=E6=9C=AA?=
 =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E7=9A=84bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../roses/kernel/auth/auth/AuthServiceImpl.java          | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/kernel-d-auth/auth-sdk/src/main/java/cn/stylefeng/roses/kernel/auth/auth/AuthServiceImpl.java b/kernel-d-auth/auth-sdk/src/main/java/cn/stylefeng/roses/kernel/auth/auth/AuthServiceImpl.java
index 71afd5e75..037143541 100644
--- a/kernel-d-auth/auth-sdk/src/main/java/cn/stylefeng/roses/kernel/auth/auth/AuthServiceImpl.java
+++ b/kernel-d-auth/auth-sdk/src/main/java/cn/stylefeng/roses/kernel/auth/auth/AuthServiceImpl.java
@@ -85,6 +85,8 @@ import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
 import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
 
 import static cn.stylefeng.roses.kernel.auth.api.exception.enums.AuthExceptionEnum.AUTH_EXPIRED_ERROR;
 import static cn.stylefeng.roses.kernel.auth.api.exception.enums.AuthExceptionEnum.TOKEN_PARSE_ERROR;
@@ -293,6 +295,13 @@ public class AuthServiceImpl implements AuthServiceApi {
             LoginUser loginUser = userLoginInfo.getLoginUser();
             loginUser.setTenantCode(tenantCode);
 
+            // 设置websocket url
+            String webSocketWsUrl = WebSocketConfigExpander.getWebSocketWsUrl();
+            Map<String, String> params = new HashMap<>(1);
+            params.put("token", token);
+            webSocketWsUrl = StrUtil.format(webSocketWsUrl, params);
+            loginUser.setWsUrl(webSocketWsUrl);
+
             // 创建用户会话信息
             sessionManagerApi.updateSession(token, loginUser);
         } finally {