From 49400da0804fd0ab56056fbd0a2c9ee36cd091b6 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Sep 2025 16:22:55 +0800
Subject: [PATCH] feat(ai): 增加公司 ID 筛选功能
---
src/main/java/cc/mrbird/febs/ai/controller/product/ViewController.java | 8 ++++++--
1 files changed, 6 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..fea3805 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 -> {
--
Gitblit v1.9.1