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 | 4 +++- 1 files changed, 3 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 7f874d9..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 @@ -112,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)){ @@ -125,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 -> { -- Gitblit v1.9.1