iailab-plat.git

潘志宝

2024-12-24 140065f50679c04ea873db3f6c958358b8dd8ddc

提交 \| 用户 \| 时间
a6de49	1	package com.iailab.module.data.point.collection.handler;
H	2
140065	3	import com.iailab.framework.common.util.string.StrUtils;
781e72	4	import com.iailab.module.data.common.enums.CommonConstant;
a6de49	5	import com.iailab.module.data.common.enums.DataTypeEnum;
H	6	import com.iailab.module.data.common.enums.JsErrorCode;
	7	import com.iailab.module.data.common.utils.JavaScriptHandler;
781e72	8	import com.iailab.module.data.point.collection.PointCollector;
a6de49	9	import com.iailab.module.data.point.collection.utils.GenInfluxPointValueUtils;
H	10	import com.iailab.module.data.point.dto.DaPointDTO;
	11	import com.iailab.module.data.point.service.DaPointService;
	12	import com.iailab.module.data.influxdb.pojo.InfluxPointValuePOJO;
	13	import lombok.extern.slf4j.Slf4j;
	14	import org.apache.commons.lang3.StringUtils;
	15	import javax.annotation.Resource;
781e72	16
潘	17	import org.springframework.beans.factory.annotation.Autowired;
	18	import org.springframework.data.redis.core.RedisTemplate;
