From dd028e18a12ad9ae7c43ed09b15ddd6bde1a43e9 Mon Sep 17 00:00:00 2001
From: guyue <1721849008@qq.com>
Date: 星期三, 03 九月 2025 11:27:50 +0800
Subject: [PATCH] 采集中状态修改提前，统计数据合并
---
src/main/java/com/linghu/controller/QuestionController.java | 142 +++++++++++++++++++++++------------------------
1 files changed, 69 insertions(+), 73 deletions(-)
diff --git a/src/main/java/com/linghu/controller/QuestionController.java b/src/main/java/com/linghu/controller/QuestionController.java
index 6c15d93..dcd03bd 100644
--- a/src/main/java/com/linghu/controller/QuestionController.java
+++ b/src/main/java/com/linghu/controller/QuestionController.java
@@ -1,21 +1,42 @@
package com.linghu.controller;
+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.mapper.KeywordMapper;
+import com.linghu.mapper.ReferenceMapper;
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.entity.Reference;
import com.linghu.model.excel.KeywordExcel;
+import com.linghu.model.excel.QuestionExcel;
+import com.linghu.model.validator.CreateGroup;
+import com.linghu.model.validator.UpdateGroup;
+import com.linghu.service.KeywordService;
import com.linghu.service.QuestionService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
import org.springframework.transaction.annotation.Transactional;
+import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
-import java.util.Arrays;
-import java.util.List;
+import javax.validation.Valid;
+import javax.validation.constraints.NotEmpty;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.util.*;
import java.util.stream.Collectors;
+
+import static jdk.nashorn.internal.runtime.regexp.joni.Config.log;
@RestController
@RequestMapping("/question")
@@ -27,89 +48,64 @@
@PostMapping
@ApiOperation(value = "添加提问词")
- @Transactional
- public ResponseResult<List<Question>> add(@RequestBody KeywordDto keywordDto) {
-
- 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);
- }
- return ResponseResult.error("添加提问词失败");
+ public ResponseResult<List<Question>> add(@Validated(CreateGroup.class) @RequestBody KeywordDto keywordDto) {
+ return questionService.addQuestion(keywordDto);
}
- @DeleteMapping("/{questionId}")
- @ApiOperation(value = "删除提问词")
- public ResponseResult<Void> delete(@PathVariable Integer questionId) {
- boolean success = questionService.removeById(questionId);
- if (success) {
- return ResponseResult.success();
- }
- return ResponseResult.error("删除提问词失败");
- }
-
- // @PutMapping
- // @Transactional
- // public ResponseResult<Void> update(@RequestBody List<Question> questions) {
- // boolean success = questionService.updateBatchById(questions);
- // // 不存在的问题id就新增
- // List<Question> newQuestions = questions.stream()
- // .filter(q -> q.getQuestion_id() == null)
- // .collect(Collectors.toList());
- // if (!newQuestions.isEmpty()) {
- // questionService.saveBatch(newQuestions);
- // }
- // 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("更新提问词失败");
+ public ResponseResult<List<Question>> update(@Validated(UpdateGroup.class) @RequestBody KeywordDto keywordDto) {
+ return questionService.updateQuestion(keywordDto);
}
- // @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());
+ @PutMapping("/update")
+ @ApiOperation(value = "修改单个提问词")
+ public ResponseResult<Void> update(@Validated(UpdateGroup.class) @RequestBody Question questions) {
+ return questionService.updateKeyWord(questions);
+ }
- // boolean success = questionService.updateBatchById(questions);
- // if (success) {
- // return ResponseResult.success();
- // }
- // return ResponseResult.error("批量删除提问词失败");
- // }
+
+ @DeleteMapping("/batch")
+ @ApiOperation(value = "批量删除提问词")
+ public ResponseResult<Void> batchDelete( @RequestBody List<Integer> questionIds) {
+ return questionService.batchDeleteQuestion(questionIds);
+ }
+
@GetMapping("/list")
@ApiOperation("根据关键词查询提问词列表")
- public ResponseResult<List<Question>> list(Integer keyword_id) {
- LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
- queryWrapper.eq(Question::getKeyword_id, keyword_id);
- List<Question> list = questionService.list(queryWrapper);
- return ResponseResult.success(list);
+ public ResponseResult<List<Question>> list(@RequestParam("keyword_id") Integer keyword_id) {
+ return questionService.getQuestionListByKeywordId(keyword_id);
}
+
+
+
+ @GetMapping("/notNullList")
+ @ApiOperation("根据关键词查询提问词列表")
+ public ResponseResult<List<Question>> notNullList(@RequestParam("keyword_id") Integer keyword_id) {
+ return questionService.notNullQuestionList(keyword_id);
+
+ }
+
+
+ // 下载模板
+ @PostMapping("/downloadTemplate")
+ @ApiOperation("下载模板")
+ public ResponseEntity<byte[]> downloadTemplate() throws IOException {
+ return questionService.downloadQuestion();
+ }
+
+
+
+ // 导入文件
+ @PostMapping("/import")
+ @ApiOperation("导入提问词词数据")
+ public ResponseResult<String> importPlatforms(@RequestParam("file") MultipartFile file) {
+ return questionService.importQuestionsExcel(file);
+ }
+
+
}
\ No newline at end of file
--
Gitblit v1.7.1