From c7cee264bccb5026fd42a9f0dc83d274aaee40cf Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 30 Oct 2025 10:05:28 +0800
Subject: [PATCH] feat(ai): 新增AI陪练流式对话V2接口

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java |   62 ++++++++++++++++++++++++++++++
 1 files changed, 61 insertions(+), 1 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
index 566e82f..63f5657 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java
@@ -1,20 +1,32 @@
 package cc.mrbird.febs.ai.service.impl;
 
+import cc.mrbird.febs.ai.entity.AiMemberProductUnlock;
 import cc.mrbird.febs.ai.entity.AiProduct;
+import cc.mrbird.febs.ai.entity.AiProductDependency;
 import cc.mrbird.febs.ai.entity.AiProductPoint;
 import cc.mrbird.febs.ai.mapper.AiProductMapper;
 import cc.mrbird.febs.ai.req.product.ApiProductInfoDto;
 import cc.mrbird.febs.ai.req.product.ApiProductPageDto;
 import cc.mrbird.febs.ai.res.product.ApiProductInfoVo;
 import cc.mrbird.febs.ai.res.product.ApiProductVo;
+import cc.mrbird.febs.ai.res.productPoint.ApiProductPointVo;
 import cc.mrbird.febs.ai.service.*;
 import cc.mrbird.febs.common.entity.FebsResponse;
+import cc.mrbird.febs.common.exception.FebsException;
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
+
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
 
 /**
  * AI产品表 Service实现类
@@ -31,6 +43,8 @@
     private final AiMemberRoleService aiMemberRoleService;
     private final AiProductCategoryService aiProductCategoryService;
     private final AiProductPointService aiProductPointService;
+    private final AiProductDependencyService aiProductDependencyService;
+    private final AiMemberProductUnlockService aiMemberProductUnlockService;
 
 
     @Override
@@ -58,15 +72,48 @@
         // 创建分页对象，传入当前页和每页大小
         Page<ApiProductVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
         Page<ApiProductVo> pageListByQuery = this.getPageListByQuery(page, dto);
+        List<ApiProductVo> records = pageListByQuery.getRecords();
+        if (CollUtil.isNotEmpty( records)){
+
+            //stream流操作records，返回一个id的集合
+            Set<String> productIds = records.stream().map(ApiProductVo::getId).collect(Collectors.toSet());
+
+            List<AiProductDependency> aiProductDependencies = aiProductDependencyService.selectListByProductIds(productIds);
+            if (CollUtil.isNotEmpty( aiProductDependencies)){
+                //stream流操作aiProductDependencies，返回一个Map<targetProductId,AiProductDependency>
+                Map<String, AiProductDependency> targetProductIdAiProductDependencyMap = aiProductDependencies.stream().collect(Collectors.toMap(AiProductDependency::getTargetProductId, aiProductDependency -> aiProductDependency));
+                for (ApiProductVo record : records){
+                    if (targetProductIdAiProductDependencyMap.containsKey(record.getId())){
+                        record.setState(0);
+                    }
+                }
+            }
+
+            if (StrUtil.isNotEmpty(dto.getMemberId())){
+                List<AiMemberProductUnlock> aiMemberProductUnlocks = aiMemberProductUnlockService.selectListByMemberId(dto.getMemberId());
+                if (CollUtil.isNotEmpty( aiMemberProductUnlocks)){
+                    //stream流操作aiMemberProductUnlocks，返回一个Map<targetProductId,AiMemberProductUnlock>
+                    Map<String, AiMemberProductUnlock> aiMemberProductUnlocksMap = aiMemberProductUnlocks.stream().collect(Collectors.toMap(AiMemberProductUnlock::getProductId, aiMemberProductUnlock -> aiMemberProductUnlock));
+                    for (ApiProductVo record : records){
+                        if (aiMemberProductUnlocksMap.containsKey(record.getId())){
+                            record.setState(1);
+                        }
+                    }
+                }
+            }
+        }
         return new FebsResponse().success().data(pageListByQuery);
     }
 
     @Override
     public FebsResponse productInfo(ApiProductInfoDto dto) {
 
+        ApiProductInfoVo vo = new ApiProductInfoVo();
         String id = dto.getId();
         AiProduct entity = this.getById(id);
-        ApiProductInfoVo vo = new ApiProductInfoVo();
+        if (ObjectUtil.isNull( entity)){
+            throw new FebsException("产品不存在");
+        }
         vo.setId(entity.getId());
         vo.setProductCategoryId(entity.getProductCategoryId());
         vo.setName(entity.getName());
@@ -80,4 +127,17 @@
         vo.setProductPointList(aiProductPointService.listByProductId(id));
         return new FebsResponse().success().data(vo);
     }
+
+    @Override
+    public FebsResponse pointInfoList(ApiProductInfoDto dto) {
+        String id = dto.getId();
+        AiProduct entity = this.getById(id);
+        List<ApiProductPointVo> apiProductPointVos = aiProductPointService.listByProductId(id);
+        return new FebsResponse().success().data(apiProductPointVos);
+    }
+
+    @Override
+    public List<AiProduct> getProductListByQuery(LambdaQueryWrapper<AiProduct> productQuery) {
+        return aiProductMapper.selectList(productQuery);
+    }
 }

--
Gitblit v1.9.1