package com.gkhy.assess.system.excel;
|
|
import cn.hutool.extra.spring.SpringUtil;
|
import com.alibaba.excel.context.AnalysisContext;
|
import com.alibaba.excel.event.AnalysisEventListener;
|
import com.alibaba.excel.util.ListUtils;
|
import com.baomidou.mybatisplus.core.conditions.Wrapper;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.gkhy.assess.common.utils.StringUtils;
|
import com.gkhy.assess.system.domain.SysExpertInfo;
|
import com.gkhy.assess.system.domain.vo.SysExpertInfoExcelVO;
|
import com.gkhy.assess.system.service.SysExpertInfoService;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
@Slf4j
|
|
public class ExpertExcelListener extends AnalysisEventListener<SysExpertInfoExcelVO> {
|
|
|
private static final int BATCH_COUNT=100;
|
|
private List<SysExpertInfoExcelVO> cachedDateList= ListUtils.newArrayListWithExpectedSize(BATCH_COUNT);
|
|
@Override
|
public void onException(Exception exception, AnalysisContext context) throws Exception {
|
log.error("解析异常:",exception);
|
throw exception;
|
}
|
|
@Override
|
public void invoke(SysExpertInfoExcelVO sysExpertInfoExcelVO, AnalysisContext analysisContext) {
|
if(!StringUtils.isBlank(sysExpertInfoExcelVO.getName())){
|
cachedDateList.add(sysExpertInfoExcelVO);
|
}else{
|
System.out.println("ffffff");
|
}
|
if(cachedDateList.size()>=BATCH_COUNT){
|
saveData();
|
cachedDateList.clear();
|
}
|
}
|
|
private void saveData(){
|
for(SysExpertInfoExcelVO sysExpertInfoExcelVO:cachedDateList){
|
String direction=sysExpertInfoExcelVO.getSupportDirectionSafety();
|
String name=sysExpertInfoExcelVO.getName();
|
String idCard = sysExpertInfoExcelVO.getIdCard();
|
SysExpertInfo existExpertInfo = SpringUtil.getBean(SysExpertInfoService.class).getOne(Wrappers.<SysExpertInfo>lambdaQuery()
|
.eq(true, SysExpertInfo::getIdCard, idCard)
|
.last(" limit 1"));
|
List<String> safetyArray = new ArrayList<>();
|
if(!StringUtils.isBlank(direction)) {
|
direction = direction.replace("安全生产：", "").trim();
|
String[] splits = direction.split("、");
|
for (String str : splits) {
|
str = str.trim();
|
String code = "1";
|
if (str.equals("现场检查")) {
|
code = "1";
|
} else if (str.equals("调查评估")) {
|
code = "2";
|
} else if (str.equals("咨询服务")) {
|
code = "3";
|
} else if (str.equals("教育培训")) {
|
code = "4";
|
} else {
|
code = "5";
|
}
|
safetyArray.add(code);
|
}
|
}else{
|
safetyArray.add("5");
|
}
|
if(existExpertInfo!=null){
|
existExpertInfo.setState(2);
|
existExpertInfo.setSupportDirectionSafety(String.join( ",",safetyArray));
|
existExpertInfo.setDomain(sysExpertInfoExcelVO.getDomain());
|
existExpertInfo.setLevel(sysExpertInfoExcelVO.getLevel());
|
SpringUtil.getBean(SysExpertInfoService.class).updateById(existExpertInfo);
|
}else{
|
SysExpertInfo sysExpertInfo=new SysExpertInfo();
|
BeanUtils.copyProperties(sysExpertInfoExcelVO,sysExpertInfo);
|
sysExpertInfo.setSupportDirectionSafety(String.join( ",",safetyArray));
|
sysExpertInfo.setState(2);
|
SpringUtil.getBean(SysExpertInfoService.class).save(sysExpertInfo);
|
}
|
}
|
}
|
|
@Override
|
public void doAfterAllAnalysed(AnalysisContext analysisContext) {
|
saveData();
|
log.info("sheet={}解析完成",analysisContext.readSheetHolder().getSheetName());
|
}
|
}
|
