代码规范

guyue

2025-08-12 a271ed5a232236383bc96c8c1e380a5c91dc1c3c

 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);
    }


}