test/MeterSphere
1
0
Fork 0
mirror of https://gitee.com/fit2cloud-feizhiyun/MeterSphere.git synced 2024-12-05 05:29:29 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

refactor(gateway): 重写is-login方法，不用响应式，防止阻塞redisson线程

Browse Source
This commit is contained in:
liqiang-fit2cloud 2023-04-28 20:49:44 +08:00
parent 166cde0b11
commit 45c733c30e
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
framework/gateway/src/main/java/io/metersphere/gateway/controller/LoginController.java
View File

@ -5,6 +5,7 @@ import io.metersphere.commons.constants.OperLogConstants;
import io.metersphere.commons.constants.OperLogModule;
import io.metersphere.commons.constants.SessionConstants;
import io.metersphere.commons.user.SessionUser;
import io.metersphere.commons.utils.LogUtil;
import io.metersphere.commons.utils.RsaUtil;
import io.metersphere.controller.handler.ResultHolder;
import io.metersphere.dto.ServiceDTO;
@ -56,15 +57,21 @@ public class LoginController {
if (StringUtils.isNotBlank(sessionId) && StringUtils.isNotBlank(csrfToken)) {
userLoginService.validateCsrfToken(sessionId, csrfToken);
Object userFromSession = redisSessionRepository.getSessionRedisOperations().opsForHash().get("spring:session:sessions:" + sessionId, "sessionAttr:user");
if (userFromSession instanceof User) {
// 用户只有工作空间权限
if (StringUtils.isBlank(((User) userFromSession).getLastProjectId())) {
((User) userFromSession).setLastProjectId("no_such_project");
if (userFromSession != null) {
if (userFromSession instanceof User) {
// 用户只有工作空间权限
if (StringUtils.isBlank(((User) userFromSession).getLastProjectId())) {
((User) userFromSession).setLastProjectId("no_such_project");
}
// 使用数据库里的最新用户权限，不同的tab sessionId 不变
UserDTO userDTO = userLoginService.getUserDTO(((User) userFromSession).getId());
SessionUser sessionUser = SessionUser.fromUser(userDTO, sessionId);
return Mono.just(ResultHolder.success(sessionUser));
}
// 使用数据库里的最新用户权限，不同的tab sessionId 不变
UserDTO userDTO = userLoginService.getUserDTO(((User) userFromSession).getId());
SessionUser sessionUser = SessionUser.fromUser(userDTO, sessionId);
return Mono.just(ResultHolder.success(sessionUser));
LogUtil.info("userFromSession.class: " + userFromSession.getClass().getName());
return Mono.just(ResultHolder.success(userFromSession));
} else {
LogUtil.info("userFromSession is null");
}
}
return Mono.just(ResultHolder.error(RsaUtil.getRsaKey().getPublicKey()));