From 2fb72d56082e2ee19aa187707751dd0dacd34f4b Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 16 Sep 2025 11:29:00 +0800
Subject: [PATCH] feat(ai): 增加公司列表接口并优化相关服务

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java |   41 ++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 40 insertions(+), 1 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..771f4ef 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,18 @@
 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.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,6 +74,35 @@
     }
 
     @Override
+    public FebsResponse hotV1(ApiProductCategoryHotDto dto) {
+
+        List<ApiProductCategoryVo> list = new ArrayList<>();
+
+        LambdaQueryWrapper<AiProductCategory> query = Wrappers.lambdaQuery(AiProductCategory.class);
+        if (StrUtil.isNotEmpty(dto.getCompanyId())){
+            query.eq(AiProductCategory::getCompanyId, dto.getCompanyId());
+        }else{
+            query.isNull(AiProductCategory::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 categoryList(ApiProductCategoryPageDto dto) {
         // 创建分页对象,传入当前页和每页大小
         Page<ApiProductCategoryVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
@@ -81,6 +114,7 @@
     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 +126,14 @@
     }
 
     @Override
-    public FebsResponse allList() {
+    public FebsResponse allList(ApiProductCategoryAllDto dto) {
         List<ApiProductCategoryVo> list = new ArrayList<>();
         LambdaQueryWrapper<AiProductCategory> query = Wrappers.lambdaQuery(AiProductCategory.class);
+        if (StrUtil.isNotEmpty(dto.getCompanyId())){
+            query.eq(AiProductCategory::getCompanyId, dto.getCompanyId());
+        }else{
+            query.isNull(AiProductCategory::getCompanyId);
+        }
         query.eq(AiProductCategory::getState, 1);
         query.eq(AiProductCategory::getLevel, ProductCategoryLevelEnum.LEVEL_ONE.getLevel());
         query.orderByAsc(AiProductCategory::getSort);

--
Gitblit v1.9.1