iailab-plat.git

Merge branch 'master' of http://dlindusit.com:53929/r/iailab-plat

潘志宝

2024-12-25 874dd8d939b4ae61efbb799855c9ddf328722972

提交 \| 用户 \| 时间
7fd198	1	package com.iailab.module.model.api;
潘	2
	3	import com.alibaba.fastjson.JSON;
	4	import com.iailab.module.model.api.mdk.MdkApi;
	5	import com.iailab.module.model.api.mdk.dto.*;
	6	import com.iailab.module.model.mcs.pre.entity.DmModuleEntity;
	7	import com.iailab.module.model.mcs.pre.service.DmModuleService;
	8	import com.iailab.module.model.mcs.pre.service.MmPredictItemService;
ac52ae	9	import com.iailab.module.model.mcs.sche.service.StScheduleRecordService;
潘	10	import com.iailab.module.model.mcs.sche.service.StScheduleSchemeService;
9162d9	11	import com.iailab.module.model.mdk.factory.ItemEntityFactory;
潘	12	import com.iailab.module.model.mdk.factory.PredictItemFactory;
7fd198	13	import com.iailab.module.model.mdk.predict.PredictModuleHandler;
9162d9	14	import com.iailab.module.model.mdk.predict.PredictResultHandler;
054fb9	15	import com.iailab.module.model.mdk.schedule.ScheduleModelHandler;
9162d9	16	import com.iailab.module.model.mdk.vo.DataValueVO;
7fd198	17	import com.iailab.module.model.mdk.vo.ItemVO;
潘	18	import com.iailab.module.model.mdk.vo.PredictResultVO;
054fb9	19	import com.iailab.module.model.mdk.vo.ScheduleResultVO;
7fd198	20	import lombok.extern.slf4j.Slf4j;
潘	21	import org.springframework.beans.factory.annotation.Autowired;
9162d9	22	import org.springframework.util.CollectionUtils;
7fd198	23	import org.springframework.validation.annotation.Validated;
潘	24	import org.springframework.web.bind.annotation.RestController;
	25
