From 3b05389a5bcf93ead612761835952699a4b2e302 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Sep 2025 16:50:21 +0800
Subject: [PATCH] refactor(ai): 重构产品配置相关功能

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java |   15 ++++++++++++---
 1 files changed, 12 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..9d0c44e 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
@@ -9,6 +9,7 @@
 import cc.mrbird.febs.common.entity.FebsResponse;
 import cc.mrbird.febs.common.entity.QueryRequest;
 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;
@@ -44,6 +45,9 @@
     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);
         return pages;
     }
@@ -56,6 +60,7 @@
         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());
@@ -74,6 +79,7 @@
                             .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,9 +100,12 @@
     }
 
     @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);
     }
 
 

--
Gitblit v1.9.1