From 65f10033448a07535d610ad56a628d7b3f137480 Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Wed, 27 Aug 2025 17:22:14 +0800 Subject: [PATCH] feat(ai): 优化对话项添加逻辑并增强接口返回值 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java | 34 ++++++++++++++++++++++++++++++++++ 1 files changed, 34 insertions(+), 0 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java index 02dde7b..5a577b8 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductQuestionServiceImpl.java @@ -1,9 +1,14 @@ package cc.mrbird.febs.ai.service.impl; import cc.mrbird.febs.ai.entity.AiProductQuestion; +import cc.mrbird.febs.ai.entity.AiProductQuestionLink; import cc.mrbird.febs.ai.mapper.AiProductQuestionMapper; +import cc.mrbird.febs.ai.service.AiProductQuestionLinkService; import cc.mrbird.febs.ai.service.AiProductQuestionService; +import cn.hutool.core.collection.CollUtil; +import cn.hutool.core.util.RandomUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; @@ -11,6 +16,8 @@ import org.springframework.transaction.annotation.Transactional; import java.util.List; +import java.util.Set; +import java.util.stream.Collectors; /** * AI产品题目 Service实现类 @@ -24,6 +31,33 @@ public class AiProductQuestionServiceImpl extends ServiceImpl<AiProductQuestionMapper, AiProductQuestion> implements AiProductQuestionService { private final AiProductQuestionMapper aiProductQuestionMapper; + private final AiProductQuestionLinkService aiProductQuestionLinkService; + @Override + public AiProductQuestion getById(String id) { + return aiProductQuestionMapper.selectById( id); + } + + @Override + public List<AiProductQuestion> getListByQuery(LambdaQueryWrapper<AiProductQuestion> questionLambdaQueryWrapper) { + return aiProductQuestionMapper.selectList(questionLambdaQueryWrapper); + } + + @Override + public AiProductQuestion createQuestion(String productId) { + AiProductQuestion aiProductQuestion = new AiProductQuestion(); + + LambdaQueryWrapper<AiProductQuestionLink> aiProductQuestionLinkLambdaQueryWrapper = Wrappers.lambdaQuery(AiProductQuestionLink.class); + aiProductQuestionLinkLambdaQueryWrapper.eq(AiProductQuestionLink::getProductId, productId); + List<AiProductQuestionLink> linkList = aiProductQuestionLinkService.getListByQuery(aiProductQuestionLinkLambdaQueryWrapper); + if(CollUtil.isNotEmpty(linkList)){ + List<String> collect = linkList.stream().map(AiProductQuestionLink::getProductQuestionId).collect(Collectors.toList()); + if(CollUtil.isNotEmpty( collect)){ + String questionId = RandomUtil.randomEle(collect); + aiProductQuestion = this.getById(questionId); + } + } + return aiProductQuestion; + } } -- Gitblit v1.9.1