package com.panzhihua.common.listen;
|
|
import com.alibaba.excel.context.AnalysisContext;
|
import com.alibaba.excel.event.AnalysisEventListener;
|
import com.alibaba.fastjson.JSONArray;
|
import com.google.common.collect.Lists;
|
import com.panzhihua.common.constants.Constants;
|
import com.panzhihua.common.exceptions.ServiceException;
|
import com.panzhihua.common.model.vos.R;
|
import com.panzhihua.common.model.vos.partybuilding.excel.ComPbCheckUnitErrorExcelVO;
|
import com.panzhihua.common.model.vos.partybuilding.excel.ComPbCheckUnitExcelVO;
|
import com.panzhihua.common.model.vos.partybuilding.excel.ComPbMemberRoleErrorExcelVo;
|
import com.panzhihua.common.model.vos.partybuilding.excel.ComPbMemberRoleExcelVo;
|
import com.panzhihua.common.service.partybuilding.ComPbCheckUnitFeign;
|
import com.panzhihua.common.service.partybuilding.PartyBuildingService;
|
import com.panzhihua.common.utlis.ListUtils;
|
import com.panzhihua.common.utlis.StringUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
import org.springframework.data.redis.core.ValueOperations;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.TimeUnit;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
import java.util.stream.Collectors;
|
|
/**
|
* title: 社区党委导入监听
|
* @author : lyq
|
*/
|
@Slf4j
|
public class ComPbMemberRoleExcelListen extends AnalysisEventListener<Map<Integer, String>> {
|
/**
|
* 每隔5条存储数据库，实际使用中可以3000条，然后清理list ，方便内存回收
|
*/
|
private static final int BATCH_COUNT = 3000;
|
private static int headSize = 0;
|
List<Map<Integer, String>> list = new ArrayList<Map<Integer, String>>();
|
private PartyBuildingService partyBuildingService;;
|
private Long communityId;
|
private Long userId;
|
private Map<Integer, String> headData;
|
private StringRedisTemplate stringRedisTemplate;
|
|
public ComPbMemberRoleExcelListen(PartyBuildingService partyBuildingService, Long communityId, Long userId,
|
StringRedisTemplate stringRedisTemplate) {
|
this.partyBuildingService = partyBuildingService;
|
this.communityId = communityId;
|
this.userId = userId;
|
this.stringRedisTemplate = stringRedisTemplate;
|
}
|
|
@Override
|
public void invoke(Map<Integer, String> data, AnalysisContext context) {
|
list.add(data);
|
if (list.size() >= BATCH_COUNT) {
|
saveData();
|
list.clear();
|
}
|
}
|
|
/**
|
* 这里会一行行的返回头
|
*
|
* @param headMap
|
* @param context
|
*/
|
@Override
|
public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {
|
headSize = headMap.size();
|
headData = headMap;
|
}
|
|
@Override
|
public void doAfterAllAnalysed(AnalysisContext context) {
|
saveData();
|
log.info("所有数据解析完成！");
|
}
|
|
/**
|
* 不是固定的列只能手动处理
|
*/
|
private void saveData() {
|
log.info("表格总数据：" + list.size());
|
if (list.size() == 0) {
|
throw new ServiceException("000", "导入数据为空！");
|
}
|
ValueOperations<String, String> valueOperations = stringRedisTemplate.opsForValue();
|
String key = Constants.PB_MEMBER_ROLE_ERROR_LIST + communityId;
|
int index = 2;
|
try {
|
ArrayList<ComPbMemberRoleExcelVo> voList = Lists.newArrayList();
|
ArrayList<ComPbMemberRoleErrorExcelVo> mistakes = Lists.newArrayList();
|
for (Map<Integer, String> oneData : list) {
|
ComPbMemberRoleExcelVo vo = new ComPbMemberRoleExcelVo();
|
|
if (StringUtils.isEmpty(oneData.get(0))) {
|
index++;
|
mistakes.add(setErrorObject(oneData,"党员姓名不可为空，请填写党员姓名"));
|
continue;
|
}else{
|
//判断导入的名字的长度是否超过30
|
if(oneData.get(0).length() > 10){
|
index++;
|
mistakes.add(setErrorObject(oneData,"党员姓名长度不可超过10，请重新填写党员姓名"));
|
continue;
|
}
|
}
|
vo.setName(oneData.get(0));
|
if (StringUtils.isEmpty(oneData.get(1))) {
|
index++;
|
mistakes.add(setErrorObject(oneData,"身份证号不可为空，请填写身份证号"));
|
continue;
|
}
|
vo.setIdCard(oneData.get(1));
|
|
if (StringUtils.isEmpty(oneData.get(2))) {
|
index++;
|
mistakes.add(setErrorObject(oneData,"手机号不可为空，请填写手机号"));
|
continue;
|
}
|
vo.setPhone(oneData.get(2));
|
|
if (StringUtils.isNotEmpty(oneData.get(3))) {
|
vo.setTypeName(oneData.get(3));
|
}
|
|
if (StringUtils.isEmpty(oneData.get(4))) {
|
index++;
|
mistakes.add(setErrorObject(oneData,"入党日期不可为空，请填写入党日期"));
|
continue;
|
}
|
vo.setJoinTime(oneData.get(4));
|
|
if (StringUtils.isNotEmpty(oneData.get(5))) {
|
vo.setEmploymentTime(oneData.get(5));
|
}
|
if (StringUtils.isNotEmpty(oneData.get(6))) {
|
vo.setPosition(oneData.get(6));
|
}
|
if (StringUtils.isNotEmpty(oneData.get(7))) {
|
vo.setJobResponsibilities(oneData.get(7));
|
}
|
if (StringUtils.isNotEmpty(oneData.get(8))) {
|
vo.setOrg(oneData.get(8));
|
}
|
voList.add(vo);
|
index++;
|
}
|
List<ComPbMemberRoleExcelVo> newVoList =
|
voList.stream().filter(ListUtils.distinctByKey(ComPbMemberRoleExcelVo::getIdCard))
|
.collect(Collectors.toList());
|
R r = partyBuildingService.importPbMemberRole(newVoList, communityId, userId);
|
if (!R.isOk(r)) {
|
List<ComPbMemberRoleErrorExcelVo> list =
|
JSONArray.parseArray(JSONArray.toJSONString(r.getData()), ComPbMemberRoleErrorExcelVo.class);
|
mistakes.addAll(list);
|
valueOperations.set(key, JSONArray.toJSONString(mistakes), 1, TimeUnit.HOURS);
|
throw new ServiceException("500", key);
|
} else {
|
if (!mistakes.isEmpty()) {
|
valueOperations.set(key, JSONArray.toJSONString(mistakes), 1, TimeUnit.HOURS);
|
throw new ServiceException("500", key);
|
}
|
}
|
} catch (NumberFormatException e) {
|
e.printStackTrace();
|
throw new ServiceException("500", "导入失败");
|
}
|
}
|
|
private void setError(Map<Integer, String> map, ComPbMemberRoleErrorExcelVo vo) {
|
vo.setName(map.get(0));
|
vo.setIdCard(map.get(1));
|
vo.setPhone(map.get(2));
|
vo.setTypeName(map.get(3));
|
vo.setJoinTime(map.get(4));
|
vo.setEmploymentTime(map.get(5));
|
vo.setPosition(map.get(6));
|
vo.setJobResponsibilities(map.get(7));
|
}
|
|
/**
|
* 组装错误信息
|
* @param oneData 数据表格对象
|
* @param error 错误信息
|
* @return 错误对象
|
*/
|
private ComPbMemberRoleErrorExcelVo setErrorObject(Map<Integer, String> oneData, String error){
|
ComPbMemberRoleErrorExcelVo mistake = new ComPbMemberRoleErrorExcelVo();
|
setError(oneData, mistake);
|
mistake.setError(error);
|
return mistake;
|
}
|
}
|
