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 | 118 ++++++++++++++++++++++++++++++++++++-----------------------
1 files changed, 72 insertions(+), 46 deletions(-)
diff --git a/src/main/java/com/linghu/controller/QuestionController.java b/src/main/java/com/linghu/controller/QuestionController.java
index 52aa5fc..cd11e19 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;
@@ -25,7 +28,10 @@
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
+import java.util.Objects;
import java.util.stream.Collectors;
+
+import static jdk.nashorn.internal.runtime.regexp.joni.Config.log;
@RestController
@RequestMapping("/question")
@@ -34,6 +40,8 @@
@Autowired
private QuestionService questionService;
+ @Autowired
+ private KeywordService keywordService;
@PostMapping
@ApiOperation(value = "添加提问词")
@@ -73,59 +81,77 @@
@ApiOperation(value = "修改提问词")
@Transactional
public ResponseResult<List<Question>> update(@RequestBody KeywordDto keywordDto) {
- if (keywordDto.getStatus() != "customPage"){
- return ResponseResult.error("该关键词已提交或者已采集完成不允许修改提问词！");
- }
+ //查询question的状态
+ //遍历
+
+ if ("submitted".equals(keywordDto.getStatus()) ) {
+ return ResponseResult.error("该关键词在采集中");
+ }
+
+ questionService.updateBatchById(keywordDto.getQuestionList());
+
+// 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("/update")
+ @ApiOperation(value = "修改单个提问词")
+ public ResponseResult<Void> update(@RequestBody Question questions) {
+
LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(Question::getKeyword_id, keywordDto.getKeyword_id());
- questionService.remove(queryWrapper);
+ queryWrapper.eq(Question::getQuestion_id, questions.getQuestion_id());
+ Question keyword = questionService.getOne(queryWrapper);
+ if ("submitted".equals(keyword.getStatus()) ) {
+ return ResponseResult.error("该关键词在采集中");
+ }
- 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());
-
- boolean success = questionService.saveBatch(questionList);
-
-
- if(success) {
- return ResponseResult.success(questionList);
+ boolean success = questionService.updateById(questions);
+ if (success) {
+ return ResponseResult.success();
}
return ResponseResult.error("更新提问词失败");
}
-// @PutMapping
-// @ApiOperation(value = "修改提问词")
-// public ResponseResult<Void> update(@RequestBody Question questions) {
-// 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) {
- // @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, questionIds);
+ List<Question> questions = questionService.list(queryWrapper);
+ Question question = questions.get(0);
+ Keyword keyword = keywordService.getById(question.getKeyword_id());
- // boolean success = questionService.updateBatchById(questions);
- // if (success) {
- // return ResponseResult.success();
- // }
- // return ResponseResult.error("批量删除提问词失败");
- // }
+
+ if ("submitted".equals(keyword.getStatus()) ) {
+ return ResponseResult.error("该关键词在采集中");
+ }
+ System.out.println(questionIds.toString());
+ questionService.removeByIds(questionIds);
+ return ResponseResult.success();
+ }
@GetMapping("/list")
@ApiOperation("根据关键词查询提问词列表")
@@ -145,7 +171,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