diff --git a/kernel-d-auth/auth-sdk/src/main/java/cn/stylefeng/roses/kernel/auth/config/InitConfigWebSocketCallbackApiImpl.java b/kernel-d-auth/auth-sdk/src/main/java/cn/stylefeng/roses/kernel/auth/config/InitConfigWebSocketCallbackApiImpl.java new file mode 100644 index 000000000..f830b4331 --- /dev/null +++ b/kernel-d-auth/auth-sdk/src/main/java/cn/stylefeng/roses/kernel/auth/config/InitConfigWebSocketCallbackApiImpl.java @@ -0,0 +1,26 @@ +package cn.stylefeng.roses.kernel.auth.config; + +import cn.stylefeng.roses.kernel.auth.api.context.LoginContext; +import cn.stylefeng.roses.kernel.config.api.ConfigInitCallbackApi; +import cn.stylefeng.roses.kernel.message.api.expander.WebSocketConfigExpander; +import org.springframework.stereotype.Component; + +/** + * 项目初始化完成以后,修改用户websocket地址的配置 + * + * @author majianguo + * @date 2021/10/19 17:07 + */ +@Component +public class InitConfigWebSocketCallbackApiImpl implements ConfigInitCallbackApi { + + @Override + public void initBefore() { + + } + + @Override + public void initAfter() { + LoginContext.me().getLoginUser().setWsUrl(WebSocketConfigExpander.getWebSocketWsUrl()); + } +}