From 63446c356df28f927e4d4735b2f867b278c1fec6 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 26 Aug 2025 11:38:30 +0800
Subject: [PATCH] refactor(ai): 优化 AI聊天功能代码
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java | 133 +++++++++++++++++++++++++++++++++++++++-----
1 files changed, 117 insertions(+), 16 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java
index 39fe521..58c9830 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java
@@ -53,6 +53,79 @@
return aiMemberAnswerMapper.selectOne(aiMemberAnswerLambdaQueryWrapper);
}
+ @Override
+ public FebsResponse startAnswerV2(ApiMemberAnswerDto dto) {
+ String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ ApiMemberAnswerVoV2 apiMemberAnswerVoV2 = new ApiMemberAnswerVoV2();
+
+ String productId = dto.getId();
+ AiProduct aiProduct = aiProductService.getById(productId);
+ if(ObjectUtil.isNull(aiProduct)){
+ throw new RuntimeException("产品不存在");
+ }
+ //验证产品分类下是否有题目
+ String productCategoryId = aiProduct.getProductCategoryId();
+ LambdaQueryWrapper<AiProductQuestion> questionLambdaQueryWrapper = Wrappers.lambdaQuery(AiProductQuestion.class);
+ questionLambdaQueryWrapper.eq(AiProductQuestion::getProductCategoryId,productCategoryId);
+ questionLambdaQueryWrapper.eq(AiProductQuestion::getState,1);
+ questionLambdaQueryWrapper.last("limit 1");
+ List<AiProductQuestion> aiProductQuestions = aiProductQuestionService.getListByQuery(questionLambdaQueryWrapper);
+ if(CollUtil.isEmpty(aiProductQuestions)){
+ // todo 没有题目采用AI生成题目
+ throw new RuntimeException("产品没有题目");
+ }
+ //判断该产品是否有正在进行的练习
+ Date createdTime = new Date();
+ LambdaQueryWrapper<AiMemberAnswer> aiMemberAnswerLambdaQueryWrapper = Wrappers.lambdaQuery(AiMemberAnswer.class);
+ aiMemberAnswerLambdaQueryWrapper.eq(AiMemberAnswer::getMemberId,memberUuid);
+ aiMemberAnswerLambdaQueryWrapper.eq(AiMemberAnswer::getProductId,aiProduct.getId());
+ aiMemberAnswerLambdaQueryWrapper.eq(AiMemberAnswer::getState,0);
+ AiMemberAnswer aiMemberAnswer = this.getOneByQuery(aiMemberAnswerLambdaQueryWrapper);
+ if(ObjectUtil.isNull(aiMemberAnswer)){
+ Integer questionCount = aiProduct.getQuestionCount();
+ aiMemberAnswer = new AiMemberAnswer();
+ aiMemberAnswer.setId(UUID.getSimpleUUIDString());
+ aiMemberAnswer.setCreatedTime(createdTime);
+ aiMemberAnswer.setUpdatedTime(createdTime);
+ aiMemberAnswer.setProductId(aiProduct.getId());
+ aiMemberAnswer.setMemberId(memberUuid);
+ aiMemberAnswer.setState(0);
+ aiMemberAnswer.setScore(0);
+ aiMemberAnswer.setQuestionCnt(questionCount);
+ aiMemberAnswer.setDoneCnt(0);
+ aiMemberAnswer.setCorrectCnt(0);
+ aiMemberAnswerMapper.insert(aiMemberAnswer);
+ List<AiMemberAnswerItem> aiMemberAnswerItems = new ArrayList<>();
+ for (int i = 1; i <= questionCount; i++){
+ //初始化答题记录
+ AiProductQuestion aiProductQuestion = aiProductQuestionService.createQuestion(aiProduct.getId());
+ AiMemberAnswerItem aiMemberAnswerItem = new AiMemberAnswerItem();
+ aiMemberAnswerItem.setId(UUID.getSimpleUUIDString());
+ aiMemberAnswerItem.setCreatedTime(createdTime);
+ aiMemberAnswerItem.setAnswerId(aiMemberAnswer.getId());
+ aiMemberAnswerItem.setProductId(aiMemberAnswer.getProductId());
+ aiMemberAnswerItem.setMemberId(memberUuid);
+ aiMemberAnswerItem.setProductQuestionId(aiProductQuestion.getId());
+ aiMemberAnswerItem.setTitle(aiProductQuestion.getTitle());
+ aiMemberAnswerItem.setDifficulty(aiProductQuestion.getDifficulty());
+ aiMemberAnswerItem.setMemberAnswerState(0);
+ aiMemberAnswerItem.setSort( i );
+ aiMemberAnswerItems.add(aiMemberAnswerItem);
+ }
+ aiMemberAnswerItemService.saveBatch(aiMemberAnswerItems);
+ apiMemberAnswerVoV2.setCurrent(1);
+ }else{
+ if (aiMemberAnswer.getQuestionCnt() > aiMemberAnswer.getDoneCnt()){
+ apiMemberAnswerVoV2.setCurrent(aiMemberAnswer.getDoneCnt() + 1);
+ }else{
+ apiMemberAnswerVoV2.setCurrent(aiMemberAnswer.getDoneCnt());
+ }
+ }
+ apiMemberAnswerVoV2.setMemberAnswerId(aiMemberAnswer.getId());
+
+ return new FebsResponse().success().data(apiMemberAnswerVoV2);
+ }
+
@Override
public FebsResponse startAnswer(ApiMemberAnswerDto dto) {
@@ -75,6 +148,7 @@
throw new RuntimeException("产品没有题目");
}
//判断该产品是否有正在进行的练习
+ Date createdTime = new Date();
LambdaQueryWrapper<AiMemberAnswer> aiMemberAnswerLambdaQueryWrapper = Wrappers.lambdaQuery(AiMemberAnswer.class);
aiMemberAnswerLambdaQueryWrapper.eq(AiMemberAnswer::getMemberId,memberUuid);
aiMemberAnswerLambdaQueryWrapper.eq(AiMemberAnswer::getProductId,aiProduct.getId());
@@ -83,7 +157,6 @@
if(ObjectUtil.isNull(aiMemberAnswer)){
aiMemberAnswer = new AiMemberAnswer();
aiMemberAnswer.setId(UUID.getSimpleUUIDString());
- Date createdTime = new Date();
aiMemberAnswer.setCreatedTime(createdTime);
aiMemberAnswer.setUpdatedTime(createdTime);
aiMemberAnswer.setProductId(aiProduct.getId());
@@ -113,6 +186,21 @@
apiMemberAnswerVo.setProductQuestionId(aiProductQuestion.getId());
apiMemberAnswerVo.setTitle(aiProductQuestion.getTitle());
apiMemberAnswerVo.setDifficulty(aiProductQuestion.getDifficulty());
+
+ //初始化答题记录
+ AiMemberAnswerItem aiMemberAnswerItem = new AiMemberAnswerItem();
+ aiMemberAnswerItem.setId(UUID.getSimpleUUIDString());
+ aiMemberAnswerItem.setCreatedTime(createdTime);
+ aiMemberAnswerItem.setAnswerId(aiMemberAnswer.getId());
+ aiMemberAnswerItem.setProductId(aiMemberAnswer.getProductId());
+ aiMemberAnswerItem.setMemberId(memberUuid);
+ aiMemberAnswerItem.setProductQuestionId(aiProductQuestion.getId());
+ aiMemberAnswerItem.setTitle(aiProductQuestion.getTitle());
+ aiMemberAnswerItem.setDifficulty(aiProductQuestion.getDifficulty());
+ aiMemberAnswerItem.setMemberAnswerState(0);
+ aiMemberAnswerItemService.save(aiMemberAnswerItem);
+ apiMemberAnswerVo.setMemberAnswerState(aiMemberAnswerItem.getMemberAnswerState());
+ apiMemberAnswerVo.setMemberAnswerItemId(aiMemberAnswerItem.getId());
List<ApiMemberAnswerItemVo> answerList = new ArrayList<>();
LambdaQueryWrapper<AiProductQuestionItem> productQuestionItemLambdaQueryWrapper = Wrappers.lambdaQuery(AiProductQuestionItem.class);
@@ -154,6 +242,7 @@
String productQuestionItemId = dto.getId();
String productQuestionId = dto.getProductQuestionId();
String memberAnswerId = dto.getMemberAnswerId();
+ String memberAnswerItemId = dto.getMemberAnswerItemId();
AiMemberAnswer aiMemberAnswer = this.getById(memberAnswerId);
if(ObjectUtil.isNull(aiMemberAnswer)){
@@ -161,17 +250,9 @@
}
Date createdTime = new Date();
- AiMemberAnswerItem aiMemberAnswerItem = new AiMemberAnswerItem();
- aiMemberAnswerItem.setId(UUID.getSimpleUUIDString());
- aiMemberAnswerItem.setCreatedTime(createdTime);
- aiMemberAnswerItem.setAnswerId(aiMemberAnswer.getId());
- aiMemberAnswerItem.setProductId(aiMemberAnswer.getProductId());
- aiMemberAnswerItem.setMemberId(memberUuid);
- aiMemberAnswerItem.setProductQuestionId(productQuestionId);
-
+ AiMemberAnswerItem aiMemberAnswerItem = aiMemberAnswerItemService.getById(memberAnswerItemId);
+ aiMemberAnswerItem.setUpdatedTime(createdTime);
AiProductQuestion aiProductQuestion = aiProductQuestionService.getById(productQuestionId);
- aiMemberAnswerItem.setTitle(aiProductQuestion.getTitle());
- aiMemberAnswerItem.setDifficulty(aiProductQuestion.getDifficulty());
LambdaQueryWrapper<AiProductQuestionItem> productQuestionItemLambdaQueryWrapper = Wrappers.lambdaQuery(AiProductQuestionItem.class);
productQuestionItemLambdaQueryWrapper.eq(AiProductQuestionItem::getProductQuestionId,aiProductQuestion.getId());
@@ -195,14 +276,14 @@
}else{
aiMemberAnswerItem.setIsCollected(0);
}
- aiMemberAnswerItemService.save(aiMemberAnswerItem);
+ aiMemberAnswerItemService.updateById(aiMemberAnswerItem);
aiMemberAnswer.setDoneCnt(aiMemberAnswer.getDoneCnt() + 1);
if (aiMemberAnswerItem.getMemberAnswerState() == 1){
aiMemberAnswer.setCorrectCnt(aiMemberAnswer.getCorrectCnt() + 1);
}
aiMemberAnswer.setUpdatedTime(createdTime);
- this.updateById(aiMemberAnswer);
+ aiMemberAnswerMapper.updateById(aiMemberAnswer);
apiQuestionItemInfoVo.setAnswerAnalysis(aiMemberAnswerItem.getAnswerAnalysis());
apiQuestionItemInfoVo.setMemberAnswerState(aiMemberAnswerItem.getMemberAnswerState());
@@ -232,6 +313,15 @@
long between = DateUtil.between(createdTime, updatedTime, DateUnit.MINUTE);
apiMemberAnswerInsureVo.setUseTime(between);
+ apiMemberAnswerInsureVo.setProductId(aiMemberAnswer.getProductId());
+
+ aiMemberAnswerMapper.update(null,
+ Wrappers.lambdaUpdate(AiMemberAnswer.class)
+ .set(AiMemberAnswer::getUpdatedTime, new Date())
+ .set(AiMemberAnswer::getState, 1)
+ .set(AiMemberAnswer::getScore, percentage.intValue())
+ .eq(AiMemberAnswer::getId, memberAnswerId)
+ );
return new FebsResponse().success().data(apiMemberAnswerInsureVo);
}
@@ -239,14 +329,13 @@
public FebsResponse previous(ApiMemberAnswerPreviousDto dto) {
String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
- ApiMemberAnswerPreviousVo apiMemberAnswerPreviousVo = new ApiMemberAnswerPreviousVo();
// 创建分页对象,传入当前页和每页大小
Page<ApiMemberAnswerPreviousVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
Page<ApiMemberAnswerPreviousVo> pageListByQuery = aiMemberAnswerItemService.getPageListByQuery(page, dto);
List<ApiMemberAnswerPreviousVo> records = pageListByQuery.getRecords();
if (CollUtil.isNotEmpty(records)){
- apiMemberAnswerPreviousVo = records.get(0);
+ ApiMemberAnswerPreviousVo apiMemberAnswerPreviousVo = records.get(0);
String memberAnswerId = apiMemberAnswerPreviousVo.getMemberAnswerId();
AiMemberAnswer aiMemberAnswer = this.getById(memberAnswerId);
@@ -254,6 +343,7 @@
AiProduct aiProduct = aiProductService.getById(productId);
apiMemberAnswerPreviousVo.setName(aiProduct.getName());
apiMemberAnswerPreviousVo.setQuestionCnt(aiMemberAnswer.getQuestionCnt());
+ apiMemberAnswerPreviousVo.setDoneCnt(aiMemberAnswer.getDoneCnt());
String productQuestionId = apiMemberAnswerPreviousVo.getProductQuestionId();
AiProductQuestion aiProductQuestion = aiProductQuestionService.getById(productQuestionId);
@@ -277,7 +367,7 @@
apiMemberAnswerPreviousVo.setAnswerList(answerList);
}
}
- return new FebsResponse().success().data(apiMemberAnswerPreviousVo);
+ return new FebsResponse().success().data(pageListByQuery);
}
@Override
@@ -288,4 +378,15 @@
dto.setMemberUuid(memberUuid);
return aiMemberAnswerItemService.wrongPage(dto);
}
+
+ @Override
+ public FebsResponse myWork(ApiMemberProductWorkPageDto dto) {
+ String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ dto.setMemberUuid(memberUuid);
+ // 创建分页对象,传入当前页和每页大小
+ Page<ApiMemberProductWorkVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
+ Page<ApiMemberProductWorkVo> pageListByQuery = this.getBaseMapper().getPageMyWorkListByQuery(page, dto);
+
+ return new FebsResponse().success().data(pageListByQuery);
+ }
}
--
Gitblit v1.9.1