From 9a34ef1f3c2aea28705939eb1dc32c28f1222ad9 Mon Sep 17 00:00:00 2001
From: guyue <1721849008@qq.com>
Date: 星期四, 10 七月 2025 18:41:48 +0800
Subject: [PATCH] 修改
---
src/main/java/com/linghu/controller/QuestionController.java | 125 +++++++++++++++++++++++++++++++----------
1 files changed, 94 insertions(+), 31 deletions(-)
diff --git a/src/main/java/com/linghu/controller/QuestionController.java b/src/main/java/com/linghu/controller/QuestionController.java
index aeaa35d..77330f8 100644
--- a/src/main/java/com/linghu/controller/QuestionController.java
+++ b/src/main/java/com/linghu/controller/QuestionController.java
@@ -1,17 +1,28 @@
package com.linghu.controller;
+import com.alibaba.excel.EasyExcel;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
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.Question;
+import com.linghu.model.excel.KeywordExcel;
+import com.linghu.model.excel.QuestionExcel;
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.web.bind.annotation.*;
-import com.linghu.model.dto.KeywordDto;
+import org.springframework.web.multipart.MultipartFile;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
@@ -47,43 +58,57 @@
return ResponseResult.error("添加提问词失败");
}
- // @DeleteMapping("/{questionId}")
- // public ResponseResult<Void> delete(@PathVariable Integer questionId) {
- // Question question = new Question();
- // question.setQuestion_id(questionId);
- // question.setDel_flag(1);
- // boolean success = questionService.updateById(question);
- // if (success) {
- // return ResponseResult.success();
- // }
- // return ResponseResult.error("删除提问词失败");
- // }
+// @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
+ @ApiOperation(value = "修改提问词")
@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);
+ public ResponseResult<List<Question>> update(@RequestBody KeywordDto keywordDto) {
+ if (!"notSubmitted".equals(keywordDto.getStatus() )){
+ return ResponseResult.error("该关键词已提交或者已采集完成不允许修改提问词！");
}
- if (success) {
- return ResponseResult.success();
+ 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());
+
+ boolean success = questionService.saveBatch(questionList);
+
+
+ if(success) {
+ return ResponseResult.success(questionList);
}
return ResponseResult.error("更新提问词失败");
}
- @GetMapping("/{questionId}")
- public ResponseResult<Question> getById(@PathVariable Integer questionId) {
- Question question = questionService.getById(questionId);
- if (question != null) {
- return ResponseResult.success(question);
- }
- 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
@@ -103,10 +128,48 @@
// }
@GetMapping("/list")
- public ResponseResult<List<Question>> 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);
}
+
+ // 下载模板
+ @PostMapping("/downloadTemplate")
+ @ApiOperation("下载模板")
+ public ResponseEntity<byte[]> downloadTemplate() throws IOException {
+ ByteArrayOutputStream out = new ByteArrayOutputStream();
+ EasyExcel.write(out, QuestionExcel.class).sheet("提问词模板").doWrite(new ArrayList<>());
+
+ return ResponseEntity.ok()
+ .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=question_template.xlsx")
+ .contentType(MediaType.APPLICATION_OCTET_STREAM)
+ .body(out.toByteArray());
+ }
+
+ // 导入文件
+ @PostMapping("/import")
+ @ApiOperation("导入提问词词数据")
+ public ResponseResult<String> importPlatforms(@RequestParam("file") MultipartFile file) {
+ try {
+ if (file.isEmpty()) {
+ return ResponseResult.error("上传文件不能为空");
+ }
+ // 创建数据监听器
+ QuestionExcelListener listener = new QuestionExcelListener();
+ EasyExcel.read(file.getInputStream(), QuestionExcel.class, listener)
+ .sheet()
+ .doRead();
+ // 获取并合并关键词
+ String mergedKeywords = String.join("\n", listener.getMergedKeywords());
+ return ResponseResult.success(mergedKeywords);
+ } catch (IOException e) {
+ return ResponseResult.error("文件读取失败：" + e.getMessage());
+ } catch (Exception e) {
+ return ResponseResult.error("导入失败：" + e.getMessage());
+ }
+ }
}
\ No newline at end of file
--
Gitblit v1.7.1