From 05105809d740885fd952cb052fabad84b886b5df Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Sat, 11 Jan 2025 16:59:41 +0800 Subject: [PATCH] =?UTF-8?q?bug=2025=20=E5=BA=94=E7=94=A8=E4=B8=AD=E5=BF=83?= =?UTF-8?q?=E9=A1=B5=E9=9D=A2=EF=BC=8C=E5=81=9C=E7=94=A8=E7=9A=84=E6=A0=87?= =?UTF-8?q?=E7=AD=BE=E4=BE=9D=E7=84=B6=E8=83=BD=E5=B1=95=E7=A4=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../admin/label/LabelController.java | 4 +-- .../llm/service/label/LabelService.java | 33 ++++++++++++------- .../llm/service/label/LabelServiceImpl.java | 21 +++++++++++- 3 files changed, 44 insertions(+), 14 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/label/LabelController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/label/LabelController.java index 3e3af3887..8b9132b6b 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/label/LabelController.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/label/LabelController.java @@ -83,7 +83,7 @@ public class LabelController { @Operation(summary = "获得标签管理列表所有") // @PreAuthorize("@ss.hasPermission('llm:label:query')") public CommonResult> getLabelList() { - List list = labelService.getLabelList(); + List list = labelService.getEnableLabelList(); return success(BeanUtils.toBean(list, LabelRespVO.class)); } @@ -101,4 +101,4 @@ public class LabelController { BeanUtils.toBean(list, LabelRespVO.class)); } -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/label/LabelService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/label/LabelService.java index f599dae6d..5275c6bd4 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/label/LabelService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/label/LabelService.java @@ -1,10 +1,10 @@ package cn.iocoder.yudao.module.llm.service.label; -import cn.iocoder.yudao.module.llm.controller.admin.label.vo.*; -import cn.iocoder.yudao.module.llm.dal.dataobject.label.LabelDO; import cn.iocoder.yudao.framework.common.pojo.PageResult; -import cn.iocoder.yudao.framework.common.pojo.PageParam; +import cn.iocoder.yudao.module.llm.controller.admin.label.vo.LabelPageReqVO; +import cn.iocoder.yudao.module.llm.controller.admin.label.vo.LabelSaveReqVO; +import cn.iocoder.yudao.module.llm.dal.dataobject.label.LabelDO; import javax.validation.Valid; import java.util.List; @@ -22,21 +22,21 @@ public interface LabelService { * @param createReqVO 创建信息 * @return 编号 */ - Long createLabel(@Valid LabelSaveReqVO createReqVO); + Long createLabel (@Valid LabelSaveReqVO createReqVO); /** * 更新标签管理 * * @param updateReqVO 更新信息 */ - void updateLabel(@Valid LabelSaveReqVO updateReqVO); + void updateLabel (@Valid LabelSaveReqVO updateReqVO); /** * 删除标签管理 * * @param id 编号 */ - void deleteLabel(Long id); + void deleteLabel (Long id); /** * 获得标签管理 @@ -44,7 +44,7 @@ public interface LabelService { * @param id 编号 * @return 标签管理 */ - LabelDO getLabel(Long id); + LabelDO getLabel (Long id); /** * 获得标签管理分页 @@ -52,8 +52,19 @@ public interface LabelService { * @param pageReqVO 分页查询 * @return 标签管理分页 */ - PageResult getLabelPage(LabelPageReqVO pageReqVO); - // 获取所有标签 - List getLabelList(); + PageResult getLabelPage (LabelPageReqVO pageReqVO); -} \ No newline at end of file + /** + * 获取所有标签 + * + * @return 标签列表 + */ + List getLabelList (); + + /** + * 获取所有已开启的标签 + * + * @return 标签列表 + */ + List getEnableLabelList (); +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/label/LabelServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/label/LabelServiceImpl.java index 825629fe6..72105a90d 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/label/LabelServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/label/LabelServiceImpl.java @@ -1,10 +1,13 @@ package cn.iocoder.yudao.module.llm.service.label; +import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; import org.springframework.stereotype.Service; import org.springframework.validation.annotation.Validated; import org.springframework.transaction.annotation.Transactional; import java.util.*; +import java.util.stream.Collectors; + import cn.iocoder.yudao.module.llm.controller.admin.label.vo.*; import cn.iocoder.yudao.module.llm.dal.dataobject.label.LabelDO; import cn.iocoder.yudao.framework.common.pojo.PageResult; @@ -77,4 +80,20 @@ public class LabelServiceImpl implements LabelService { return labelMapper.selectList(); } -} \ No newline at end of file + /** + * 获取所有已开启的标签 + * + * @return 标签列表 + */ + @Override + public List getEnableLabelList () { + List labels = labelMapper.selectList(); + if (CollectionUtils.isEmpty(labels)){ + return Collections.emptyList(); + } + + // status 0: 启用，1: 停用 + return labels.stream().filter(labelDO -> labelDO.getStatus() == 0).collect(Collectors.toList()); + } + +}