From c7cee264bccb5026fd42a9f0dc83d274aaee40cf Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 30 Oct 2025 10:05:28 +0800
Subject: [PATCH] feat(ai): 新增AI陪练流式对话V2接口
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java | 134 ++++++++++++++++++++++++++++++++++++++++----
1 files changed, 122 insertions(+), 12 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java
index ac90ba3..1ea68f5 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberServiceImpl.java
@@ -1,14 +1,11 @@
package cc.mrbird.febs.ai.service.impl;
import cc.mrbird.febs.ai.entity.*;
+import cc.mrbird.febs.ai.enumerates.AiTypeEnum;
import cc.mrbird.febs.ai.mapper.AiMemberMapper;
-import cc.mrbird.febs.ai.req.member.ApiMemberTeamPageDto;
-import cc.mrbird.febs.ai.req.member.ApiMemberTeamPracticeDto;
-import cc.mrbird.febs.ai.req.member.ApiMemberTeamStudyDto;
-import cc.mrbird.febs.ai.res.member.ApiMemberTeamPageVo;
-import cc.mrbird.febs.ai.res.member.ApiMemberTeamPracticeVo;
-import cc.mrbird.febs.ai.res.member.ApiMemberTeamStudyVo;
-import cc.mrbird.febs.ai.res.productPoint.ApiProductPointListVo;
+import cc.mrbird.febs.ai.req.member.*;
+import cc.mrbird.febs.ai.res.member.*;
+import cc.mrbird.febs.ai.res.memberAnswer.ApiMemberAnswerPreviousItemVo;
import cc.mrbird.febs.ai.service.*;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.exception.FebsException;
@@ -40,9 +37,11 @@
private final MallMemberMapper mallMemberMapper;
private final AiMemberTalkService aiMemberTalkService;
private final AiMemberAnswerService aiMemberAnswerService;
+ private final AiMemberAnswerItemService aiMemberAnswerItemService;
private final AiMemberPointService aiMemberPointService;
private final AiProductService aiProductService;
private final AiProductPointService aiProductPointService;
+ private final AiProductQuestionItemService aiProductQuestionItemService;
@Override
public AiMember getById(String id) {
return aiMemberMapper.selectById( id);
@@ -158,7 +157,17 @@
String companyId = LoginUserUtil.getLoginUser().getCompanyId();
String memberUuid = dto.getMemberUuid();
- List<AiMemberTalk> aiMemberTalks = aiMemberTalkService.getListByCompanyIdAndMemberUuid(companyId, memberUuid);
+ // 创建分页对象，传入当前页和每页大小
+ Page<AiMemberTalk> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ LambdaQueryWrapper<AiMemberTalk> queryWrapper = Wrappers.lambdaQuery(AiMemberTalk.class);
+ queryWrapper.eq(AiMemberTalk::getCompanyId, companyId);
+ queryWrapper.eq(AiMemberTalk::getMemberId, memberUuid);
+ queryWrapper.eq(AiMemberTalk::getState, AiTypeEnum.AI_MEMBER_TALK_STATE_DONE.getCode());
+ Page<AiMemberTalk> pageListByQuery = aiMemberTalkService.getBaseMapper().selectPage(page, queryWrapper);
+ List<AiMemberTalk> aiMemberTalks = pageListByQuery.getRecords();
+
+// List<AiMemberTalk> aiMemberTalks = aiMemberTalkService.getListByCompanyIdAndMemberUuid(companyId, memberUuid);
+
Map<String, List<AiMemberTalk>> aiMemberTalkMap = new HashMap<>();
if (CollUtil.isNotEmpty(aiMemberTalks)){
//stream操作aiMemberTalks，返回一个根据productId分组的集合
@@ -196,6 +205,8 @@
String productId = aiProduct.getId();
ApiMemberTeamPracticeVo vo = new ApiMemberTeamPracticeVo();
+ vo.setMemberUuid(memberUuid);
+ vo.setProductId(aiProduct.getId());
vo.setName(aiProduct.getName());
vo.setTarget(aiProduct.getTarget());
@@ -226,13 +237,24 @@
String companyId = LoginUserUtil.getLoginUser().getCompanyId();
String memberUuid = dto.getMemberUuid();
- List<AiMemberPoint> aiMemberPoints = aiMemberPointService.getListByCompanyIdAndMemberUuid(companyId, memberUuid);
+
+
+ // 创建分页对象，传入当前页和每页大小
+ Page<AiMemberPoint> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ LambdaQueryWrapper<AiMemberPoint> queryWrapper = Wrappers.lambdaQuery(AiMemberPoint.class);
+ queryWrapper.eq(AiMemberPoint::getCompanyId, companyId);
+ queryWrapper.eq(AiMemberPoint::getMemberId, memberUuid);
+ Page<AiMemberPoint> pageListByQuery = aiMemberPointService.getBaseMapper().selectPage(page, queryWrapper);
+ List<AiMemberPoint> aiMemberPoints = pageListByQuery.getRecords();
+
+// List<AiMemberPoint> aiMemberPoints = aiMemberPointService.getListByCompanyIdAndMemberUuid(companyId, memberUuid);
if (CollUtil.isEmpty(aiMemberPoints)){
return new FebsResponse().success().data(vos);
}
- Map<String, AiMemberPoint> aiMemberPointMap = new HashMap<>();
+ Map<String, List<AiMemberPoint>> aiMemberPointMap = new HashMap<>();
if (CollUtil.isNotEmpty(aiMemberPoints)){
- aiMemberPoints.forEach(aiMemberPoint -> aiMemberPointMap.put(aiMemberPoint.getProductPointId(),aiMemberPoint));
+ //stream操作aiMemberPoints，返回一个Map<productPointId,List<AiMemberPoint>>
+ aiMemberPointMap = aiMemberPoints.stream().collect(Collectors.groupingBy(AiMemberPoint::getProductPointId));
}
//stream流操作aiMemberPoints，返回一个productPointId的set集合
@@ -252,7 +274,10 @@
vo.setTitle(aiProductPoint.getTitle());
Integer totalTime = 0;
if (aiMemberPointMap.containsKey(aiProductPoint.getId())){
- totalTime = aiMemberPointMap.get(aiProductPoint.getId()).getTotalTime();
+ List<AiMemberPoint> memberPoints = aiMemberPointMap.get(aiProductPoint.getId());
+ for (AiMemberPoint memberPoint : memberPoints){
+ totalTime += memberPoint.getTotalTime();
+ }
}
vo.setTotalTime(DateUtil.secondToTime(totalTime));
@@ -261,4 +286,89 @@
return new FebsResponse().success().data(vos);
}
+
+ @Override
+ public FebsResponse talkPage(ApiMemberTalkPageDto dto) {
+
+ String companyId = LoginUserUtil.getLoginUser().getCompanyId();
+ dto.setCompanyId(companyId);
+ // 创建分页对象，传入当前页和每页大小
+ Page<ApiMemberTalkPageVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ Page<ApiMemberTalkPageVo> pages = aiMemberTalkService.getTalkPage(page, dto);
+
+ return new FebsResponse().success().data(pages);
+ }
+
+ @Override
+ public FebsResponse answerPage(ApiMemberAnswerPageDto dto) {
+
+ String companyId = LoginUserUtil.getLoginUser().getCompanyId();
+ dto.setCompanyId(companyId);
+ // 创建分页对象，传入当前页和每页大小
+ Page<ApiMemberAnswerPageVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ Page<ApiMemberAnswerPageVo> pages = aiMemberAnswerService.getAnswerPage(page, dto);
+
+ return new FebsResponse().success().data(pages);
+ }
+
+ @Override
+ public FebsResponse answerInfo(ApiMemberAnswerInfoDto dto) {
+ List<ApiMemberAnswerInfoVo> vos = new ArrayList<>();
+
+ String answerId = dto.getAnswerId();
+ AiMemberAnswer memberAnswer = aiMemberAnswerService.getById(answerId);
+ if (memberAnswer == null){
+ return new FebsResponse().success().data(vos);
+ }
+
+ List<AiMemberAnswerItem> records = aiMemberAnswerItemService.getBaseMapper().selectList(
+ Wrappers.lambdaQuery(AiMemberAnswerItem.class)
+ .eq(AiMemberAnswerItem::getAnswerId, memberAnswer.getId())
+ .orderByAsc(AiMemberAnswerItem::getCreatedTime)
+ );
+
+ if (CollUtil.isNotEmpty(records)){
+ Set<String> collect = records.stream().map(AiMemberAnswerItem::getProductQuestionId).collect(Collectors.toSet());
+
+ LambdaQueryWrapper<AiProductQuestionItem> questionItemQuery = Wrappers.lambdaQuery(AiProductQuestionItem.class);
+ questionItemQuery.in(AiProductQuestionItem::getProductQuestionId,collect);
+ List<AiProductQuestionItem> listByQuery = aiProductQuestionItemService.getListByQuery(questionItemQuery);
+
+ if (CollUtil.isNotEmpty(listByQuery)){
+ //利用stream流，操作集合listByQuery，返回一个map对象，key为productQuestionId，value为List<AiProductQuestionItem>对象
+ Map<String, List<AiProductQuestionItem>> questionItemMap =
+ listByQuery.stream().collect(Collectors.groupingBy(AiProductQuestionItem::getProductQuestionId));
+
+ for (AiMemberAnswerItem record : records){
+ String productQuestionId = record.getProductQuestionId();
+ ApiMemberAnswerInfoVo vo = new ApiMemberAnswerInfoVo();
+ vo.setTitle(record.getTitle());
+ vo.setDifficulty(record.getDifficulty());
+ vo.setMemberAnswerQuestionId(record.getMemberAnswerQuestionId());
+ vo.setAnswerAnalysis(record.getAnswerAnalysis());
+ vo.setProductQuestionId(record.getProductQuestionId());
+ vo.setCreatedTime(record.getUpdatedTime());
+
+
+ List<AiProductQuestionItem> questionItemList = questionItemMap.get(productQuestionId);
+ List<ApiMemberAnswerPreviousItemVo> answerList = new ArrayList<>();
+ if (CollUtil.isNotEmpty(questionItemList)){
+ for (AiProductQuestionItem questionItem : questionItemList){
+ ApiMemberAnswerPreviousItemVo itemVo = new ApiMemberAnswerPreviousItemVo();
+ itemVo.setId(questionItem.getId());
+ itemVo.setAnswer(questionItem.getAnswer());
+ itemVo.setAnswerAnalysis(questionItem.getAnswerAnalysis());
+ itemVo.setCorrectAnswer(questionItem.getCorrectAnswer());
+ answerList.add(itemVo);
+ }
+ }
+ vo.setAnswerList(answerList);
+
+ vos.add( vo);
+ }
+ }
+ }
+
+ return new FebsResponse().success().data(vos);
+ }
}
--
Gitblit v1.9.1