ps：最近在做的考试试题导入，留个痕
本人有点菜，实现方式有些简单，各位有什么优化也可以拿去改，0.0；
这个方法可以获取对象试题字段的字段值 和字段名 并且可以根据字段上的注解判断字段是否要处理 有类似的业务还是可以用用的
一.demo测试代码：
public static void main(String[] args) {
AssetsApply assetsApply = new AssetsApply();
Class<?> clazz = assetsApply.getClass();
Field[] fields = clazz.getDeclaredFields();
for (Field field : fields) {
String fieldName = field.getName();
if (field.isAnnotationPresent(ApiModelProperty.class) && fieldName.contains("ageName")) {
field.setAccessible(true);
try {
String fieldValue = (String) field.get(assetsApply);
//字段值不为空
if (StringUtils.isNotBlank(fieldName)) {
System.out.println("fieldName = " + fieldName);
System.out.println("fieldValue = " + fieldValue);
String last = fieldName.substring(fieldName.length() - 1);
System.out.println("last = " + last);
ArrayList<String> strList = new ArrayList<>();
strList.add("a");
System.out.println("strList.contains(last) = " + strList.contains(last));
strList.add("b");
System.out.println("strList.contains(last) = " + strList.contains(last));
strList.add("e");
System.out.println("strList.contains(last) = " + strList.contains(last));
strList.add("c");
System.out.println("strList.contains(last) = " + strList.contains(last));
}
} catch (IllegalAccessException e) {
e.printStackTrace();
}
}
}
}
|
二.业务逻辑代码
我个人主要是使用的easyExcel加上hutu和ruoyi的工具类直接做的比较方便
1.service代码
/**
* 模板导入试题
* @param file 导入excel文件
* @return
*/
@Override
@Transactional(rollbackFor = Exception.class)
public Boolean importQuestions(MultipartFile file, String repoId) {
Assert.isTrue(!file.isEmpty(), "导入数据为空");
importQuestionsListener.setQuestions(new LinkedList<>());
importQuestionsListener.setRepoId(repoId);
String filename = file.getOriginalFilename();
// 判断文件类型
Assert.isTrue(StringUtils.endsWith(filename, "xlsx") || StringUtils.endsWith(filename, "xls"),
"只支持.xlsx或者.xls类型文件导入");
try {
EasyExcel.read(file.getInputStream(), ExamQuestionDto.class, importQuestionsListener)
.sheet().doRead();
return true;
} catch (IOException e) {
return false;
}
}
|
2.easyExcel的监听器
package com.zy.learn.exam.listener;
import cn.hutool.core.util.IdUtil;
import com.alibaba.excel.annotation.ExcelProperty;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.zy.common.dto.learn.exam.ExamQuestionDto;
import com.zy.common.entity.learn.exam.ExamAnswer;
import com.zy.common.entity.learn.exam.ExamQuestion;
import com.zy.common.enums.learn.exam.ExamQuestionCategoryEnum;
import com.zy.common.utils.StringUtils;
import com.zy.learn.exam.service.ExamAnswerService;
import com.zy.learn.exam.service.ExamQuestionService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* 导入试题监听器
*
* @Name ImportQuestionslistener
* @Author hhx
* @Date 2023-11-15 10:28
*/
@Slf4j
@Component
@Scope("prototype")
@RequiredArgsConstructor
public class ImportQuestionsListener extends AnalysisEventListener<ExamQuestionDto> {
private final ExamQuestionService questionService;
private final ExamAnswerService answerService;
private List<ExamQuestion> questions;
private String repoId;
public void setQuestions(List<ExamQuestion> questions) {
this.questions = questions;
}
public List<ExamQuestion> getQuestions() {
return questions;
}
public void setRepoId(String repoId) {
this.repoId = repoId;
}
public String getRepoId() {
return repoId;
}
public static Map<String, Object> map = new HashMap<>();
/**
* 读取
*
* @param questionDto
* @param analysisContext
*/
@Override
@Transactional(rollbackFor = Exception.class)
public void invoke(ExamQuestionDto questionDto, AnalysisContext analysisContext) {
ExamQuestion question = new ExamQuestion();
//处理读取到的答案
List<ExamAnswer> answers = handAnswers(questionDto);
//先都不给问题和答案id
question.setCategory(ExamQuestionCategoryEnum.getByValue(questionDto.getCategory()).getCategory())
.setText(questionDto.getQuestionText())
.setRemark(questionDto.getRemark())
.setAnswers(answers)
.setRepoId(getRepoId());
this.questions.add(question);
}
/**
* 执行完毕
* 保存试卷
*
* @param analysisContext
*/
@Override
public void doAfterAllAnalysed(AnalysisContext analysisContext) {
//1.准备答案集合
ArrayList<ExamAnswer> examAnswers = new ArrayList<>();
//2.处理问题/答案 完善实体
questions.stream().forEach(question -> {
String questionId = IdUtil.getSnowflakeNextIdStr();
question.setQuestionId(questionId);
List<ExamAnswer> answers = question.getAnswers();
if (answers.size() > 0) {
answers.stream().forEach(answer -> {
answer.setAnswerId(IdUtil.getSnowflakeNextIdStr())
.setQuestionId(questionId);
});
examAnswers.addAll(answers);
}
});
//3.保存
//保存试题
questionService.saveBatch(questions);
//保存答案
answerService.saveBatch(examAnswers);
//执行完毕后，清空数据
setQuestions(null);
setRepoId(null);
log.info("试题导入成功");
}
|