From 49400da0804fd0ab56056fbd0a2c9ee36cd091b6 Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Mon, 15 Sep 2025 16:22:55 +0800 Subject: [PATCH] feat(ai): 增加公司 ID 筛选功能 --- src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java | 9 +++++++-- 1 files changed, 7 insertions(+), 2 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java b/src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java index 0c53cf9..12a8535 100644 --- a/src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java +++ b/src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java @@ -29,7 +29,8 @@ @GetMapping("list") public FebsResponse list(AiProductPoint dto, QueryRequest request) { - + String companyId = getCurrentUserCompanyId(); + dto.setCompanyId(companyId); Map<String, Object> data = getDataTable(service.listInPage(dto, request)); return new FebsResponse().success().data(data); } @@ -38,6 +39,8 @@ @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败") public FebsResponse add(@RequestBody @Valid AiProductPoint dto) { + String companyId = getCurrentUserCompanyId(); + dto.setCompanyId(companyId); return service.add(dto); } @@ -60,6 +63,8 @@ @GetMapping(value = "/pointTree") public FebsResponse pointTree() { - return new FebsResponse().success().data(service.pointTree()); + + String companyId = getCurrentUserCompanyId(); + return new FebsResponse().success().data(service.pointTree(companyId)); } } -- Gitblit v1.9.1