From 8298b88330abc838dd6484c96317de68b95aeb6a Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 16 Sep 2025 16:49:44 +0800
Subject: [PATCH] feat(ai): 新增 AI 用户选择角色关联产品功能
---
src/main/java/cc/mrbird/febs/ai/controller/productCategory/AiProductCategoryController.java | 11 ++++++++---
1 files changed, 8 insertions(+), 3 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/productCategory/AiProductCategoryController.java b/src/main/java/cc/mrbird/febs/ai/controller/productCategory/AiProductCategoryController.java
index 8063a2a..4cb3c8b 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/productCategory/AiProductCategoryController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/productCategory/AiProductCategoryController.java
@@ -31,6 +31,8 @@
@GetMapping("list")
public FebsResponse list(AiProductCategory dto, QueryRequest request) {
+ String companyId = getCurrentUserCompanyId();
+ dto.setCompanyId(companyId);
Map<String, Object> data = getDataTable(service.listInPage(dto, request));
return new FebsResponse().success().data(data);
@@ -50,7 +52,8 @@
@PostMapping("add")
@ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
public FebsResponse add(@RequestBody @Valid AiProductCategory dto) {
-
+ String companyId = getCurrentUserCompanyId();
+ dto.setCompanyId(companyId);
return service.add(dto);
}
@@ -73,13 +76,15 @@
@GetMapping(value = "/categoryTree")
public FebsResponse categoryTree() {
- return new FebsResponse().success().data(service.categoryTree());
+ String companyId = getCurrentUserCompanyId();
+ return new FebsResponse().success().data(service.categoryTree(companyId));
}
@GetMapping("categoryTree/parent")
@ControllerEndpoint(exceptionMessage = "获取分类失败")
public List<AiProductCategory> parent(){
- return service.parent();
+ String companyId = getCurrentUserCompanyId();
+ return service.parent(companyId);
}
}
--
Gitblit v1.9.1