ziin/tkcrawl-client
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1.添加AI聊天工具登录接口

Browse Source 
2.删除 xmlsql 无用判断
3.修改 dev配置文件开启 swagger
This commit is contained in:
Ziin
2025-07-03 15:25:33 +08:00
parent 55e9f3d538
commit 4f7e0e4576
8 changed files with 69 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/yupi/springbootinit/config/SaTokenConfigure.java
View File

@@ -45,7 +45,8 @@ public class SaTokenConfigure implements WebMvcConfigurer {
// 你的其他放行路径，例如登录接口
"/user/doLogin",
"/tenant/get-id-by-name",
"/user/bigbrother-doLogin"
"/user/bigbrother-doLogin",
"/user/aiChat-doLogin"
};
}

32
src/main/java/com/yupi/springbootinit/controller/UserController.java
View File

@@ -88,6 +88,38 @@ public class UserController {
BeanUtil.copyProperties(user, systemUsersVO);
// 赋予用户 Id
StpUtil.login(user.getId(),"bigbrother");
// 设置 token 有效期为当前日期和套餐有效期的差值
StpUtil.renewTimeout(second);
systemUsersVO.setTokenName(StpUtil.getTokenName());
systemUsersVO.setTokenValue(StpUtil.getTokenValue());
return ResultUtils.success(systemUsersVO);
}
// 用户登陆接口
@PostMapping("aiChat-doLogin")
public BaseResponse<SystemUsersVO> aiChatDoLogin(@RequestBody SystemUsersDTO usersDTO) {
SystemUsers user = usersService.getUserByUserName(usersDTO.getUsername(),usersDTO.getTenantId());
if (user == null) {
throw new BusinessException(ErrorCode.USERNAME_OR_PASSWORD_ERROR);
}
if (!usersService.isPasswordMatch(usersDTO.getPassword(), user.getPassword())) {
throw new BusinessException(ErrorCode.USERNAME_OR_PASSWORD_ERROR);
}
if (CommonStatusEnum.isDisable(Integer.valueOf(user.getStatus()))) {
throw new BusinessException(ErrorCode.USER_DISABLE);
}
if (usersService.isExpired(usersDTO.getTenantId())){
throw new BusinessException(ErrorCode.PACKAGE_EXPIRED);
}
if (!usersService.checkAiCHatLoginRole(user.getId())){
throw new BusinessException(ErrorCode.LOGIN_NOW_ALLOWED);
}
Long second = usersService.getTenantExpiredTime(usersDTO.getTenantId());
SystemUsersVO systemUsersVO = new SystemUsersVO();
BeanUtil.copyProperties(user, systemUsersVO);
// 赋予用户 Id
StpUtil.login(user.getId(),"aiChat");
// 设置 token 有效期为当前日期和套餐有效期的差值
StpUtil.renewTimeout(second);
systemUsersVO.setTokenName(StpUtil.getTokenName());

7
src/main/java/com/yupi/springbootinit/model/entity/SystemUsers.java
View File

@@ -174,4 +174,11 @@ public class SystemUsers {
@TableField(value = "big_brother")
@ApiModelProperty(value = "能否登录大哥爬虫客户端")
private Byte bigBrother;
/**
* 能否登录大哥爬虫客户端
*/
@TableField(value = "ai_chat")
@ApiModelProperty(value = "能否登录大哥爬虫客户端")
private Byte aiChat;
}

1
src/main/java/com/yupi/springbootinit/service/SystemUsersService.java
View File

@@ -23,4 +23,5 @@ public interface SystemUsersService extends IService<SystemUsers> {
boolean checkbigBrotherlRole(Long userId);
boolean checkAiCHatLoginRole(Long userId);
}

6
src/main/java/com/yupi/springbootinit/service/impl/SystemUsersServiceImpl.java
View File

@@ -76,5 +76,11 @@ public class SystemUsersServiceImpl extends ServiceImpl<SystemUsersMapper,System
return systemUsers.getBigBrother() == 1;
}
@Override
public boolean checkAiCHatLoginRole(Long userId) {
SystemUsers systemUsers = baseMapper.selectById(userId);
return systemUsers.getAiChat() == 1;
}
}