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