From a271ed5a232236383bc96c8c1e380a5c91dc1c3c Mon Sep 17 00:00:00 2001
From: guyue <1721849008@qq.com>
Date: 星期二, 12 八月 2025 01:07:19 +0800
Subject: [PATCH] 代码规范
---
src/main/java/com/linghu/service/impl/QuestionServiceImpl.java | 157 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 157 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/linghu/service/impl/QuestionServiceImpl.java b/src/main/java/com/linghu/service/impl/QuestionServiceImpl.java
index f6ec1bf..fccec91 100644
--- a/src/main/java/com/linghu/service/impl/QuestionServiceImpl.java
+++ b/src/main/java/com/linghu/service/impl/QuestionServiceImpl.java
@@ -1,10 +1,35 @@
package com.linghu.service.impl;
+import com.alibaba.excel.EasyExcel;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.linghu.config.FinalStatus;
+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.QuestionExcel;
+import com.linghu.service.KeywordService;
import com.linghu.service.QuestionService;
import com.linghu.mapper.QuestionMapper;
+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.stereotype.Service;
+import org.springframework.util.CollectionUtils;
+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;
/**
* @author xy
@@ -14,6 +39,138 @@
@Service
public class QuestionServiceImpl extends ServiceImpl<QuestionMapper, Question>
implements QuestionService{
+ @Autowired
+ private KeywordService keywordService;
+ @Autowired
+ private KeywordMapper keywordMapper;
+ @Autowired
+ private ReferenceMapper referenceMapper;
+
+ @Override
+ public ResponseResult<List<Question>> addQuestion(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(FinalStatus.PENDING.getValue());
+
+ return question;
+ }).collect(Collectors.toList());
+
+ boolean success = this.saveBatch(questionList);
+ if (success) {
+ return ResponseResult.success(questionList);
+ }
+ return ResponseResult.error("添加提问词失败");
+ }
+
+ @Override
+ public ResponseResult<Void> updateKeyWord(Question questions) {
+ LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
+ queryWrapper.eq(Question::getQuestion_id, questions.getQuestion_id());
+ Question keyword = this.getOne(queryWrapper);
+ if (FinalStatus.SUBMITTED.getValue().equals(keyword.getStatus()) ) {
+ return ResponseResult.error("该关键词在采集中");
+ }
+
+ boolean success = this.updateById(questions);
+ if (success) {
+ return ResponseResult.success();
+ }
+ return ResponseResult.error("更新提问词失败");
+ }
+
+ @Override
+ public ResponseResult<Void> batchDeleteQuestion(List<Integer> questionIds) {
+ // 新增判断：如果集合为空，直接返回错误
+ if (CollectionUtils.isEmpty(questionIds)) {
+ return ResponseResult.error("删除失败，至少选择一个提问词");
+ }
+ // 2. 批量查询所有提问词
+ List<Question> questions = this.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 (FinalStatus.PENDING.getValue().equals(keyword.getStatus())) {
+ return ResponseResult.error("该关键词在采集中，无法删除");
+ }
+ }
+ this.removeByIds(questionIds);
+ return ResponseResult.success();
+ }
+
+ @Override
+ public ResponseResult<List<Question>> notNullQuestionList(Integer keyword_id) {
+ Keyword keyword = keywordMapper.selectById(keyword_id);
+
+ List<Reference> references = referenceMapper.selectList(new LambdaQueryWrapper<Reference>()
+ .eq(Reference::getKeyword_id, keyword_id)
+ .eq(Reference::getNum, keyword.getNum()));
+ List<Integer> questionIds = references.stream().map(Reference::getQuestion_id).distinct().collect(Collectors.toList());
+ // 查询success状态的 当前轮数的
+ LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
+ queryWrapper.eq(Question::getKeyword_id, keyword_id);
+ List<Question> list = this.list(queryWrapper);
+ return ResponseResult.success(list);
+ }
+
+ @Override
+ public ResponseResult<List<Question>> getQuestionListByKeywordId(Integer keyword_id) {
+ LambdaQueryWrapper<Question> queryWrapper = new LambdaQueryWrapper<>();
+ queryWrapper.eq(Question::getKeyword_id, keyword_id);
+ List<Question> list = this.list(queryWrapper);
+ return ResponseResult.success(list);
+ }
+
+ @Override
+ public ResponseResult<String> importQuestionsExcel(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());
+ }
+ }
+
+ @Override
+ public ResponseResult<List<Question>> updateQuestion(KeywordDto keywordDto) {
+ if (FinalStatus.SUBMITTED.getValue().equals(keywordDto.getStatus()) ) {
+ return ResponseResult.error("该关键词在采集中");
+ }
+ this.updateBatchById(keywordDto.getQuestionList());
+
+ return ResponseResult.success();
+ }
+
+ @Override
+ public ResponseEntity<byte[]> downloadQuestion() {
+ 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());
+ }
+
}
--
Gitblit v1.7.1