feat(ai): 添加知识点推荐功能- 新增 AiProductPointService接口的 recommend 方法 - 实现 AiProduc...

Administrator

2 days ago 841f1631b790d2c4caf24a40eb4830f57a9bafa5

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

@@ -1,14 +1,13 @@
package cc.mrbird.febs.ai.service.impl;

import cc.mrbird.febs.ai.entity.AiMemberRoleProduct;
import cc.mrbird.febs.ai.entity.AiProduct;
import cc.mrbird.febs.ai.entity.AiProductPoint;
import cc.mrbird.febs.ai.entity.AiProductPointLink;
import cc.mrbird.febs.ai.entity.*;
import cc.mrbird.febs.ai.mapper.AiProductPointMapper;
import cc.mrbird.febs.ai.req.productPoint.ApiProductPointInfoDto;
import cc.mrbird.febs.ai.req.productPoint.ApiProductPointPageDto;
import cc.mrbird.febs.ai.req.productPoint.ApiProductPointRecommendDto;
import cc.mrbird.febs.ai.res.productPoint.ApiProductPointInfoVo;
import cc.mrbird.febs.ai.res.productPoint.ApiProductPointListVo;
import cc.mrbird.febs.ai.res.productPoint.ApiProductPointRecommendVo;
import cc.mrbird.febs.ai.res.productPoint.ApiProductPointVo;
import cc.mrbird.febs.ai.service.*;
import cc.mrbird.febs.common.entity.FebsResponse;
@@ -43,6 +42,7 @@
    private final AiProductPointLinkService aiProductPointLinkService;
    private final AiProductCategoryService aiProductCategoryService;
    private final AiMemberRoleService aiMemberRoleService;
    private final AiMemberTalkService aiMemberTalkService;

    @Override
    public AiProductPoint getById(String id) {
@@ -112,4 +112,44 @@
        }
        return new FebsResponse().success().data(apiProductPointInfoVo);
    }

    @Override
    public FebsResponse recommend(ApiProductPointRecommendDto dto) {

        List<ApiProductPointRecommendVo> vos = new  ArrayList<>();

        String memberTalkId = dto.getMemberTalkId();
        AiMemberTalk aiMemberTalk = aiMemberTalkService.getById(memberTalkId);
        if (ObjectUtil.isNotNull(aiMemberTalk)){
            return new FebsResponse().success().data(vos);
        }
        String productId = aiMemberTalk.getProductId();
        List<AiProductPointLink> aiProductPointLinkList = aiProductPointLinkService.getListByProductId(
                Wrappers.lambdaQuery(AiProductPointLink.class)
                        .eq(AiProductPointLink::getProductId, productId)
        );
        if (CollUtil.isEmpty(aiProductPointLinkList)){
            return new FebsResponse().success().data(vos);
        }

        //stream流操作aiProductPointLinkList,返回一个Set<productPointId>的set集合
        Set<String> productPointIdSet = aiProductPointLinkList.stream().map(AiProductPointLink::getProductPointId).collect(Collectors.toSet());

        List<AiProductPoint> listByPointIds = aiProductPointMapper.selectList(
                Wrappers.lambdaQuery(AiProductPoint.class)
                .select(AiProductPoint::getId,AiProductPoint::getTitle)
                .in(AiProductPoint::getId, productPointIdSet)
        );
        if (CollUtil.isEmpty(listByPointIds)){
            return new FebsResponse().success().data(vos);
        }

        for (AiProductPoint aiProductPoint : listByPointIds){
            ApiProductPointRecommendVo vo = new ApiProductPointRecommendVo();
            vo.setId(aiProductPoint.getId());
            vo.setTitle(aiProductPoint.getTitle());
            vos.add(vo);
        }
        return new FebsResponse().success().data(vos);
    }
}