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

---
 src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java |    8 ++++++--
 1 files changed, 6 insertions(+), 2 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 4300e07..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,7 +50,8 @@
     @PostMapping("add")
     @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
     public FebsResponse add(@RequestBody @Valid AiMemberRole dto) {
-
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         return aiMemberRoleService.memberRoleAdd(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