修改取消任务关键词状态

guyue

昨天 70806ef4ad3587cc23113ed7b2a1e60f47c2abc0

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

@@ -2,13 +2,16 @@

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.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.excel.KeywordExcel;
import com.linghu.model.excel.QuestionExcel;
import com.linghu.service.KeywordService;
import com.linghu.service.QuestionService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
@@ -25,7 +28,10 @@
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

import static jdk.nashorn.internal.runtime.regexp.joni.Config.log;

@RestController
@RequestMapping("/question")
@@ -34,6 +40,8 @@

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

    @PostMapping
    @ApiOperation(value = "添加提问词")
@@ -73,59 +81,77 @@
    @ApiOperation(value = "修改提问词")
    @Transactional
    public ResponseResult<List<Question>> update(@RequestBody KeywordDto keywordDto) {
        if (!"notSubmitted".equals(keywordDto.getStatus() )){
            return ResponseResult.error("该关键词已提交或者已采集完成不允许修改提问词！");
        }
        //查询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();


    }

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

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

        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);
            boolean success = questionService.updateById(questions);
        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("更新提问词失败");
//    }
     @DeleteMapping("/batch")
     @Transactional
     @ApiOperation(value = "批量删除提问词")
     public ResponseResult<Void> batchDelete(@RequestBody List<Integer> questionIds) {

    // @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());
        LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Question::getQuestion_id, questionIds);
        List<Question> questions = questionService.list(queryWrapper);
        Question question = questions.get(0);
         Keyword keyword = keywordService.getById(question.getKeyword_id());

    // boolean success = questionService.updateBatchById(questions);
    // if (success) {
    // return ResponseResult.success();
    // }
    // return ResponseResult.error("批量删除提问词失败");
    // }

         if ("submitted".equals(keyword.getStatus()) ) {
             return ResponseResult.error("该关键词在采集中");
         }
             System.out.println(questionIds.toString());
        questionService.removeByIds(questionIds);
        return ResponseResult.success();
     }

    @GetMapping("/list")
    @ApiOperation("根据关键词查询提问词列表")
@@ -145,7 +171,7 @@
        EasyExcel.write(out, QuestionExcel.class).sheet("提问词模板").doWrite(new ArrayList<>());

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