diff --git a/src/main/java/com/yupi/springbootinit/service/impl/LoginService.java b/src/main/java/com/yupi/springbootinit/service/impl/LoginService.java index 72b10ac..1d7f878 100644 --- a/src/main/java/com/yupi/springbootinit/service/impl/LoginService.java +++ b/src/main/java/com/yupi/springbootinit/service/impl/LoginService.java @@ -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; } } \ No newline at end of file diff --git a/src/main/java/com/yupi/springbootinit/utils/RedisUtils.java b/src/main/java/com/yupi/springbootinit/utils/RedisUtils.java new file mode 100644 index 0000000..1d9b5bf --- /dev/null +++ b/src/main/java/com/yupi/springbootinit/utils/RedisUtils.java @@ -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 redisTemplate; + + public boolean hasAiLoginKeys(String prefix) { + Set keys = redisTemplate.keys(prefix); // 获取匹配的键集合 + return !keys.isEmpty(); // 如果有键匹配,返回true,否则返回false + } +}