package com.chinaztt.mes.technology.excel;
|
|
import com.alibaba.excel.context.AnalysisContext;
|
import com.alibaba.excel.event.AnalysisEventListener;
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
import com.chinaztt.mes.basic.util.DictUtils;
|
import com.chinaztt.mes.technology.service.RoutingService;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* @Author: liuth
|
* @Date: 2020-11-30 9:46
|
*/
|
public class RoutingUploadListener extends AnalysisEventListener<RoutingExcelData> {
|
|
/**
|
* 每隔5条存储数据库，实际使用中可以3000条，然后清理list ，方便内存回收
|
*/
|
private static final int BATCH_COUNT = 1000;
|
|
List<RoutingExcelData> list = new ArrayList<RoutingExcelData>();
|
|
/**
|
* 假设这个是一个DAO，当然有业务逻辑这个也可以是一个service。当然如果不用存储这个对象没用。
|
*/
|
private RoutingService routingService;
|
private DictUtils dictUtils;
|
|
/**
|
* 如果使用了spring,请使用这个构造方法。每次创建Listener的时候需要把spring管理的类传进来
|
*
|
* @param routingService
|
*/
|
public RoutingUploadListener(RoutingService routingService,DictUtils dictUtils) {
|
this.routingService = routingService;
|
this.dictUtils = dictUtils;
|
}
|
|
/**
|
* 这个每一条数据解析都会来调用
|
*
|
* @param data one row value. Is is same as {@link AnalysisContext#readRowHolder()}
|
* @param context
|
*/
|
@Override
|
public void invoke(RoutingExcelData data, AnalysisContext context) {
|
Map<String, String> bomTypeDb = dictUtils.getDicKey("bom_type_db");
|
//因素单位数据字典
|
Map<String, String> dicruntimedb = dictUtils.getDicKey("runtimecodedb_type");
|
if (StringUtils.isBlank(data.getPartNo()) || StringUtils.isBlank(data.getPartVersion()) || StringUtils.isBlank(data.getChildPartNo()) || StringUtils.isBlank(data.getChildPartVersion())) {
|
throw new RuntimeException("零件数据编号或者版本缺失");
|
}
|
if (StringUtils.isBlank(data.getWorkCenter())) {
|
throw new RuntimeException("工作中心缺失");
|
}
|
if (StringUtils.isBlank(data.getLaborClassNo())) {
|
throw new RuntimeException("人工类别缺失");
|
}
|
if (StringUtils.isBlank(data.getRunTimeCodeDb())) {
|
throw new RuntimeException("因素单位缺失");
|
}
|
if (StringUtils.isBlank(data.getOperationName())) {
|
throw new RuntimeException("工序缺失");
|
}
|
if (StringUtils.isBlank(data.getPhaseInDate())) {
|
throw new RuntimeException("逐步采用时间缺失");
|
}
|
if (StringUtils.isBlank(data.getOperationOrder())) {
|
throw new RuntimeException("工序号缺失");
|
}
|
//设置默认值 替代默认为 *
|
if (StringUtils.isBlank(data.getAlternativeNo())) {
|
data.setAlternativeNo("*");
|
}
|
//默认bom类型是M
|
data.setBomTypeDb(StringUtils.isBlank(data.getBomTypeDb()) ? "M" : bomTypeDb.get(data.getBomTypeDb()));
|
//设置单位
|
data.setRunTimeCodeDb(dicruntimedb.get(data.getRunTimeCodeDb()));
|
//四个时间设置为0
|
if (data.getLaborRunFactor() == null) {
|
data.setLaborRunFactor("0");
|
}
|
if (data.getMachRunFactor() == null) {
|
data.setMachRunFactor("0");
|
}
|
if (data.getLaborSetupTime() == null) {
|
data.setLaborSetupTime("0");
|
}
|
if (data.getMachSetupTime() == null) {
|
data.setMachSetupTime("0");
|
}
|
//零件号+版本号+替代唯一
|
data.setRoutingKey(data.getPartNo() + "_" + data.getPartVersion() + "_" + data.getAlternativeNo());
|
list.add(data);
|
// 达到BATCH_COUNT了，需要去存储一次数据库，防止数据几万条数据在内存，容易OOM
|
if (list.size() >= BATCH_COUNT) {
|
saveData();
|
// 存储完成清理 list
|
list.clear();
|
}
|
}
|
|
/**
|
* 所有数据解析完成了 都会来调用
|
*
|
* @param context
|
*/
|
@Override
|
public void doAfterAllAnalysed(AnalysisContext context) {
|
// 这里也要保存数据，确保最后遗留的数据也存储到数据库
|
saveData();
|
}
|
|
/**
|
* 加上存储数据库
|
*/
|
private void saveData() {
|
routingService.importExcel(list);
|
}
|
}
|
