From cb78ad5f3aa9436cf32ae74f07dd1a5d18b8f046 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 08 Jan 2026 13:44:41 +0800
Subject: [PATCH] refactor(ai): 调整公司ID查询逻辑和端口配置

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java |   91 ++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 85 insertions(+), 6 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java
index 2f0697e..54f3c5c 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java
@@ -1,14 +1,19 @@
 package cc.mrbird.febs.ai.service.impl;
 
+import cc.mrbird.febs.ai.entity.AiMemberRole;
 import cc.mrbird.febs.ai.entity.AiProductCategory;
+import cc.mrbird.febs.ai.enumerates.AiCommonEnum;
 import cc.mrbird.febs.ai.enumerates.ProductCategoryLevelEnum;
 import cc.mrbird.febs.ai.mapper.AiProductCategoryMapper;
+import cc.mrbird.febs.ai.req.productCategory.ApiProductCategoryAllDto;
+import cc.mrbird.febs.ai.req.productCategory.ApiProductCategoryHotDto;
 import cc.mrbird.febs.ai.req.productCategory.ApiProductCategoryPageDto;
 import cc.mrbird.febs.ai.res.productCategory.ApiProductCategoryVo;
 import cc.mrbird.febs.ai.service.AiProductCategoryService;
 import cc.mrbird.febs.common.entity.FebsResponse;
 import cc.mrbird.febs.mall.vo.ApiActivityInfoVo;
 import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
@@ -70,17 +75,87 @@
     }
 
     @Override
-    public FebsResponse categoryList(ApiProductCategoryPageDto dto) {
-        // 创建分页对象,传入当前页和每页大小
-        Page<ApiProductCategoryVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
-        Page<ApiProductCategoryVo> pageListByQuery = this.getPageListByQuery(page, dto);
-        return new FebsResponse().success().data(pageListByQuery);
+    public FebsResponse hotV1(ApiProductCategoryHotDto dto) {
+
+        List<ApiProductCategoryVo> list = new ArrayList<>();
+
+        LambdaQueryWrapper<AiProductCategory> query = Wrappers.lambdaQuery(AiProductCategory.class);
+        if (StrUtil.isEmpty(dto.getCompanyId())){
+            dto.setCompanyId(AiCommonEnum.COMPANY_ID.getPrompt());
+        }
+        query.eq(AiProductCategory::getCompanyId, dto.getCompanyId());
+        query.eq(AiProductCategory::getLevel, ProductCategoryLevelEnum.LEVEL_TWO.getLevel());
+        query.eq(AiProductCategory::getState, 1);
+        query.eq(AiProductCategory::getHotState, 1);
+        query.orderByAsc(AiProductCategory::getSort);
+        List<AiProductCategory> listByQuery = this.getListByQuery(query);
+        if (CollUtil.isNotEmpty(listByQuery)){
+            for (AiProductCategory aiProductCategory : listByQuery){
+                ApiProductCategoryVo apiProductCategoryVo = new ApiProductCategoryVo();
+                apiProductCategoryVo.setId(aiProductCategory.getId());
+                apiProductCategoryVo.setName(aiProductCategory.getName());
+                apiProductCategoryVo.setBackImg(aiProductCategory.getBackImg());
+                apiProductCategoryVo.setIconImg(aiProductCategory.getIconImg());
+                list.add(apiProductCategoryVo);
+            }
+        }
+        return new FebsResponse().success().data(list);
+    }
+
+    @Override
+    public FebsResponse hotV2(ApiProductCategoryHotDto dto) {
+
+        List<ApiProductCategoryVo> list = new ArrayList<>();
+
+        LambdaQueryWrapper<AiProductCategory> query = Wrappers.lambdaQuery(AiProductCategory.class);
+        if (StrUtil.isEmpty(dto.getCompanyId())){
+            dto.setCompanyId(AiCommonEnum.COMPANY_ID.getPrompt());
+        }
+        query.eq(AiProductCategory::getCompanyId, dto.getCompanyId());
+        query.eq(AiProductCategory::getLevel, ProductCategoryLevelEnum.LEVEL_ONE.getLevel());
+        query.eq(AiProductCategory::getState, 1);
+        query.eq(AiProductCategory::getHotState, 1);
+        query.orderByAsc(AiProductCategory::getSort);
+        List<AiProductCategory> listByQuery = this.getListByQuery(query);
+        if (CollUtil.isNotEmpty(listByQuery)){
+            for (AiProductCategory aiProductCategory : listByQuery){
+                ApiProductCategoryVo apiProductCategoryVo = new ApiProductCategoryVo();
+                apiProductCategoryVo.setId(aiProductCategory.getId());
+                apiProductCategoryVo.setName(aiProductCategory.getName());
+                apiProductCategoryVo.setBackImg(aiProductCategory.getBackImg());
+                apiProductCategoryVo.setIconImg(aiProductCategory.getIconImg());
+                list.add(apiProductCategoryVo);
+            }
+        }
+        return new FebsResponse().success().data(list);
+    }
+
+    @Override
+    public FebsResponse categoryChildList(ApiProductCategoryPageDto dto) {
+        List<ApiProductCategoryVo> vos = new ArrayList<>();
+
+        List<AiProductCategory> aiProductCategories = aiProductCategoryMapper.selectList(
+                Wrappers.lambdaQuery(AiProductCategory.class)
+                        .eq(AiProductCategory::getParentId, dto.getParentId())
+                        .eq(AiProductCategory::getState, 1)
+                .orderByAsc(AiProductCategory::getSort)
+        );
+        if (CollUtil.isNotEmpty(aiProductCategories)){
+            for (AiProductCategory aiProductCategory : aiProductCategories) {
+                ApiProductCategoryVo vo = new ApiProductCategoryVo();
+                vo.setId(aiProductCategory.getId());
+                vo.setName(aiProductCategory.getName());
+                vos.add(vo);
+            }
+        }
+        return new FebsResponse().success().data(vos);
     }
 
     @Override
     public String getDefaultProductCategoryId() {
         LambdaQueryWrapper<AiProductCategory> query = Wrappers.lambdaQuery(AiProductCategory.class);
         query.eq(AiProductCategory::getState, 1);
+        query.isNull(AiProductCategory::getCompanyId);
         query.eq(AiProductCategory::getHotState, 1);
         query.orderByAsc(AiProductCategory::getSort);
         query.last("limit 1");
@@ -92,9 +167,13 @@
     }
 
     @Override
-    public FebsResponse allList() {
+    public FebsResponse allList(ApiProductCategoryAllDto dto) {
         List<ApiProductCategoryVo> list = new ArrayList<>();
         LambdaQueryWrapper<AiProductCategory> query = Wrappers.lambdaQuery(AiProductCategory.class);
+        if (StrUtil.isEmpty(dto.getCompanyId())){
+            dto.setCompanyId(AiCommonEnum.COMPANY_ID.getPrompt());
+        }
+        query.eq(AiProductCategory::getCompanyId, dto.getCompanyId());
         query.eq(AiProductCategory::getState, 1);
         query.eq(AiProductCategory::getLevel, ProductCategoryLevelEnum.LEVEL_ONE.getLevel());
         query.orderByAsc(AiProductCategory::getSort);

--
Gitblit v1.9.1