From 8130bed63a505cf718481c0d95d3fcf7b037db96 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 16 Sep 2025 15:11:45 +0800
Subject: [PATCH] refactor(ai): 优化代码结构和更新时间处理

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberAnswerServiceImpl.java |  124 +++++++++++++++++++++++++++++++++++++++--
 1 files changed, 117 insertions(+), 7 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 44a88fd..2434673 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
@@ -9,6 +9,7 @@
 import cc.mrbird.febs.ai.service.*;
 import cc.mrbird.febs.ai.utils.UUID;
 import cc.mrbird.febs.common.entity.FebsResponse;
+import cc.mrbird.febs.common.exception.FebsException;
 import cc.mrbird.febs.common.utils.LoginUserUtil;
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.date.DateUnit;
@@ -53,6 +54,82 @@
         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 FebsException("产品不存在");
+        }
+        String companyId = aiProduct.getCompanyId();
+        //验证产品分类下是否有题目
+        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 FebsException("产品没有题目");
+        }
+        //判断该产品是否有正在进行的练习
+        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.setCompanyId(companyId);
+            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.setCompanyId(companyId);
+                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) {
@@ -61,7 +138,7 @@
         String productId = dto.getId();
         AiProduct aiProduct = aiProductService.getById(productId);
         if(ObjectUtil.isNull(aiProduct)){
-            throw new RuntimeException("产品不存在");
+            throw new FebsException("产品不存在");
         }
         //验证产品分类下是否有题目
         String productCategoryId = aiProduct.getProductCategoryId();
@@ -72,7 +149,7 @@
         List<AiProductQuestion> aiProductQuestions = aiProductQuestionService.getListByQuery(questionLambdaQueryWrapper);
         if(CollUtil.isEmpty(aiProductQuestions)){
             // todo 没有题目采用AI生成题目
-            throw new RuntimeException("产品没有题目");
+            throw new FebsException("产品没有题目");
         }
         //判断该产品是否有正在进行的练习
         Date createdTime = new Date();
@@ -173,12 +250,14 @@
         AiMemberAnswer aiMemberAnswer = this.getById(memberAnswerId);
 
         if(ObjectUtil.isNull(aiMemberAnswer)){
-            throw new RuntimeException("答题记录不存在");
+            throw new FebsException("答题记录不存在");
         }
 
-        Date createdTime = new Date();
+        Date nowTime = new Date();
+
+        String companyId = aiMemberAnswer.getCompanyId();
         AiMemberAnswerItem aiMemberAnswerItem = aiMemberAnswerItemService.getById(memberAnswerItemId);
-        aiMemberAnswerItem.setUpdatedTime(createdTime);
+        aiMemberAnswerItem.setUpdatedTime(nowTime);
         AiProductQuestion aiProductQuestion = aiProductQuestionService.getById(productQuestionId);
 
         LambdaQueryWrapper<AiProductQuestionItem> productQuestionItemLambdaQueryWrapper = Wrappers.lambdaQuery(AiProductQuestionItem.class);
@@ -203,13 +282,14 @@
         }else{
             aiMemberAnswerItem.setIsCollected(0);
         }
-        aiMemberAnswerItemService.save(aiMemberAnswerItem);
+        aiMemberAnswerItem.setCompanyId(companyId);
+        aiMemberAnswerItemService.updateById(aiMemberAnswerItem);
 
         aiMemberAnswer.setDoneCnt(aiMemberAnswer.getDoneCnt() + 1);
         if (aiMemberAnswerItem.getMemberAnswerState() == 1){
             aiMemberAnswer.setCorrectCnt(aiMemberAnswer.getCorrectCnt() + 1);
         }
-        aiMemberAnswer.setUpdatedTime(createdTime);
+        aiMemberAnswer.setUpdatedTime(nowTime);
         aiMemberAnswerMapper.updateById(aiMemberAnswer);
 
         apiQuestionItemInfoVo.setAnswerAnalysis(aiMemberAnswerItem.getAnswerAnalysis());
@@ -240,6 +320,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);
     }
 
@@ -296,4 +385,25 @@
         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);
+    }
+
+    @Override
+    public void updateMemberAnswerUpdateTime(String memberAnswerId, Date updateTime) {
+        aiMemberAnswerMapper.update(
+                null,
+                Wrappers.lambdaUpdate(AiMemberAnswer.class)
+                .set(AiMemberAnswer::getUpdatedTime,updateTime)
+                .eq(AiMemberAnswer::getId,memberAnswerId)
+        );
+    }
 }

--
Gitblit v1.9.1