opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-02 21:50:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix some team project bugs

Browse Source
This commit is contained in:
JiaJu Zhuang
2023-09-09 11:33:09 +08:00
parent 0a8ab3f2e8
commit 26ebc23cf6
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/config/Chat2dbWebMvcConfigurer.java
View File

@ -92,6 +92,7 @@ public class Chat2dbWebMvcConfigurer implements WebMvcConfigurer {
return null;
}
if (!ValidStatusEnum.VALID.getCode().equals(user.getStatus())) {
StpUtil.logout();
throw new BusinessException("oauth.invalidUserName");
}
boolean admin = RoleCodeEnum.ADMIN.getCode().equals(user.getRoleCode());

4
chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/controller/oauth/OauthController.java
View File

@ -1,6 +1,7 @@
package ai.chat2db.server.start.controller.oauth;
import ai.chat2db.server.domain.api.enums.RoleCodeEnum;
import ai.chat2db.server.domain.api.enums.ValidStatusEnum;
import jakarta.annotation.Resource;
import ai.chat2db.server.domain.api.model.User;
@ -50,6 +51,9 @@ public class OauthController {
if (user == null) {
throw new BusinessException("oauth.userNameNotExits");
}
if (!ValidStatusEnum.VALID.getCode().equals(user.getStatus())) {
throw new BusinessException("oauth.invalidUserName");
}
if (RoleCodeEnum.DESKTOP.getDefaultUserId().equals(user.getId())) {
throw new BusinessException("oauth.IllegalUserName");
}