gitee
/
roses
mirror of https://gitee.com/stylefeng/roses
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

修复初始化配置以后登录用户需重新登录才可以使用Websocket的问题

pull/22/head
rays 2021-10-20 11:30:19 +08:00
parent d69565eb5c
commit c9477f5f93
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
kernel-d-auth/auth-sdk/src/main/java/cn/stylefeng/roses/kernel/auth/config/InitConfigWebSocketCallbackApiImpl.java
View File

@ -1,8 +1,11 @@
package cn.stylefeng.roses.kernel.auth.config;
import cn.stylefeng.roses.kernel.auth.api.SessionManagerApi;
import cn.stylefeng.roses.kernel.auth.api.context.LoginContext;
import cn.stylefeng.roses.kernel.auth.api.pojo.login.LoginUser;
import cn.stylefeng.roses.kernel.config.api.ConfigInitCallbackApi;
import cn.stylefeng.roses.kernel.message.api.expander.WebSocketConfigExpander;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
/**
@ -14,6 +17,9 @@ import org.springframework.stereotype.Component;
@Component
public class InitConfigWebSocketCallbackApiImpl implements ConfigInitCallbackApi {
@Autowired
private SessionManagerApi sessionManagerApi;
@Override
public void initBefore() {
@ -21,6 +27,8 @@ public class InitConfigWebSocketCallbackApiImpl implements ConfigInitCallbackApi
@Override
public void initAfter() {
LoginContext.me().getLoginUser().setWsUrl(WebSocketConfigExpander.getWebSocketWsUrl());
LoginUser loginUser = LoginContext.me().getLoginUser();
loginUser.setWsUrl(WebSocketConfigExpander.getWebSocketWsUrl());
sessionManagerApi.updateSession(loginUser.getToken(), loginUser);
}
}