From e4451cbe7eea81c397353e8d5649e52dcbd3b7d1 Mon Sep 17 00:00:00 2001 From: guyue <1721849008@qq.com> Date: 星期五, 05 九月 2025 13:14:00 +0800 Subject: [PATCH] 平台筛选 --- src/main/java/com/linghu/controller/QuestionController.java | 147 ++++++++++++++++++++++++------------------------ 1 files changed, 73 insertions(+), 74 deletions(-) diff --git a/src/main/java/com/linghu/controller/QuestionController.java b/src/main/java/com/linghu/controller/QuestionController.java index 5922e03..dcd03bd 100644 --- a/src/main/java/com/linghu/controller/QuestionController.java +++ b/src/main/java/com/linghu/controller/QuestionController.java @@ -1,24 +1,46 @@ 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 com.linghu.model.dto.KeywordDto; +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") -@Api(value = "提问词相关接口", tags = "设置-提问词") +@Api(value = "提问词相关接口", tags = "订单管理-提问词") public class QuestionController { @Autowired @@ -26,87 +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}") - // 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("删除提问词失败"); - // } + @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("更新提问词失败"); + @ApiOperation(value = "修改提问词") + public ResponseResult<List<Question>> update(@Validated(UpdateGroup.class) @RequestBody KeywordDto keywordDto) { + return questionService.updateQuestion(keywordDto); } - @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("/update") + @ApiOperation(value = "修改单个提问词") + public ResponseResult<Void> update(@Validated(UpdateGroup.class) @RequestBody Question questions) { + return questionService.updateKeyWord(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()); - // 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") - public ResponseResult<List<Question>> list() { - LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>(); - List<Question> list = questionService.list(queryWrapper); - return ResponseResult.success(list); + @ApiOperation("根据关键词查询提问词列表") + 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