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

Administrator

1 days ago 841f1631b790d2c4caf24a40eb4830f57a9bafa5

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

@@ -1,23 +1,26 @@
package cc.mrbird.febs.ai.service.impl;

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.AiProductPointLinkService;
import cc.mrbird.febs.ai.service.AiProductPointService;
import cc.mrbird.febs.ai.service.*;
import cc.mrbird.febs.common.entity.FebsResponse;
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.core.toolkit.Wrappers;
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.ArrayList;
import java.util.List;
@@ -37,6 +40,9 @@

    private final AiProductPointMapper aiProductPointMapper;
    private final AiProductPointLinkService aiProductPointLinkService;
    private final AiProductCategoryService aiProductCategoryService;
    private final AiMemberRoleService aiMemberRoleService;
    private final AiMemberTalkService aiMemberTalkService;

    @Override
    public AiProductPoint getById(String id) {
@@ -74,6 +80,24 @@
    }

    @Override
    public FebsResponse productPointList(ApiProductPointPageDto 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<ApiProductPointListVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
        Page<ApiProductPointListVo> pageListByQuery = aiProductPointMapper.selectPageListByQuery(page, dto);
        return new FebsResponse().success().data(pageListByQuery);
    }

    @Override
    public FebsResponse productPointInfo(ApiProductPointInfoDto dto) {
        ApiProductPointInfoVo apiProductPointInfoVo = new ApiProductPointInfoVo();
        String id = dto.getId();
@@ -88,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);
    }
}