From 33fb9a2fdb0e6cd68a987bbdcbe22ef587fa5813 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 17 Sep 2025 16:30:32 +0800
Subject: [PATCH] refactor(mallMember): 修改会员列表中管理员字段的标题

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java |   14 +++++++++++---
 1 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
index 0f361cb..4d2a6bc 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
@@ -8,6 +8,7 @@
 import cc.mrbird.febs.common.entity.FebsResponse;
 import cc.mrbird.febs.common.entity.QueryRequest;
 import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
@@ -44,6 +45,9 @@
 
         Page<AiProductRole> page = new Page<>(request.getPageNum(), request.getPageSize());
         LambdaQueryWrapper<AiProductRole> query = Wrappers.lambdaQuery(AiProductRole.class);
+        if (StrUtil.isNotEmpty(dto.getCompanyId())){
+            query.eq(AiProductRole::getCompanyId, dto.getCompanyId());
+        }
         Page<AiProductRole> pages = aiProductRoleMapper.selectPage(page, query);
         return pages;
     }
@@ -73,7 +77,6 @@
         if (ObjectUtil.isNotNull( entity)){
             this.update(null,
                     Wrappers.lambdaUpdate(AiProductRole.class)
-                            .set(AiProductRole::getCompanyId, dto.getCompanyId())
                             .set(AiProductRole::getName, dto.getName())
                             .set(AiProductRole::getDescription, dto.getDescription())
                             .set(AiProductRole::getPromptHead, dto.getPromptHead())
@@ -101,8 +104,13 @@
     }
 
     @Override
-    public List<AiProductRole> productRoleTree() {
-        return aiProductRoleMapper.selectList(null);
+    public List<AiProductRole> productRoleTree(String companyId) {
+
+        LambdaQueryWrapper<AiProductRole> query = Wrappers.lambdaQuery(AiProductRole.class);
+        if (StrUtil.isNotEmpty(companyId)){
+            query.eq(AiProductRole::getCompanyId, companyId);
+        }
+        return aiProductRoleMapper.selectList(query);
     }
 
 }

--
Gitblit v1.9.1