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 | 210 ++++++++++++++++++++++++----------------------------
1 files changed, 96 insertions(+), 114 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 17566dc..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,16 +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.service.AiProductService;
+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 org.springframework.transaction.annotation.Transactional;
import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
/**
* AI产品表 Service实现类
@@ -24,138 +40,104 @@
public class AiProductServiceImpl extends ServiceImpl<AiProductMapper, AiProduct> implements AiProductService {
private final AiProductMapper aiProductMapper;
+ private final AiMemberRoleService aiMemberRoleService;
+ private final AiProductCategoryService aiProductCategoryService;
+ private final AiProductPointService aiProductPointService;
+ private final AiProductDependencyService aiProductDependencyService;
+ private final AiMemberProductUnlockService aiMemberProductUnlockService;
+
@Override
public AiProduct getById(String id) {
- return this.getById(id);
+ return aiProductMapper.selectById( id);
}
@Override
- public List<AiProduct> getByCompanyId(String companyId) {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiProduct::getCompanyId, companyId);
- queryWrapper.orderByDesc(AiProduct::getSort);
- queryWrapper.orderByDesc(AiProduct::getCreatedTime);
- return this.list(queryWrapper);
+ public Page<ApiProductVo> getPageListByQuery(Page<ApiProductVo> page, ApiProductPageDto dto) {
+ return aiProductMapper.selectPageListByQuery(page, dto);
}
@Override
- public AiProduct getByCode(String code) {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiProduct::getCode, code);
- return this.getOne(queryWrapper);
- }
+ public FebsResponse productList(ApiProductPageDto dto) {
- @Override
- public List<AiProduct> getByName(String name) {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.like(AiProduct::getName, name);
- queryWrapper.orderByDesc(AiProduct::getSort);
- queryWrapper.orderByDesc(AiProduct::getCreatedTime);
- return this.list(queryWrapper);
- }
-
- @Override
- public List<AiProduct> getByProductCategoryId(String productCategoryId) {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiProduct::getProductCategoryId, productCategoryId);
- queryWrapper.orderByDesc(AiProduct::getSort);
- queryWrapper.orderByDesc(AiProduct::getCreatedTime);
- return this.list(queryWrapper);
- }
-
- @Override
- public List<AiProduct> getByState(Integer state) {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiProduct::getState, state);
- queryWrapper.orderByDesc(AiProduct::getSort);
- queryWrapper.orderByDesc(AiProduct::getCreatedTime);
- return this.list(queryWrapper);
- }
-
- @Override
- public List<AiProduct> getByHotState(Integer hotState) {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiProduct::getHotState, hotState);
- queryWrapper.orderByDesc(AiProduct::getSort);
- queryWrapper.orderByDesc(AiProduct::getCreatedTime);
- return this.list(queryWrapper);
- }
-
- @Override
- public List<AiProduct> getByCompanyIdAndState(String companyId, Integer state) {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiProduct::getCompanyId, companyId);
- queryWrapper.eq(AiProduct::getState, state);
- queryWrapper.orderByDesc(AiProduct::getSort);
- queryWrapper.orderByDesc(AiProduct::getCreatedTime);
- return this.list(queryWrapper);
- }
-
- @Override
- public List<AiProduct> getByCompanyIdAndHotState(String companyId, Integer hotState) {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiProduct::getCompanyId, companyId);
- queryWrapper.eq(AiProduct::getHotState, hotState);
- queryWrapper.orderByDesc(AiProduct::getSort);
- queryWrapper.orderByDesc(AiProduct::getCreatedTime);
- return this.list(queryWrapper);
- }
-
- @Override
- @Transactional(rollbackFor = Exception.class)
- public boolean saveProduct(AiProduct aiProduct) {
- try {
- return this.save(aiProduct);
- } catch (Exception e) {
- log.error("保存AI产品失败: ", e);
- return false;
+ if(StrUtil.isEmpty(dto.getMemberRoleId())){
+ String memberRoleId = aiMemberRoleService.getDefaultMemberRoleId();
+ dto.setMemberRoleId(memberRoleId);
}
+
+ if(StrUtil.isEmpty(dto.getCategoryId())){
+ String categoryId = aiProductCategoryService.getDefaultProductCategoryId();
+ dto.setCategoryId(categoryId);
+ }
+ // 创建分页对象，传入当前页和每页大小
+ 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
- @Transactional(rollbackFor = Exception.class)
- public boolean saveBatchProducts(List<AiProduct> products) {
- try {
- return this.saveBatch(products);
- } catch (Exception e) {
- log.error("批量保存AI产品失败: ", e);
- return false;
+ public FebsResponse productInfo(ApiProductInfoDto dto) {
+
+ ApiProductInfoVo vo = new ApiProductInfoVo();
+ String id = dto.getId();
+ AiProduct entity = this.getById(id);
+ if (ObjectUtil.isNull( entity)){
+ throw new FebsException("产品不存在");
}
+ vo.setId(entity.getId());
+ vo.setProductCategoryId(entity.getProductCategoryId());
+ vo.setName(entity.getName());
+ vo.setScene(entity.getScene());
+ vo.setTarget(entity.getTarget());
+ vo.setBackImg(entity.getBackImg());
+ vo.setIconImg(entity.getIconImg());
+ vo.setDescription(entity.getDescription());
+ vo.setQuestionCount(entity.getQuestionCount());
+
+ vo.setProductPointList(aiProductPointService.listByProductId(id));
+ return new FebsResponse().success().data(vo);
}
@Override
- @Transactional(rollbackFor = Exception.class)
- public boolean updateProduct(AiProduct aiProduct) {
- try {
- return this.updateById(aiProduct);
- } catch (Exception e) {
- log.error("更新AI产品失败: ", e);
- return false;
- }
+ 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
- @Transactional(rollbackFor = Exception.class)
- public boolean deleteById(String id) {
- try {
- return this.removeById(id);
- } catch (Exception e) {
- log.error("删除AI产品失败: ", e);
- return false;
- }
- }
-
- @Override
- @Transactional(rollbackFor = Exception.class)
- public boolean deleteByCompanyId(String companyId) {
- try {
- LambdaQueryWrapper<AiProduct> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(AiProduct::getCompanyId, companyId);
- return this.remove(queryWrapper);
- } catch (Exception e) {
- log.error("根据公司ID删除AI产品失败: ", e);
- return false;
- }
+ public List<AiProduct> getProductListByQuery(LambdaQueryWrapper<AiProduct> productQuery) {
+ return aiProductMapper.selectList(productQuery);
}
}
--
Gitblit v1.9.1