From 490bceb5d9e911a5c4f687dd6570b6699ca11915 Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Tue, 16 Sep 2025 17:27:11 +0800 Subject: [PATCH] feat(ai): 添加会员角色知识点配置功能 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java | 46 +++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 43 insertions(+), 3 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java index 8bead4e..4bde82d 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java @@ -1,14 +1,18 @@ package cc.mrbird.febs.ai.service.impl; import cc.mrbird.febs.ai.entity.AiProduct; +import cc.mrbird.febs.ai.entity.AiProductCategory; import cc.mrbird.febs.ai.entity.AiProductPoint; import cc.mrbird.febs.ai.mapper.AiProductPointMapper; +import cc.mrbird.febs.ai.service.AiProductCategoryService; import cc.mrbird.febs.ai.service.AiProductPointLinkService; import cc.mrbird.febs.ai.service.AiProductPointService; 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; @@ -20,6 +24,9 @@ import java.util.Date; import java.util.List; +import java.util.Map; +import java.util.Set; +import java.util.stream.Collectors; /** * AI产品知识点 Service实现类 @@ -34,6 +41,7 @@ private final AiProductPointMapper aiProductPointMapper; private final AiProductPointLinkService aiProductPointLinkService; + private final AiProductCategoryService aiProductCategoryService; @Override public AiProductPoint getById(String id) { @@ -44,7 +52,24 @@ public IPage<AiProductPoint> listInPage(AiProductPoint dto, QueryRequest request) { Page<AiProductPoint> page = new Page<>(request.getPageNum(), request.getPageSize()); LambdaQueryWrapper<AiProductPoint> query = Wrappers.lambdaQuery(AiProductPoint.class); + if (StrUtil.isNotEmpty(dto.getCompanyId())){ + query.eq(AiProductPoint::getCompanyId, dto.getCompanyId()); + } Page<AiProductPoint> pages = aiProductPointMapper.selectPage(page, query); + List<AiProductPoint> records = pages.getRecords(); + if (CollUtil.isNotEmpty( records)){ + Set<String> collect = records.stream().map(AiProductPoint::getProductCategoryId).collect(Collectors.toSet()); + + if(CollUtil.isNotEmpty( collect)){ + Map<String, AiProductCategory> map = aiProductCategoryService.selectMapByIds(collect); + for (AiProductPoint record : records){ + AiProductCategory orDefault = map.getOrDefault(record.getProductCategoryId(), null); + if(ObjectUtil.isNotNull(orDefault)){ + record.setProductCategoryName(orDefault.getName()); + } + } + } + } return pages; } @@ -52,10 +77,12 @@ public FebsResponse add(AiProductPoint dto) { AiProductPoint entity = new AiProductPoint(); entity.setId(UUID.getSimpleUUIDString()); + entity.setProductCategoryId(dto.getProductCategoryId()); entity.setCompanyId(dto.getCompanyId()); entity.setIsNormal(dto.getIsNormal() ); entity.setFinderUserName(dto.getFinderUserName()); entity.setFeedId(dto.getFeedId()); + entity.setFeedImg(dto.getFeedImg()); entity.setTitle(dto.getTitle()); entity.setDescription(dto.getDescription()); entity.setCreatedTime(new Date()); @@ -71,9 +98,11 @@ this.update(null, Wrappers.lambdaUpdate(AiProductPoint.class) .set(AiProductPoint::getIsNormal, dto.getIsNormal()) + .set(AiProductPoint::getProductCategoryId, dto.getProductCategoryId()) .set(AiProductPoint::getFinderUserName, dto.getFinderUserName()) .set(AiProductPoint::getFeedId, dto.getFeedId()) .set(AiProductPoint::getTitle, dto.getTitle()) + .set(AiProductPoint::getFeedImg, dto.getFeedImg()) .set(AiProductPoint::getDescription, dto.getDescription()) .set(AiProductPoint::getUpdatedTime, new Date()) .eq(AiProductPoint::getId, id) @@ -94,11 +123,22 @@ } @Override - public List<AiProductPoint> pointTree() { - - return aiProductPointMapper.selectList(null); + public List<AiProductPoint> pointTree(String companyId) { + LambdaQueryWrapper<AiProductPoint> query = Wrappers.lambdaQuery(AiProductPoint.class); + if (StrUtil.isNotEmpty(companyId)){ + query.eq(AiProductPoint::getCompanyId, companyId); + } + return aiProductPointMapper.selectList(query); } + @Override + public List<AiProductPoint> selectList(String companyId) { + LambdaQueryWrapper<AiProductPoint> query = Wrappers.lambdaQuery(AiProductPoint.class); + if (StrUtil.isNotEmpty(companyId)){ + query.eq(AiProductPoint::getCompanyId, companyId); + } + return aiProductPointMapper.selectList(query); + } } -- Gitblit v1.9.1