b2aca2	26	import java.util.*;
7fd198	27	import java.util.stream.Collectors;
潘	28
	29	/**
	30	* @author PanZhibao
	31	* @Description
	32	* @createTime 2024年08月26日
	33	*/
	34	@Slf4j
	35	@RestController
	36	@Validated
	37	public class MdkApiImpl implements MdkApi {
	38
	39	@Autowired
	40	private DmModuleService dmModuleService;
	41
	42	@Autowired
	43	private MmPredictItemService mmPredictItemService;
	44
	45	@Autowired
	46	private PredictModuleHandler predictModuleHandler;
9162d9	47
潘	48	@Autowired
	49	private PredictResultHandler predictResultHandler;
054fb9	50
潘	51	@Autowired
	52	private ScheduleModelHandler scheduleModelHandler;
ac52ae	53
潘	54	@Autowired
	55	private StScheduleRecordService stScheduleRecordService;
	56
	57	@Autowired
	58	private StScheduleSchemeService stScheduleSchemeService;
7fd198	59
潘	60	/**
	61	* 按模块预测
	62	*
	63	* @param reqDTO
	64	* @return
	65	*/
	66	@Override
148842	67	public MdkPredictModuleRespDTO predictModule(MdkPredictReqDTO reqDTO) {
7fd198	68	MdkPredictModuleRespDTO resp = new MdkPredictModuleRespDTO();
69bd5e	69	resp.setPredictTime(reqDTO.getPredictTime());
D	70	resp.setModuleType(reqDTO.getModuleType());
	71
7fd198	72	Map<String, MdkPredictItemRespDTO> predictItemRespMap = new HashMap<>();
潘	73	try {
	74	if (reqDTO.getPredictTime() == null) {
	75	throw new Exception("PredictTime不能为空");
	76	}
	77	if (reqDTO.getModuleType() == null) {
	78	throw new Exception("ModuleType不能为空");
	79	}
b2aca2	80	Calendar calendar = Calendar.getInstance();
D	81	calendar.setTime(reqDTO.getPredictTime());
	82	calendar.set(Calendar.MILLISECOND, 0);
	83	calendar.set(Calendar.SECOND, 0);
	84	reqDTO.setPredictTime(calendar.getTime());
7fd198	85	log.info("预测参数：" + JSON.toJSONString(reqDTO));
潘	86	MdkPredictModuleRespDTO result = new MdkPredictModuleRespDTO();
	87	result.setPredictTime(reqDTO.getPredictTime());
	88	result.setModuleType(reqDTO.getModuleType());
	89	List<DmModuleEntity> moduleList = dmModuleService.getModuleByModuleType(reqDTO.getModuleType());
	90	log.info("预测计算开始: " + System.currentTimeMillis());
	91	for (DmModuleEntity module : moduleList) {
	92	int intervalTime = 0;
	93	if (module.getPredicttime() != null) {
	94	intervalTime = (int) (reqDTO.getPredictTime().getTime() - module.getPredicttime().getTime()) / (1000 * 60);
	95	}
	96	List<ItemVO> predictItemList = mmPredictItemService.getByModuleId(module.getId());
07890e	97	Map<String, PredictResultVO> predictResultMap = new HashMap<>(predictItemList.size());
D	98	// 分组，先运行normal预测项，再将结果传递给merge预测项
	99	List<ItemVO> normalItems = predictItemList.stream().filter(e -> e.getItemType().equals("NormalItem")).collect(Collectors.toList());
	100	if (!CollectionUtils.isEmpty(normalItems)) {
ac52ae	101	predictModuleHandler.predict(normalItems, reqDTO.getPredictTime(), intervalTime, predictResultMap);
fde993	102	List<ItemVO> mergeItems = predictItemList.stream().filter(e -> e.getItemType().equals("MergeItem")).collect(Collectors.toList());
D	103	if (!CollectionUtils.isEmpty(mergeItems)) {
ac52ae	104	predictModuleHandler.predict(mergeItems, reqDTO.getPredictTime(), intervalTime, predictResultMap);
07890e	105	}
D	106	}
4f1717	107	// 更新Module时间
潘	108	dmModuleService.updatePredictTime(module.getId(), reqDTO.getPredictTime());
	109	if (reqDTO.getIsResult() == null \|\| !reqDTO.getIsResult()) {
	110	return resp;
	111	}
	112	for (Map.Entry<String, PredictResultVO> entry : predictResultMap.entrySet()) {
	113	MdkPredictItemRespDTO itemResp = new MdkPredictItemRespDTO();
	114	itemResp.setItemId(entry.getKey());
	115	itemResp.setPredictTime(reqDTO.getPredictTime());
	116	Map<String, List<MdkPredictDataDTO>> itemPredictData = new HashMap<>();
	117
	118	Map<String, List<DataValueVO>> predictLists = predictResultHandler.convertToPredictData2(entry.getValue());
	119	for (Map.Entry<String, List<DataValueVO>> dataListEntry : predictLists.entrySet()) {
	120	List<MdkPredictDataDTO> predictData = dataListEntry.getValue().stream().map(t -> {
	121	MdkPredictDataDTO dto1 = new MdkPredictDataDTO();
	122	dto1.setDataTime(t.getDataTime());
	123	dto1.setDataValue(t.getDataValue());
	124	return dto1;
	125	}).collect(Collectors.toList());
	126	itemPredictData.put(dataListEntry.getKey(), predictData);
	127	}
	128	itemResp.setPredictData(itemPredictData);
	129	predictItemRespMap.put(entry.getKey(), itemResp);
	130	}
7fd198	131	}
潘	132	log.info("预测计算结束: " + System.currentTimeMillis());
	133	} catch (Exception ex) {
148842	134	ex.printStackTrace();
潘	135	return resp;
7fd198	136	}
潘	137	resp.setPredictItemRespMap(predictItemRespMap);
148842	138	return resp;
7fd198	139	}
潘	140
	141	/**
	142	* 单个预测
	143	*
	144	* @param reqDTO
	145	* @return
	146	*/
	147	@Override
148842	148	public MdkPredictItemRespDTO predictItem(MdkPredictReqDTO reqDTO) {
7fd198	149	MdkPredictItemRespDTO resp = new MdkPredictItemRespDTO();
9162d9	150	try {
1178da	151
D	152	ItemVO itemByItemNo = mmPredictItemService.getItemByItemNo(reqDTO.getItemNo());
	153	List<ItemVO> predictItemList = new ArrayList<>();
	154	predictItemList.add(itemByItemNo);
	155	Map<String, PredictResultVO> predictResultMap = new HashMap<>(predictItemList.size());
ac52ae	156	predictModuleHandler.predict(predictItemList, reqDTO.getPredictTime(), 0, predictResultMap);
1178da	157
D	158	Map<String, List<MdkPredictDataDTO>> itemPredictData = new HashMap<>();
	159
	160	Map<String, List<DataValueVO>> predictLists = predictResultHandler.convertToPredictData2(predictResultMap.get(reqDTO.getItemNo()));
	161	for (Map.Entry<String, List<DataValueVO>> dataListEntry : predictLists.entrySet()) {
	162	List<MdkPredictDataDTO> predictData = dataListEntry.getValue().stream().map(t -> {
	163	MdkPredictDataDTO dto1 = new MdkPredictDataDTO();
	164	dto1.setDataTime(t.getDataTime());
	165	dto1.setDataValue(t.getDataValue());
	166	return dto1;
	167	}).collect(Collectors.toList());
	168	itemPredictData.put(dataListEntry.getKey(), predictData);
9162d9	169	}
1178da	170	resp.setItemId(reqDTO.getItemNo());
9162d9	171	resp.setPredictTime(reqDTO.getPredictTime());
1178da	172	resp.setPredictData(itemPredictData);
D	173	} catch (Exception e) {
	174	throw new RuntimeException(e);
9162d9	175	}
潘	176
148842	177	return resp;
7fd198	178	}
潘	179
	180	/**
	181	* 预测调整
	182	*
	183	* @param reqDTO
	184	* @return
	185	*/
	186	@Override
148842	187	public Boolean predictAutoAdjust(MdkPredictReqDTO reqDTO) {
7fd198	188
潘	189
148842	190	return true;
7fd198	191	}
潘	192
	193	/**
	194	* 执行调度模型
	195	*
	196	* @param reqDTO
	197	* @return
	198	*/
	199	@Override
148842	200	public MdkScheduleRespDTO doSchedule(MdkScheduleReqDTO reqDTO) {
7fd198	201	MdkScheduleRespDTO resp = new MdkScheduleRespDTO();
054fb9	202	resp.setScheduleCode(reqDTO.getScheduleCode());
潘	203	resp.setScheduleTime(reqDTO.getScheduleTime());
	204	try {
	205	log.info("调度计算开始: " + System.currentTimeMillis());
51c1c2	206	ScheduleResultVO scheduleResult = scheduleModelHandler.doSchedule(reqDTO.getScheduleCode(), reqDTO.getScheduleTime());
054fb9	207	resp.setResult(scheduleResult.getResult());
ac52ae	208	stScheduleRecordService.create(scheduleResult);
潘	209	stScheduleSchemeService.updateTime(scheduleResult.getSchemeId(), scheduleResult.getScheduleTime());
054fb9	210	log.info("预测计算结束: " + System.currentTimeMillis());
潘	211	} catch (Exception ex) {
	212	log.info("调度计算异常: " + System.currentTimeMillis());
ac52ae	213	ex.printStackTrace();
148842	214	return resp;
054fb9	215	}
148842	216	return resp;
7fd198	217	}
潘	218	}