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);
	}
}