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