From c0b8cf32504dd1a42780bb1ee06ae8a7b0d7b120 Mon Sep 17 00:00:00 2001
From: 潘志宝 <979469083@qq.com>
Date: 星期三, 09 十月 2024 23:36:11 +0800
Subject: [PATCH] 指标数据
---
iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/handler/CalItemHandler.java | 178 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 178 insertions(+), 0 deletions(-)
diff --git a/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/handler/CalItemHandler.java b/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/handler/CalItemHandler.java
new file mode 100644
index 0000000..5775d02
--- /dev/null
+++ b/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/handler/CalItemHandler.java
@@ -0,0 +1,178 @@
+package com.iailab.module.data.ind.collection.handler;
+
+import com.alibaba.cloud.commons.lang.StringUtils;
+import com.iailab.framework.common.constant.CommonConstant;
+import com.iailab.module.data.common.enums.ItemTypeEnum;
+import com.iailab.module.data.common.enums.JsErrorCode;
+import com.iailab.module.data.common.utils.JavaScriptHandler;
+import com.iailab.module.data.ind.item.entity.IndItemEntity;
+import com.iailab.module.data.ind.item.service.IndItemCalService;
+import com.iailab.module.data.ind.item.service.IndItemService;
+import com.iailab.module.data.ind.item.vo.IndItemCalVO;
+import com.iailab.module.data.ind.item.vo.IndItemValueVO;
+import com.iailab.module.data.ind.value.service.IndItemValueService;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.math.BigDecimal;
+import java.util.*;
+
+/**
+ * @author PanZhibao
+ * @Description
+ * @createTime 2024年10月04日
+ */
+@Slf4j
+@Component
+public class CalItemHandler {
+
+ @Autowired
+ private JavaScriptHandler javaScriptHandler;
+
+ public static final String regex = "[+\\-\\*\\/\\(\\)\\&\\|\\>\\<]";
+
+ @Autowired
+ private IndItemCalService indItemCalService;
+
+ @Autowired
+ private AtomItemHandler atomItemHandler;
+
+ @Autowired
+ private DerItemHandler derItemHandler;
+
+ @Autowired
+ private IndItemService indItemService;
+
+ public List<IndItemValueVO> queryValue(String itemId) {
+ List<IndItemValueVO> result = new ArrayList<IndItemValueVO>();
+ try {
+ IndItemCalVO indItem = indItemCalService.getByItemId(itemId);
+ String expression = indItem.getExpression();
+ if (StringUtils.isBlank(expression)) {
+ return result;
+ }
+ String[] arr = expression.split(regex);
+ int dataLength = 0;
+ List<List<IndItemValueVO>> valueMix = new ArrayList<>();
+ for (int i = 0; i < arr.length; i++) {
+ String s = arr[i];
+ if (StringUtils.isBlank(s)) {
+ continue;
+ }
+ IndItemEntity subItem = indItemService.getInfoByNo(s);
+ ItemTypeEnum itemType = ItemTypeEnum.getEumByCode(subItem.getItemType());
+ switch (itemType) {
+ case ATOM:
+ valueMix.add(atomItemHandler.queryValue(subItem.getId()));
+ break;
+ case DER:
+ valueMix.add(derItemHandler.queryValue(subItem.getId()));
+ break;
+ default:
+ break;
+ }
+ }
+ dataLength = valueMix.get(0).size();
+
+ for (int i = 0; i < dataLength; i++) {
+ Map<String, IndItemValueVO> dataMap = new HashMap<>();
+ for (int j = 0; j < valueMix.size(); j++) {
+ IndItemValueVO valueDTO = valueMix.get(j).get(i);
+ dataMap.put(valueDTO.getItemNo(), valueDTO);
+ }
+ IndItemValueVO itemValue = this.singleCompute(indItem, dataMap);
+ result.add(itemValue);
+ }
+
+ } catch (Exception ex) {
+ ex.printStackTrace();
+ }
+ return result;
+ }
+
+ public List<IndItemValueVO> queryValue(String itemId, Date startTime, Date endTime) {
+ List<IndItemValueVO> result = new ArrayList<IndItemValueVO>();
+ try {
+ IndItemCalVO indItem = indItemCalService.getByItemId(itemId);
+ String expression = indItem.getExpression();
+ if (StringUtils.isBlank(expression)) {
+ return result;
+ }
+ String[] arr = expression.split(regex);
+ int dataLength = 0;
+ List<List<IndItemValueVO>> valueMix = new ArrayList<>();
+ for (int i = 0; i < arr.length; i++) {
+ String s = arr[i];
+ if (StringUtils.isBlank(s)) {
+ continue;
+ }
+ IndItemEntity subItem = indItemService.getInfoByNo(s);
+ ItemTypeEnum itemType = ItemTypeEnum.getEumByCode(subItem.getItemType());
+ switch (itemType) {
+ case ATOM:
+ valueMix.add(atomItemHandler.queryValue(subItem.getId()));
+ break;
+ case DER:
+ valueMix.add(derItemHandler.queryValue(subItem.getId(), startTime, endTime));
+ break;
+ default:
+ break;
+ }
+ }
+ dataLength = valueMix.get(0).size();
+
+ for (int i = 0; i < dataLength; i++) {
+ Map<String, IndItemValueVO> dataMap = new HashMap<>();
+ for (int j = 0; j < valueMix.size(); j++) {
+ IndItemValueVO valueDTO = valueMix.get(j).get(i);
+ dataMap.put(valueDTO.getItemNo(), valueDTO);
+ }
+ IndItemValueVO itemValue = this.singleCompute(indItem, dataMap);
+ result.add(itemValue);
+ }
+
+ } catch (Exception ex) {
+ ex.printStackTrace();
+ }
+ return result;
+ }
+
+ private IndItemValueVO singleCompute(IndItemCalVO dto, Map<String, IndItemValueVO> dataMap) {
+ IndItemValueVO resultDto = new IndItemValueVO();
+ String dataTime = "";
+ BigDecimal dataValue = CommonConstant.ZERO_VALUE;
+ String expression = dto.getExpression();
+ String[] arr = expression.split(regex);
+
+ for (int i = 0; i < arr.length; i++) {
+ String s = arr[i];
+ if (StringUtils.isNotBlank(s) && dataMap.containsKey(s)) {
+ dataTime = dataMap.get(s).getDataTime();
+ if (dataMap.get(s) == null || dataMap.get(s).getDataValue() == null) {
+ resultDto.setDataTime(dataTime);
+ return resultDto;
+ }
+ expression = expression.replace(s, dataMap.get(s).getDataValue().toString());
+ }
+ }
+ expression = expression.replace("&", "&&");
+ expression = expression.replace("|", "||");
+ expression = expression.replace("False", "false");
+ expression = expression.replace("True", "true");
+ log.info("ItemId=" + dto.getItemId() + ";expression=" + expression);
+ String result = javaScriptHandler.eval(expression);
+ log.info("result=" + result);
+ if (result == null) {
+ return null;
+ } else if (result.contains(JsErrorCode.Infinity.name()) ||
+ result.contains(JsErrorCode.NaN.name())) {
+ log.info("计算异常，使用默认值");
+ } else {
+ dataValue = new BigDecimal(result);
+ }
+ resultDto.setDataTime(dataTime);
+ resultDto.setDataValue(dataValue);
+ return resultDto;
+ }
+}
\ No newline at end of file
--
Gitblit v1.9.3