refactor: 使用 Token 作为 WebSocket Session Key

This commit is contained in:
2024-06-15 11:40:15 +08:00
parent e30e32aa94
commit e6dbe1a06d
2 changed files with 10 additions and 4 deletions

View File

@@ -16,7 +16,6 @@
package top.continew.admin.common.config.websocket;
import cn.hutool.core.convert.Convert;
import jakarta.servlet.http.HttpServletRequest;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.stereotype.Component;
@@ -40,7 +39,7 @@ public class CurrentUserProviderImpl implements CurrentUserProvider {
String token = servletRequest.getParameter("token");
LoginUser loginUser = LoginHelper.getLoginUser(token);
CurrentUser currentUser = new CurrentUser();
currentUser.setUserId(Convert.toStr(loginUser.getId()));
currentUser.setUserId(loginUser.getToken());
return currentUser;
}
}