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 params = new HashMap<>(1); + params.put("token", token); + webSocketWsUrl = StrUtil.format(webSocketWsUrl, params); + loginUser.setWsUrl(webSocketWsUrl); + // 创建用户会话信息 sessionManagerApi.updateSession(token, loginUser); } finally {