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 | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++---- 1 files changed, 50 insertions(+), 4 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 26895c7..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,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; @@ -42,6 +41,8 @@ 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) { @@ -81,6 +82,11 @@ @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); @@ -106,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.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