From cdd076be08ba86bbe196b4edc4fee7b4c8ae840c Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Thu, 18 Sep 2025 15:25:51 +0800 Subject: [PATCH] ``` refactor(AiProductServiceImpl): 优化产品信息查询逻辑 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java | 75 +++++++++++++++++++++++++++++++++++-- 1 files changed, 70 insertions(+), 5 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java index 5bf24cf..4f73869 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductPointServiceImpl.java +++ b/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(); @@ -82,9 +106,50 @@ apiProductPointInfoVo.setIsNormal(entity.getIsNormal()); apiProductPointInfoVo.setFinderUserName(entity.getFinderUserName()); apiProductPointInfoVo.setFeedId(entity.getFeedId()); + apiProductPointInfoVo.setFeedImg(entity.getFeedImg()); apiProductPointInfoVo.setTitle(entity.getTitle()); apiProductPointInfoVo.setDescription(entity.getDescription()); } 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.isEmpty(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); + } } -- Gitblit v1.9.1