| | |
| | | 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实现类 |
| | |
| | | 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); |
| | | } |
| | | } |