From 8298b88330abc838dd6484c96317de68b95aeb6a Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 16 Sep 2025 16:49:44 +0800
Subject: [PATCH] feat(ai): 新增 AI 用户选择角色关联产品功能

---
 src/main/java/cc/mrbird/febs/ai/controller/product/ViewController.java |   11 +++++++++--
 1 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/controller/product/ViewController.java b/src/main/java/cc/mrbird/febs/ai/controller/product/ViewController.java
index 7f874d9..1514d86 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/product/ViewController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/product/ViewController.java
@@ -72,11 +72,13 @@
         Set<String> productIds = new HashSet<>();
 
         AiProduct entity = aiProductService.getById(id);
+        String companyId = entity.getCompanyId();
         if(ObjectUtil.isNotNull(entity)){
             //右侧数据
             LambdaQueryWrapper<AiProductPointLink> query = Wrappers.lambdaQuery(AiProductPointLink.class);
             if(StrUtil.isNotEmpty(id)){
                 query.eq(AiProductPointLink::getProductId, id);
+                query.eq(AiProductPointLink::getCompanyId, companyId);
             }
             List<AiProductPointLink> selectedList = aiProductPointLinkService.selectListByQuery(query);
             if(CollUtil.isNotEmpty(selectedList)){
@@ -85,7 +87,7 @@
             }
 
             //左侧数据
-            List<AiProductPoint> allList = aiProductPointService.pointTree();
+            List<AiProductPoint> allList = aiProductPointService.pointTree(companyId);
             if(CollUtil.isNotEmpty(allList)){
                 //stream流操作mallMembers,生成一个新的List<MallMemberVo>
                 vos = allList.stream().map(AiProductPoint -> {
@@ -112,11 +114,13 @@
         Set<String> productIds = new HashSet<>();
 
         AiProduct entity = aiProductService.getById(id);
+        String companyId = entity.getCompanyId();
         if(ObjectUtil.isNotNull(entity)){
             //右侧数据
             LambdaQueryWrapper<AiProductRoleLink> query = Wrappers.lambdaQuery(AiProductRoleLink.class);
             if(StrUtil.isNotEmpty(id)){
                 query.eq(AiProductRoleLink::getProductId, id);
+                query.eq(AiProductRoleLink::getCompanyId, companyId);
             }
             List<AiProductRoleLink> selectedList = aiProductRoleLinkService.selectListByQuery(query);
             if(CollUtil.isNotEmpty(selectedList)){
@@ -125,7 +129,7 @@
             }
 
             //左侧数据
-            List<AiProductRole> allList = aiProductRoleService.productRoleTree();
+            List<AiProductRole> allList = aiProductRoleService.productRoleTree(companyId);
             if(CollUtil.isNotEmpty(allList)){
                 //stream流操作mallMembers,生成一个新的List<MallMemberVo>
                 vos = allList.stream().map(AiProductRole -> {
@@ -152,11 +156,13 @@
         Set<String> productIds = new HashSet<>();
 
         AiProduct entity = aiProductService.getById(id);
+        String companyId = entity.getCompanyId();
         if(ObjectUtil.isNotNull(entity)){
             //右侧数据
             LambdaQueryWrapper<AiProductQuestionLink> query = Wrappers.lambdaQuery(AiProductQuestionLink.class);
             if(StrUtil.isNotEmpty(id)){
                 query.eq(AiProductQuestionLink::getProductId, id);
+                query.eq(AiProductQuestionLink::getCompanyId, companyId);
             }
             List<AiProductQuestionLink> selectedList = aiProductQuestionLinkService.selectListByQuery(query);
             if(CollUtil.isNotEmpty(selectedList)){
@@ -168,6 +174,7 @@
             LambdaQueryWrapper<AiProductQuestion> aiProductQuestionLambdaQueryWrapper = Wrappers.lambdaQuery(AiProductQuestion.class);
             aiProductQuestionLambdaQueryWrapper.eq(AiProductQuestion::getProductCategoryId, entity.getProductCategoryId());
             aiProductQuestionLambdaQueryWrapper.eq(AiProductQuestion::getState, 1);
+            aiProductQuestionLambdaQueryWrapper.eq(AiProductQuestion::getCompanyId, companyId);
             List<AiProductQuestion> allList = aiProductQuestionService.productQuestionTree(aiProductQuestionLambdaQueryWrapper);
             if(CollUtil.isNotEmpty(allList)){
                 //stream流操作mallMembers,生成一个新的List<MallMemberVo>

--
Gitblit v1.9.1