1.修改AI退出登录后的逻辑

This commit is contained in:
2025-08-27 21:12:50 +08:00
parent 07a4142818
commit 43cbd262ea
2 changed files with 38 additions and 4 deletions

View File

@@ -10,6 +10,7 @@ import com.yupi.springbootinit.model.enums.CommonStatusEnum;
import com.yupi.springbootinit.model.enums.LoginSceneEnum;
import com.yupi.springbootinit.model.vo.user.SystemUsersVO;
import com.yupi.springbootinit.service.SystemUsersService;
import com.yupi.springbootinit.utils.RedisUtils;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.*;
@@ -37,11 +38,14 @@ public class LoginService {
@Resource
private RabbitAdmin rabbitAdmin;
@Resource
private RedisUtils redisUtils;
public SystemUsersVO login(LoginSceneEnum scene, SystemUsersDTO dto) {
SystemUsers user = validateUser(dto); // 校验用户名、密码、状态、租户过期
checkRole(scene, user.getId()); // 按场景做角色校验
if (scene.equals(LoginSceneEnum.AI_CHAT)) {
redisTemplate.opsForValue().set("ai_login:"+user.getTenantId() ,true);
redisTemplate.opsForValue().set("ai_login:"+user.getTenantId()+":"+user.getId(),true);
String queueName = "q.tenant." + user.getTenantId();
if (created.add(String.valueOf(user.getTenantId()))) {
Queue queue = QueueBuilder.durable(queueName).build();
@@ -89,10 +93,14 @@ public class LoginService {
}
public Boolean aiChatLogout(SystemUsersDTO usersDTO) {
Boolean delete = redisTemplate.delete("ai_login:"+usersDTO.getTenantId());
created.remove(String.valueOf(usersDTO.getTenantId()));
log.info("删除租户:{}登录状态:{}",usersDTO.getTenantId(),delete);
Boolean delete = redisTemplate.delete("ai_login:"+usersDTO.getTenantId()+":"+usersDTO.getUserId());
StpUtil.logout(usersDTO.getUserId());
log.info("删除租户:{}登录状态:{}",usersDTO.getTenantId(),delete);
if (!redisUtils.hasAiLoginKeys("ai_login:" + usersDTO.getTenantId()+":*")) {
created.remove(String.valueOf(usersDTO.getTenantId()));
boolean b = rabbitAdmin.deleteQueue("q.tenant." + usersDTO.getTenantId());
log.info("删除租户:{}队列删除状态:{}",usersDTO.getTenantId(),b);
}
return true;
}
}

View File

@@ -0,0 +1,26 @@
package com.yupi.springbootinit.utils;
/*
* @author: ziin
* @date: 2025/8/27 20:35
*/
import org.springframework.data.redis.core.*;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.util.Set;
@Component
public class RedisUtils {
@Resource
private RedisTemplate<String,Object> redisTemplate;
public boolean hasAiLoginKeys(String prefix) {
Set<String> keys = redisTemplate.keys(prefix); // 获取匹配的键集合
return !keys.isEmpty(); // 如果有键匹配返回true否则返回false
}
}