From c2a9635771370e7ec908db7e11a5cdd1b5f7bbf0 Mon Sep 17 00:00:00 2001
From: guyue <1721849008@qq.com>
Date: 星期一, 14 七月 2025 17:23:29 +0800
Subject: [PATCH] 修改删除提问词,用户添加和导入设置创建时间

---
 src/main/java/com/linghu/controller/QuestionController.java |  118 ++++++++++++++++++++++++++++++++++++----------------------
 1 files changed, 73 insertions(+), 45 deletions(-)

diff --git a/src/main/java/com/linghu/controller/QuestionController.java b/src/main/java/com/linghu/controller/QuestionController.java
index 5b6732f..0c2090c 100644
--- a/src/main/java/com/linghu/controller/QuestionController.java
+++ b/src/main/java/com/linghu/controller/QuestionController.java
@@ -2,13 +2,16 @@
 
 import com.alibaba.excel.EasyExcel;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 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;
@@ -22,11 +25,10 @@
 
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.List;
-import java.util.Objects;
+import java.util.*;
 import java.util.stream.Collectors;
+
+import static jdk.nashorn.internal.runtime.regexp.joni.Config.log;
 
 @RestController
 @RequestMapping("/question")
@@ -35,6 +37,8 @@
 
     @Autowired
     private QuestionService questionService;
+    @Autowired
+    private KeywordService keywordService;
 
     @PostMapping
     @ApiOperation(value = "添加提问词")
@@ -74,57 +78,81 @@
     @ApiOperation(value = "修改提问词")
     @Transactional
     public ResponseResult<List<Question>> update(@RequestBody KeywordDto keywordDto) {
-        if (!"notSubmitted".equals(keywordDto.getStatus() )){
-            return ResponseResult.error("该关键词已提交或者已采集完成不允许修改提问词!");
-        }
-        LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.eq(Question::getKeyword_id, keywordDto.getKeyword_id());
-        questionService.remove(queryWrapper);
+        //查询question的状态
+       //遍历
 
-        List<Question> questionList = Arrays.stream(keywordDto.getQuestions().split("\\n"))
-                .filter(q -> !q.trim().isEmpty())
-                .map(q -> {
-                    Question question = new Question();
-                    question.setKeyword_id(keywordDto.getKeyword_id());
-                    question.setQuestion(q.trim());
-                    question.setStatus("pending");
+            if ("submitted".equals(keywordDto.getStatus()) ) {
+                return ResponseResult.error("该关键词在采集中");
+            }
 
-                    return question;
-                }).collect(Collectors.toList());
+         questionService.updateBatchById(keywordDto.getQuestionList());
 
-         questionService.saveBatch(questionList);
+//        if (!"notSubmitted".equals(keywordDto.getStatus() )){
+//            return ResponseResult.error("该关键词已提交或者已采集完成不允许修改提问词!");
+//        }
+//        LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
+//        queryWrapper.eq(Question::getKeyword_id, keywordDto.getKeyword_id());
+//        questionService.remove(queryWrapper);
+
+//        List<Question> questionList = Arrays.stream(keywordDto.getQuestions().split("\\n"))
+//                .filter(q -> !q.trim().isEmpty())
+//                .map(q -> {
+//                    Question question = new Question();
+//                    question.setKeyword_id(keywordDto.getKeyword_id());
+//                    question.setQuestion(q.trim());
+//                    question.setStatus("pending");
+//
+//                    return question;
+//                }).collect(Collectors.toList());
+
+//         questionService.saveBatch(questionList);
 
          return ResponseResult.success();
 
 
     }
 
-//    @PutMapping
-//    @ApiOperation(value = "修改提问词")
-//    public ResponseResult<Void> update(@RequestBody Question questions) {
-//        boolean success = questionService.updateById(questions);
-//        if (success) {
-//            return ResponseResult.success();
-//        }
-//        return ResponseResult.error("更新提问词失败");
-//    }
+    @PutMapping("/update")
+    @ApiOperation(value = "修改单个提问词")
+    public ResponseResult<Void> update(@RequestBody Question questions) {
 
-    // @DeleteMapping("/batch")
-    // @Transactional
-    // public ResponseResult<Void> batchDelete(@RequestBody List<Integer>
-    // questionIds) {
-    // List<Question> questions = questionIds.stream().map(id -> {
-    // Question question = new Question();
-    // question.setQuestion_id(id);
-    // return question;
-    // }).collect(Collectors.toList());
+        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.updateBatchById(questions);
-    // if (success) {
-    // return ResponseResult.success();
-    // }
-    // return ResponseResult.error("批量删除提问词失败");
-    // }
+            boolean success = questionService.updateById(questions);
+        if (success) {
+            return ResponseResult.success();
+        }
+        return ResponseResult.error("更新提问词失败");
+    }
+
+     @DeleteMapping("/batch")
+     @Transactional
+     @ApiOperation(value = "批量删除提问词")
+     public ResponseResult<Void> batchDelete(@RequestBody List<Integer> questionIds) {
+
+
+         // 2. 批量查询所有提问词
+         List<Question> questions = questionService.listByIds(questionIds);
+         // 3. 校验所有提问词关联的关键词是否处于采集中
+         for (Question question : questions) {
+             Keyword keyword = keywordService.getById(question.getKeyword_id());
+             // 关键词不存在也需处理(可选)
+             if (keyword == null) {
+                 return ResponseResult.error("提问词关联的关键词不存在,ID:" + question.getKeyword_id());
+             }
+             // 核心校验:若有任何关键词在采集中,禁止删除
+             if ("submitted".equals(keyword.getStatus())) {
+                 return ResponseResult.error("该关键词在采集中,无法删除");
+             }
+         }
+        questionService.removeByIds(questionIds);
+        return ResponseResult.success();
+     }
 
     @GetMapping("/list")
     @ApiOperation("根据关键词查询提问词列表")
@@ -144,7 +172,7 @@
         EasyExcel.write(out, QuestionExcel.class).sheet("提问词模板").doWrite(new ArrayList<>());
 
         return ResponseEntity.ok()
-                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=question_template.xlsx")
+                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=提问词条导入模板.xlsx")
                 .contentType(MediaType.APPLICATION_OCTET_STREAM)
                 .body(out.toByteArray());
     }

--
Gitblit v1.7.1