a6de49	19	import org.springframework.stereotype.Component;
H	20	import org.springframework.util.CollectionUtils;
	21
	22	import java.math.BigDecimal;
	23	import java.util.*;
	24
	25	/**
	26	* 计算点处理
	27	*
	28	* @author PanZhibao
	29	* @Description
	30	* @createTime 2023年05月03日 17:40:00
	31	*/
	32	@Slf4j
	33	@Component
	34	public class CalculateHandle {
	35
	36	@Resource
	37	private DaPointService daPointService;
	38
	39	@Resource
	40	private MeasureHandle measureHandle;
	41
	42	@Resource
	43	private ConstantHandle constantHandle;
	44
	45	@Resource
	46	private JavaScriptHandler javaScriptHandler;
781e72	47
潘	48	@Autowired
	49	private RedisTemplate<String, Object> redisTemplate;
a6de49	50
d41f14	51	public static final String regex = "[+\\-\\*/()\\&\\\|\\>\\<]";
a6de49	52
H	53	public List<InfluxPointValuePOJO> handle(Date collectTime, List<DaPointDTO> dtos, Map<String, Object> dataMap) {
	54	List<InfluxPointValuePOJO> result = new ArrayList<>();
	55	try {
	56	log.info("计算点处理开始");
	57	if (CollectionUtils.isEmpty(dtos)) {
	58	return result;
	59	}
	60	dtos.forEach(dto -> {
	61	try {
	62	Object value = singleCompute(dto, dataMap);
	63	InfluxPointValuePOJO pojo = GenInfluxPointValueUtils.getByPoint(dto, value);
	64	pojo.setTimestamp(collectTime.toInstant());
	65	result.add(pojo);
	66	} catch (Exception ex) {
	67	ex.printStackTrace();
	68	log.info("计算点异常！PointNo=" + dto.getPointNo());
	69	}
	70	});
	71	log.info("计算点处理结束");
	72
	73	} catch (Exception ex) {
	74	ex.printStackTrace();
	75	log.info("计算点处理异常！");
	76	}
	77	return result;
	78	}
	79
	80	private Object singleCompute(DaPointDTO dto, Map<String, Object> dataMap) {
	81	String expression = dto.getExpression();
	82	String[] arr = expression.split(regex);
	83
	84	for (int i = 0; i < arr.length; i++) {
	85	String s = arr[i];
	86	if (StringUtils.isNotBlank(s) && dataMap.containsKey(s)) {
	87	expression = expression.replace(s, dataMap.get(s).toString());
	88	}
	89	}
	90	expression = expression.replace("&", "&&");
	91	expression = expression.replace("\|", "\|\|");
	92	expression = expression.replace("False", "false");
	93	expression = expression.replace("True", "true");
	94	log.info("PointNo=" + dto.getPointNo() + ";expression=" + expression);
	95	String result = javaScriptHandler.eval(expression);
	96	log.info("result=" + result);
	97	if (result == null) {
781e72	98	return CommonConstant.BAD_VALUE;
a6de49	99	} else if (result.contains(JsErrorCode.Infinity.name()) \|\|
H	100	result.contains(JsErrorCode.NaN.name())) {
	101	log.info("计算异常，使用默认值");
	102	return dto.getDefaultValue() == null ? BigDecimal.ZERO : dto.getDefaultValue();
	103	} else {
	104	if (DataTypeEnum.INT.getCode().equals(dto.getDataType())) {
	105	return new BigDecimal(result).intValue();
	106	} else if (DataTypeEnum.FLOAT.getCode().equals(dto.getDataType())) {
781e72	107	return new BigDecimal(result).setScale(4, BigDecimal.ROUND_UP).doubleValue();
a6de49	108	} else if (DataTypeEnum.BOOLEAN.getCode().equals(dto.getDataType())) {
H	109	return Boolean.parseBoolean(result);
	110	}
	111	}
	112	return result;
	113	}
	114
	115	public Map<String, Object> getCurrent(List<String> pointNos) {
	116	Map<String, Object> data = new HashMap<>();
	117	List<DaPointDTO> pointMathList = daPointService.getMathPoint(pointNos);
	118	if (CollectionUtils.isEmpty(pointMathList)) {
	119	return data;
	120	}
	121	pointMathList.forEach(item -> {
781e72	122	Object value = CommonConstant.BAD_VALUE;
潘	123	if (redisTemplate.hasKey(PointCollector.PV + item.getPointNo())) {
	124	value = redisTemplate.opsForValue().get(PointCollector.PV + item.getPointNo());
	125	} else {
	126	value = singleCompute(item);
	127	}
	128	data.put(item.getPointNo(), value);
a6de49	129	});
H	130	return data;
	131	}
	132
	133	private Object singleCompute(DaPointDTO dto) {
781e72	134	String result = CommonConstant.BAD_VALUE.toString();
a6de49	135	Map<String, Object> dataMap = new HashMap<>();
H	136	String expression = dto.getExpression();
	137	String[] arr = expression.split(regex);
	138	for (int i = 0; i < arr.length; i++) {
	139	String s = arr[i];
	140	if (StringUtils.isBlank(s)) {
	141	continue;
	142	}
	143	List<String> pointNos = new ArrayList<>();
	144	pointNos.add(s);
	145	dataMap.putAll(measureHandle.getCurrent(pointNos));
	146	dataMap.putAll(constantHandle.getCurrent(pointNos));
140065	147	String valueStr = dataMap.get(s).toString();
潘	148	if (StrUtils.isNumeric(valueStr) && new BigDecimal(valueStr).compareTo(CommonConstant.BAD_VALUE) == 0) {
	149	log.info("BAD_VALUE:" + s);
	150	}
	151	if (StrUtils.isNumeric(valueStr) && new BigDecimal(valueStr).compareTo(BigDecimal.ZERO) < 0) {
	152	valueStr = "(" + valueStr + ")";
	153	}
	154	expression = expression.replace(s, valueStr);
a6de49	155	}
H	156	expression = expression.replace("&", "&&");
	157	expression = expression.replace("\|", "\|\|");
	158	expression = expression.replace("False", "false");
	159	expression = expression.replace("True", "true");
	160	log.info("PointNo=" + dto.getPointNo() + ";expression=" + expression);
781e72	161	result = javaScriptHandler.eval(expression);
a6de49	162	log.info("result=" + result);
H	163	if (result == null) {
781e72	164	return CommonConstant.BAD_VALUE;
潘	165	} else if (result.contains(JsErrorCode.Infinity.name()) \|\| result.contains(JsErrorCode.NaN.name())) {
a6de49	166	log.info("计算异常，使用默认值");
H	167	return dto.getDefaultValue() == null ? BigDecimal.ZERO : dto.getDefaultValue();
	168	} else {
	169	if (DataTypeEnum.INT.getCode().equals(dto.getDataType())) {
	170	return new BigDecimal(result).intValue();
	171	} else if (DataTypeEnum.FLOAT.getCode().equals(dto.getDataType())) {
781e72	172	return new BigDecimal(result).setScale(2, BigDecimal.ROUND_UP).doubleValue();
a6de49	173	} else if (DataTypeEnum.BOOLEAN.getCode().equals(dto.getDataType())) {
H	174	return Boolean.parseBoolean(result);
	175	}
	176	}
	177	return result;
	178	}
	179	}