From e4048873355aa30e2f4296d745972158225fddd6 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Sep 2025 15:57:06 +0800
Subject: [PATCH] feat(ai): 增加产品角色列表和树状图的公司条件过滤

---
 src/main/java/cc/mrbird/febs/ai/controller/product/ViewController.java |   49 ++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 48 insertions(+), 1 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 6a43ac0..9b71c7c 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
@@ -37,6 +37,8 @@
     private final AiProductPointLinkService aiProductPointLinkService;
     private final AiProductPointService aiProductPointService;
     private final AiProductRoleLinkService aiProductRoleLinkService;
+    private final AiProductQuestionLinkService aiProductQuestionLinkService;
+    private final AiProductQuestionService aiProductQuestionService;
     private final AiProductRoleService aiProductRoleService;
 
     @GetMapping("list")
@@ -110,11 +112,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)){
@@ -123,7 +127,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 -> {
@@ -140,4 +144,47 @@
         model.addAttribute("chooseId", id);
         return FebsUtil.view("modules/ai/product/productRoleSet");
     }
+
+
+
+    @GetMapping("productQuestionSet/{id}")
+    @RequiresPermissions("productList:productQuestionSet")
+    public String productQuestionSet(@PathVariable String id, Model model) {
+        List<AdminMoveChooseInfoVo> vos = new ArrayList<>();
+        Set<String> productIds = new HashSet<>();
+
+        AiProduct entity = aiProductService.getById(id);
+        if(ObjectUtil.isNotNull(entity)){
+            //右侧数据
+            LambdaQueryWrapper<AiProductQuestionLink> query = Wrappers.lambdaQuery(AiProductQuestionLink.class);
+            if(StrUtil.isNotEmpty(id)){
+                query.eq(AiProductQuestionLink::getProductId, id);
+            }
+            List<AiProductQuestionLink> selectedList = aiProductQuestionLinkService.selectListByQuery(query);
+            if(CollUtil.isNotEmpty(selectedList)){
+                //stream流操作happyMemberLabelRecords,获取memberId的set集合
+                productIds = selectedList.stream().map(AiProductQuestionLink::getProductQuestionId).collect(Collectors.toSet());
+            }
+
+            //左侧数据
+            LambdaQueryWrapper<AiProductQuestion> aiProductQuestionLambdaQueryWrapper = Wrappers.lambdaQuery(AiProductQuestion.class);
+            aiProductQuestionLambdaQueryWrapper.eq(AiProductQuestion::getProductCategoryId, entity.getProductCategoryId());
+            aiProductQuestionLambdaQueryWrapper.eq(AiProductQuestion::getState, 1);
+            List<AiProductQuestion> allList = aiProductQuestionService.productQuestionTree(aiProductQuestionLambdaQueryWrapper);
+            if(CollUtil.isNotEmpty(allList)){
+                //stream流操作mallMembers,生成一个新的List<MallMemberVo>
+                vos = allList.stream().map(AiProductQuestion -> {
+                    AdminMoveChooseInfoVo vo = new AdminMoveChooseInfoVo();
+                    vo.setId(AiProductQuestion.getId());
+                    vo.setName(AiProductQuestion.getTitle());
+                    return vo;
+                }).collect(Collectors.toList());
+            }
+        }
+
+        model.addAttribute("productQuestionAll", vos);
+        model.addAttribute("productQuestionSelected", productIds);
+        model.addAttribute("chooseId", id);
+        return FebsUtil.view("modules/ai/product/productQuestionSet");
+    }
 }

--
Gitblit v1.9.1