From 33fb9a2fdb0e6cd68a987bbdcbe22ef587fa5813 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 17 Sep 2025 16:30:32 +0800
Subject: [PATCH] refactor(mallMember): 修改会员列表中管理员字段的标题

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java |   16 +++++++++++++---
 1 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java
index d125cf8..06a3247 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java
@@ -11,6 +11,7 @@
 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 cn.hutool.json.JSONArray;
 import cn.hutool.json.JSONObject;
 import cn.hutool.json.JSONUtil;
@@ -55,6 +56,9 @@
     public IPage<AiProductQuestion> listInPage(AiProductQuestion dto, QueryRequest request) {
         Page<AiProductQuestion> page = new Page<>(request.getPageNum(), request.getPageSize());
         LambdaQueryWrapper<AiProductQuestion> query = Wrappers.lambdaQuery(AiProductQuestion.class);
+        if (StrUtil.isNotEmpty(dto.getCompanyId())){
+            query.eq(AiProductQuestion::getCompanyId, dto.getCompanyId());
+        }
         query.ne(AiProductQuestion::getState, 2);
         Page<AiProductQuestion> pages = aiProductQuestionMapper.selectPage(page, query);
         return pages;
@@ -99,6 +103,7 @@
         for (AiProductQuestionItem item : dto.getAiProductQuestionItems()){
             item.setId(UUID.getSimpleUUIDString());
             item.setProductQuestionId(entity.getId());
+            item.setCompanyId(dto.getCompanyId());
             item.setTitle(entity.getTitle());
             item.setCreatedTime(new Date());
             aiProductQuestionItemService.getBaseMapper().insert(item);
@@ -125,7 +130,6 @@
         if (ObjectUtil.isNotNull( entity)){
             this.update(null,
                     Wrappers.lambdaUpdate(AiProductQuestion.class)
-                            .set(AiProductQuestion::getCompanyId, dto.getCompanyId())
                             .set(AiProductQuestion::getProductCategoryId, dto.getProductCategoryId())
                             .set(AiProductQuestion::getTitle, dto.getTitle())
                             .set(AiProductQuestion::getDifficulty, dto.getDifficulty())
@@ -161,9 +165,13 @@
     }
 
     @Override
-    public List<AiProductQuestion> questionTree() {
+    public List<AiProductQuestion> questionTree(String companyId) {
+        LambdaQueryWrapper<AiProductQuestion> query = Wrappers.lambdaQuery(AiProductQuestion.class);
+        if (StrUtil.isNotEmpty(companyId)){
+            query.eq(AiProductQuestion::getCompanyId, companyId);
+        }
 
-        return aiProductQuestionMapper.selectList(null);
+        return aiProductQuestionMapper.selectList(query);
     }
 
     @Override
@@ -190,6 +198,7 @@
 
             AiProductQuestion aiProductQuestion = new AiProductQuestion();
             aiProductQuestion.setId(UUID.getSimpleUUIDString());
+            aiProductQuestion.setCompanyId(dto.getCompanyId());
             aiProductQuestion.setProductCategoryId(productCategoryId);
             aiProductQuestion.setTitle(title);
             aiProductQuestion.setDifficulty(difficulty);
@@ -205,6 +214,7 @@
                 System.out.println("分析: " + answer.getStr("analysis"));
                 AiProductQuestionItem aiProductQuestionItem = new AiProductQuestionItem();
                 aiProductQuestionItem.setId(UUID.getSimpleUUIDString());
+                aiProductQuestionItem.setCompanyId(dto.getCompanyId());
                 aiProductQuestionItem.setProductQuestionId(aiProductQuestion.getId());
                 aiProductQuestionItem.setTitle(aiProductQuestion.getTitle());
                 aiProductQuestionItem.setAnswer(answer.getStr("answer"));

--
Gitblit v1.9.1