From 70806ef4ad3587cc23113ed7b2a1e60f47c2abc0 Mon Sep 17 00:00:00 2001
From: guyue <1721849008@qq.com>
Date: 星期一, 14 七月 2025 16:19:33 +0800
Subject: [PATCH] 修改取消任务关键词状态

---
 src/main/java/com/linghu/controller/QuestionController.java |   35 +++++++++++++++++++++++++++++------
 1 files changed, 29 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/linghu/controller/QuestionController.java b/src/main/java/com/linghu/controller/QuestionController.java
index 0d90313..cd11e19 100644
--- a/src/main/java/com/linghu/controller/QuestionController.java
+++ b/src/main/java/com/linghu/controller/QuestionController.java
@@ -7,9 +7,11 @@
 import com.linghu.listener.QuestionExcelListener;
 import com.linghu.model.common.ResponseResult;
 import com.linghu.model.dto.KeywordDto;
+import com.linghu.model.entity.Keyword;
 import com.linghu.model.entity.Question;
 import com.linghu.model.excel.KeywordExcel;
 import com.linghu.model.excel.QuestionExcel;
+import com.linghu.service.KeywordService;
 import com.linghu.service.QuestionService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -38,6 +40,8 @@
 
     @Autowired
     private QuestionService questionService;
+    @Autowired
+    private KeywordService keywordService;
 
     @PostMapping
     @ApiOperation(value = "添加提问词")
@@ -79,11 +83,11 @@
     public ResponseResult<List<Question>> update(@RequestBody KeywordDto keywordDto) {
         //查询question的状态
        //遍历
-        for (Question q : keywordDto.getQuestionList()) {
-            if ("success".equals(q.getStatus())) {
-                return ResponseResult.error("该提问词已采集");
+
+            if ("submitted".equals(keywordDto.getStatus()) ) {
+                return ResponseResult.error("该关键词在采集中");
             }
-        }
+
          questionService.updateBatchById(keywordDto.getQuestionList());
 
 //        if (!"notSubmitted".equals(keywordDto.getStatus() )){
@@ -114,7 +118,15 @@
     @PutMapping("/update")
     @ApiOperation(value = "修改单个提问词")
     public ResponseResult<Void> update(@RequestBody Question questions) {
-        boolean success = questionService.updateById(questions);
+
+        LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.eq(Question::getQuestion_id, questions.getQuestion_id());
+        Question keyword = questionService.getOne(queryWrapper);
+        if ("submitted".equals(keyword.getStatus()) ) {
+            return ResponseResult.error("该关键词在采集中");
+        }
+
+            boolean success = questionService.updateById(questions);
         if (success) {
             return ResponseResult.success();
         }
@@ -125,7 +137,18 @@
      @Transactional
      @ApiOperation(value = "批量删除提问词")
      public ResponseResult<Void> batchDelete(@RequestBody List<Integer> questionIds) {
-         System.out.println(questionIds.toString());
+
+        LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.eq(Question::getQuestion_id, questionIds);
+        List<Question> questions = questionService.list(queryWrapper);
+        Question question = questions.get(0);
+         Keyword keyword = keywordService.getById(question.getKeyword_id());
+
+
+         if ("submitted".equals(keyword.getStatus()) ) {
+             return ResponseResult.error("该关键词在采集中");
+         }
+             System.out.println(questionIds.toString());
         questionService.removeByIds(questionIds);
         return ResponseResult.success();
      }

--
Gitblit v1.7.1