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 | 189 +++++++++++++++++++++------------------------- 1 files changed, 86 insertions(+), 103 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 0aae32b..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,130 +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 this.getById(id); + return aiProductPointMapper.selectById(id); } @Override - public List<AiProductPoint> getByCompanyId(String companyId) { - LambdaQueryWrapper<AiProductPoint> queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(AiProductPoint::getCompanyId, companyId); - queryWrapper.orderByDesc(AiProductPoint::getCreatedTime); - return this.list(queryWrapper); - } - - @Override - public List<AiProductPoint> getByIsNormal(Integer isNormal) { - LambdaQueryWrapper<AiProductPoint> queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(AiProductPoint::getIsNormal, isNormal); - queryWrapper.orderByDesc(AiProductPoint::getCreatedTime); - return this.list(queryWrapper); - } - - @Override - public List<AiProductPoint> getByFinderUserName(String finderUserName) { - LambdaQueryWrapper<AiProductPoint> queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(AiProductPoint::getFinderUserName, finderUserName); - queryWrapper.orderByDesc(AiProductPoint::getCreatedTime); - return this.list(queryWrapper); - } - - @Override - public List<AiProductPoint> getByTitle(String title) { - LambdaQueryWrapper<AiProductPoint> queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.like(AiProductPoint::getTitle, title); - queryWrapper.orderByDesc(AiProductPoint::getCreatedTime); - return this.list(queryWrapper); - } - - @Override - public List<AiProductPoint> getByCompanyIdAndIsNormal(String companyId, Integer isNormal) { - LambdaQueryWrapper<AiProductPoint> queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(AiProductPoint::getCompanyId, companyId); - queryWrapper.eq(AiProductPoint::getIsNormal, isNormal); - queryWrapper.orderByDesc(AiProductPoint::getCreatedTime); - return this.list(queryWrapper); - } - - @Override - public List<AiProductPoint> searchByKeyword(String keyword) { - LambdaQueryWrapper<AiProductPoint> queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.and(wrapper -> wrapper.like(AiProductPoint::getTitle, keyword) - .or() - .like(AiProductPoint::getDescription, keyword)); - queryWrapper.orderByDesc(AiProductPoint::getCreatedTime); - return this.list(queryWrapper); - } - - @Override - @Transactional(rollbackFor = Exception.class) - public boolean saveProductPoint(AiProductPoint aiProductPoint) { - try { - return this.save(aiProductPoint); - } catch (Exception e) { - log.error("保存AI产品知识点失败: ", e); - return false; + 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 - @Transactional(rollbackFor = Exception.class) - public boolean saveBatchProductPoints(List<AiProductPoint> productPoints) { - try { - return this.saveBatch(productPoints); - } catch (Exception e) { - log.error("批量保存AI产品知识点失败: ", e); - return false; - } + 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 - @Transactional(rollbackFor = Exception.class) - public boolean updateProductPoint(AiProductPoint aiProductPoint) { - try { - return this.updateById(aiProductPoint); - } catch (Exception e) { - log.error("更新AI产品知识点失败: ", e); - return false; + 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 - @Transactional(rollbackFor = Exception.class) - public boolean deleteById(String id) { - try { - return this.removeById(id); - } catch (Exception e) { - log.error("删除AI产品知识点失败: ", e); - return false; + 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 - @Transactional(rollbackFor = Exception.class) - public boolean deleteByCompanyId(String companyId) { - try { - LambdaQueryWrapper<AiProductPoint> queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(AiProductPoint::getCompanyId, companyId); - return this.remove(queryWrapper); - } catch (Exception e) { - log.error("根据公司ID删除AI产品知识点失败: ", e); - return false; + 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 - @Transactional(rollbackFor = Exception.class) - public boolean deleteByFeedId(String feedId) { - try { - LambdaQueryWrapper<AiProductPoint> queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(AiProductPoint::getFeedId, feedId); - return this.remove(queryWrapper); - } catch (Exception e) { - log.error("根据视频号FeedID删除AI产品知识点失败: ", e); - return false; - } - } + + } -- Gitblit v1.9.1