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()); + } + +}