增加json时区，修付开始采集状态，修改导出校验

guyue

2025-08-15 7c1335ff559c08bbcef479bd2b01a08a1f4e55f2

 src/main/java/com/linghu/controller/QuestionController.java

@@ -5,12 +5,17 @@
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;
@@ -20,9 +25,12 @@
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 javax.validation.Valid;
import javax.validation.constraints.NotEmpty;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.*;
@@ -37,166 +45,67 @@

    @Autowired
    private QuestionService questionService;
    @Autowired
    private KeywordService keywordService;

    @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
    @ApiOperation(value = "修改提问词")
    @Transactional
    public ResponseResult<List<Question>> update(@RequestBody KeywordDto keywordDto) {
        //查询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();


    public ResponseResult<List<Question>> update(@Validated(UpdateGroup.class) @RequestBody KeywordDto keywordDto) {
        return questionService.updateQuestion(keywordDto);
    }

    @PutMapping("/update")
    @ApiOperation(value = "修改单个提问词")
    public ResponseResult<Void> update(@RequestBody Question questions) {

        LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Question::getQuestion_id, questions.getQuestion_id());
        Question keyword = questionService.getOne(queryWrapper);
        if ("submitted".equals(keyword.getStatus()) ) {
            return ResponseResult.error("该关键词在采集中");
        }

            boolean success = questionService.updateById(questions);
        if (success) {
            return ResponseResult.success();
        }
        return ResponseResult.error("更新提问词失败");
    public ResponseResult<Void> update(@Validated(UpdateGroup.class) @RequestBody Question questions) {
        return questionService.updateKeyWord(questions);
    }

     @DeleteMapping("/batch")
     @Transactional
     @ApiOperation(value = "批量删除提问词")
     public ResponseResult<Void> batchDelete(@RequestBody List<Integer> questionIds) {

    @DeleteMapping("/batch")
    @ApiOperation(value = "批量删除提问词")
    public ResponseResult<Void> batchDelete( @RequestBody List<Integer> questionIds) {
        return questionService.batchDeleteQuestion(questionIds);
    }

         // 2. 批量查询所有提问词
         List<Question> questions = questionService.listByIds(questionIds);
         // 3. 校验所有提问词关联的关键词是否处于采集中
         for (Question question : questions) {
             Keyword keyword = keywordService.getById(question.getKeyword_id());
             // 关键词不存在也需处理（可选）
             if (keyword == null) {
                 return ResponseResult.error("提问词关联的关键词不存在，ID：" + question.getKeyword_id());
             }
             // 核心校验：若有任何关键词在采集中，禁止删除
             if ("submitted".equals(keyword.getStatus())) {
                 return ResponseResult.error("该关键词在采集中，无法删除");
             }
         }
        questionService.removeByIds(questionIds);
        return ResponseResult.success();
     }

    @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 {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        EasyExcel.write(out, QuestionExcel.class).sheet("提问词模板").doWrite(new ArrayList<>());

        return ResponseEntity.ok()
                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=提问词条导入模板.xlsx")
                .contentType(MediaType.APPLICATION_OCTET_STREAM)
                .body(out.toByteArray());
        return questionService.downloadQuestion();
    }



    // 导入文件
    @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());
        }
        return questionService.importQuestionsExcel(file);
    }


}