feat(ai): 实现产品依赖解锁功能 - 在AiMemberAnswerServiceImpl中注入AiProductDependencyServ...

Administrator

2025-10-24 413a57e2524cb0b839d5d83d961de3b59814eb6e

 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductServiceImpl.java

@@ -1,6 +1,8 @@
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;
@@ -11,6 +13,7 @@
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;
@@ -21,6 +24,9 @@
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * AI产品表 Service实现类
@@ -37,6 +43,8 @@
    private final AiMemberRoleService aiMemberRoleService;
    private final AiProductCategoryService aiProductCategoryService;
    private final AiProductPointService aiProductPointService;
    private final AiProductDependencyService aiProductDependencyService;
    private final AiMemberProductUnlockService aiMemberProductUnlockService;


    @Override
@@ -64,6 +72,36 @@
        // 创建分页对象，传入当前页和每页大小
        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);
    }