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