From db091bdf9f2e12a3962246aeeb5d20366639a21e Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 18 Mar 2026 13:42:27 +0800
Subject: [PATCH] refactor(yinhe): 替换登录用户工具类
---
src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java | 32 ++++++++------------------------
1 files changed, 8 insertions(+), 24 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java b/src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java
index 1c55aef..d7c702f 100644
--- a/src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java
@@ -1,12 +1,12 @@
package cc.mrbird.febs.yinhe.service.impl;
+import cc.mrbird.febs.ai.res.product.ApiProductVo;
import cc.mrbird.febs.ai.strategy.LlmStrategyFactory;
import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyEnum;
import cc.mrbird.febs.ai.utils.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.AppContants;
-import cc.mrbird.febs.common.utils.LoginUserUtil;
import cc.mrbird.febs.common.utils.RedisUtils;
import cc.mrbird.febs.common.utils.YHLoginUserUtil;
import cc.mrbird.febs.rabbit.producter.AgentProducer;
@@ -22,6 +22,7 @@
import com.alibaba.dashscope.common.Role;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
@@ -196,27 +197,10 @@
@Override
public FebsResponse agentSysList(YHSysAgentDto dto) {
- ArrayList<YHSysAgentVo> vos = new ArrayList<>();
- LambdaQueryWrapper<YHSysAgent> query = Wrappers.lambdaQuery(YHSysAgent.class);
- query.eq(YHSysAgent::getState,1);
- if (ObjectUtil.isNotEmpty(dto.getAgentCategoryId())){
- query.eq(YHSysAgent::getAgentCategoryId,dto.getAgentCategoryId());
- }
- query.orderByDesc(YHSysAgent::getCreateTime);
-
- List<YHSysAgent> list = yhSysAgentMapper.selectList(query);
- if (CollUtil.isNotEmpty(list)){
-
- for (YHSysAgent entity : list) {
- YHSysAgentVo vo = new YHSysAgentVo();
- vo.setId(entity.getId());
- vo.setName(entity.getName());
- vo.setDescription(entity.getDescription());
- vo.setIconImg(entity.getIconImg());
- vos.add(vo);
- }
- }
- return new FebsResponse().success().data(vos);
+ // 创建分页对象，传入当前页和每页大小
+ Page<YHSysAgentVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ Page<YHSysAgentVo> pageListByQuery = yhSysAgentMapper.getPageListByQuery(page, dto);
+ return new FebsResponse().success().data(pageListByQuery);
}
@Override
@@ -248,7 +232,7 @@
@Override
public FebsResponse logout() {
- String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ String memberUuid = YHLoginUserUtil.getLoginUser().getMemberUuid();
String redisKey = AppContants.XCX_LOGIN_PREFIX + memberUuid;
String existToken = redisUtils.getString(redisKey);
@@ -264,7 +248,7 @@
@Override
public FebsResponse initSend(YHSendInitDto dto) {
- String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ String memberUuid = YHLoginUserUtil.getLoginUser().getMemberUuid();
YHSendInitVo vo = new YHSendInitVo();
--
Gitblit v1.9.1