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实现类
|
*
|
* @author yourname
|
* @date 2025-07-29
|
*/
|
@Slf4j
|
@Service
|
@RequiredArgsConstructor
|
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 aiProductMapper.selectById( id);
|
}
|
|
@Override
|
public Page<ApiProductVo> getPageListByQuery(Page<ApiProductVo> page, ApiProductPageDto dto) {
|
return aiProductMapper.selectPageListByQuery(page, dto);
|
}
|
|
@Override
|
public FebsResponse productList(ApiProductPageDto dto) {
|
|
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
|
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
|
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);
|
}
|
}
|
