From 5b9d8648f3a0da63bb26fdb17c39d8999c9491c4 Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Mon, 15 Sep 2025 14:44:50 +0800 Subject: [PATCH] feat(ai): 增加公司ID筛选功能 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java | 37 +++++++++++++++++++++++++++++++------ 1 files changed, 31 insertions(+), 6 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java index 6feb107..51e1572 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java @@ -3,15 +3,13 @@ import cc.mrbird.febs.ai.entity.*; import cc.mrbird.febs.ai.mapper.AiProductMapper; import cc.mrbird.febs.ai.req.AdminMoveChooseInfoDto; -import cc.mrbird.febs.ai.service.AiProductCategoryService; -import cc.mrbird.febs.ai.service.AiProductPointLinkService; -import cc.mrbird.febs.ai.service.AiProductRoleLinkService; -import cc.mrbird.febs.ai.service.AiProductService; +import cc.mrbird.febs.ai.service.*; import cc.mrbird.febs.ai.util.UUID; import cc.mrbird.febs.common.entity.FebsResponse; import cc.mrbird.febs.common.entity.QueryRequest; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.util.ObjectUtil; +import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.core.toolkit.Wrappers; @@ -42,6 +40,7 @@ private final AiProductCategoryService aiProductCategoryService; private final AiProductPointLinkService aiProductPointLinkService; private final AiProductRoleLinkService aiProductRoleLinkService; + private final AiProductQuestionLinkService aiProductQuestionLinkService; @Override public AiProduct getById(String id) { @@ -53,7 +52,7 @@ Page<AiProduct> page = new Page<>(request.getPageNum(), request.getPageSize()); LambdaQueryWrapper<AiProduct> query = Wrappers.lambdaQuery(AiProduct.class); - if (ObjectUtil.isNotNull(dto.getProductCategoryId())){ + if (StrUtil.isNotEmpty(dto.getProductCategoryId())){ query.eq(AiProduct::getProductCategoryId, dto.getProductCategoryId()); } query.ne(AiProduct::getState, 2); @@ -159,9 +158,10 @@ } @Override - public List<AiProduct> selectList() { + public List<AiProduct> selectList(String companyId) { return aiProductMapper.selectList( Wrappers.lambdaQuery(AiProduct.class) + .eq(AiProduct::getCompanyId, companyId) .ne(AiProduct::getState, 2) ); } @@ -217,4 +217,29 @@ } return new FebsResponse().success().message("操作成功"); } + + @Override + public FebsResponse productQuestionSet(AdminMoveChooseInfoDto dto) { + String chooseId = dto.getChooseId(); + List<String> chooseIds = dto.getChooseIds(); + AiProduct aiProduct = this.getById(chooseId); + if (ObjectUtil.isNotNull(aiProduct)) { + aiProductQuestionLinkService.deleteByQuery( + Wrappers.lambdaQuery(AiProductQuestionLink.class) + .eq(AiProductQuestionLink::getProductId,chooseId) + ); + if(CollUtil.isNotEmpty(chooseIds)){ + Date createdTime = new Date(); + for (String item : chooseIds){ + AiProductQuestionLink entity = new AiProductQuestionLink(); + entity.setId(UUID.getSimpleUUIDString()); + entity.setProductId(chooseId); + entity.setProductQuestionId(item); + entity.setCreatedTime(createdTime); + aiProductQuestionLinkService.getBaseMapper().insert(entity); + } + } + } + return new FebsResponse().success().message("操作成功"); + } } -- Gitblit v1.9.1