From 577bef7f8ece263c4bc272bab23fcc8aa4b6083b Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Tue, 16 Sep 2025 11:56:01 +0800 Subject: [PATCH] feat(ai): 产品知识点增加分类功能 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java | 104 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 103 insertions(+), 1 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 f76b1d3..c496f82 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,16 +1,32 @@ 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; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Service; -import org.springframework.transaction.annotation.Transactional; +import java.util.Date; import java.util.List; +import java.util.Map; +import java.util.Set; +import java.util.stream.Collectors; /** * AI产品知识点 Service实现类 @@ -24,11 +40,97 @@ public class AiProductPointServiceImpl extends ServiceImpl<AiProductPointMapper, AiProductPoint> implements AiProductPointService { private final AiProductPointMapper aiProductPointMapper; + private final AiProductPointLinkService aiProductPointLinkService; + private final AiProductCategoryService aiProductCategoryService; @Override public AiProductPoint getById(String id) { return aiProductPointMapper.selectById(id); } + @Override + 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; + } + + @Override + 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()); + this.save(entity); + return new FebsResponse().success().message("操作成功"); + } + + @Override + public FebsResponse update(AiProductPoint dto) { + String id = dto.getId(); + AiProductPoint entity = this.getById(id); + if (ObjectUtil.isNotNull( entity)){ + 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) + ); + } + return new FebsResponse().success().message("操作成功"); + } + + @Override + public FebsResponse delete(String id) { + AiProductPoint entity = this.getById(id); + if(ObjectUtil.isNotNull( entity)){ + // 删除 + aiProductPointMapper.deleteById( id); + aiProductPointLinkService.deleteByProductPointId(id); + } + return new FebsResponse().success().message("操作成功"); + } + + @Override + 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); + } + + } -- Gitblit v1.9.1