From 4bfe57af40d98331721124837950c4329ad12baf Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Sep 2025 15:07:03 +0800
Subject: [PATCH] feat(ai): 增加公司 ID 筛选功能

---
 src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java |   10 +++++++---
 1 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java b/src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java
index e03c29a..24373ad 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java
@@ -31,7 +31,8 @@
 
     @GetMapping("list")
     public FebsResponse list(AiMemberRole dto, QueryRequest request) {
-
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         Map<String, Object> data = getDataTable(aiMemberRoleService.listInPage(dto, request));
         return new FebsResponse().success().data(data);
     }
@@ -49,12 +50,13 @@
     @PostMapping("add")
     @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
     public FebsResponse add(@RequestBody @Valid AiMemberRole dto) {
-
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         return aiMemberRoleService.memberRoleAdd(dto);
     }
 
     @PostMapping("update")
-    @ControllerEndpoint(operation = "分类-更新", exceptionMessage = "操作失败")
+    @ControllerEndpoint(operation = "更新", exceptionMessage = "操作失败")
     public FebsResponse update(@RequestBody @Valid AiMemberRole dto) {
 
         return aiMemberRoleService.updateMemberRole(dto);
@@ -73,6 +75,8 @@
     @ControllerEndpoint(operation = "产品配置", exceptionMessage = "操作失败")
     public FebsResponse productSet(@RequestBody @Valid AdminMoveChooseInfoDto dto) {
 
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         return aiMemberRoleService.productSet(dto);
     }
 }

--
Gitblit v1.9.1