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/productRole/AiProductRoleController.java |    9 ++++++---
 1 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/controller/productRole/AiProductRoleController.java b/src/main/java/cc/mrbird/febs/ai/controller/productRole/AiProductRoleController.java
index c249859..ee99336 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/productRole/AiProductRoleController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/productRole/AiProductRoleController.java
@@ -29,7 +29,8 @@
 
     @GetMapping("list")
     public FebsResponse list(AiProductRole dto, QueryRequest request) {
-
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         Map<String, Object> data = getDataTable(aiProductRoleService.listInPage(dto, request));
         return new FebsResponse().success().data(data);
     }
@@ -38,6 +39,8 @@
     @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
     public FebsResponse add(@RequestBody @Valid AiProductRole dto) {
 
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         return aiProductRoleService.add(dto);
     }
 
@@ -59,7 +62,7 @@
 
     @GetMapping(value = "/productRoleTree")
     public FebsResponse productRoleTree() {
-
-        return new FebsResponse().success().data(aiProductRoleService.productRoleTree());
+        String companyId = getCurrentUserCompanyId();
+        return new FebsResponse().success().data(aiProductRoleService.productRoleTree(companyId));
     }
 }

--
Gitblit v1.9.1