From 7f0bcd00c556889ba890e5e68c681f1b5d4267e1 Mon Sep 17 00:00:00 2001
From: dengzedong <dengzedong@email>
Date: 星期四, 19 十二月 2024 18:07:20 +0800
Subject: [PATCH] 预测项状态添加模型结果异常

---
 iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/point/service/impl/DaPointServiceImpl.java |  331 ++++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 279 insertions(+), 52 deletions(-)

diff --git a/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/point/service/impl/DaPointServiceImpl.java b/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/point/service/impl/DaPointServiceImpl.java
index 44402c3..b9742f0 100644
--- a/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/point/service/impl/DaPointServiceImpl.java
+++ b/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/point/service/impl/DaPointServiceImpl.java
@@ -1,30 +1,43 @@
 package com.iailab.module.data.point.service.impl;
 
+import cn.hutool.core.collection.CollUtil;
 import com.alibaba.fastjson.JSONArray;
 import com.baomidou.dynamic.datasource.annotation.DSTransactional;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.iailab.framework.common.pojo.PageResult;
+import com.iailab.framework.common.util.object.BeanUtils;
 import com.iailab.framework.common.util.object.ConvertUtils;
+import com.iailab.module.data.api.point.dto.ApiPointDTO;
+import com.iailab.module.data.channel.common.service.ChannelSourceService;
 import com.iailab.module.data.common.enums.CommonConstant;
 import com.iailab.module.data.common.enums.IsEnableEnum;
-import com.iailab.module.data.point.common.IncreaseCodeEnum;
+import com.iailab.module.data.common.enums.IncreaseCodeEnum;
 import com.iailab.module.data.point.common.PointTypeEnum;
 import com.iailab.module.data.point.dao.DaPointDao;
+import com.iailab.module.data.point.dto.DaCumulatePointDTO;
+import com.iailab.module.data.point.dto.DaMathPointDTO;
 import com.iailab.module.data.point.dto.DaMeasurePointDTO;
 import com.iailab.module.data.point.dto.DaPointDTO;
+import com.iailab.module.data.point.entity.DaMeasurePointEntity;
 import com.iailab.module.data.point.entity.DaPointEntity;
-import com.iailab.module.data.point.service.DaMathPointService;
-import com.iailab.module.data.point.service.DaMeasurePointService;
-import com.iailab.module.data.point.service.DaPointService;
-import com.iailab.module.data.point.service.DaSequenceNumService;
+import com.iailab.module.data.point.service.*;
 import com.iailab.module.data.point.vo.DaPointPageReqVO;
-import org.apache.commons.lang3.StringUtils;
+import com.iailab.module.data.point.vo.PointImportExcelVO;
+import com.iailab.module.data.point.vo.PointImportRespVO;
+import org.apache.commons.lang3.ObjectUtils;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
 import javax.annotation.Resource;
 import java.util.*;
+import java.util.concurrent.ConcurrentHashMap;
+
+import static com.iailab.framework.common.exception.util.ServiceExceptionUtil.exception;
+import static com.iailab.module.data.enums.ErrorCodeConstants.POINT_EXISTS;
+import static com.iailab.module.data.enums.ErrorCodeConstants.POINT_IMPORT_LIST_IS_EMPTY;
 
 /**
  * @author lirm
@@ -34,57 +47,109 @@
 @Service
 public class DaPointServiceImpl extends ServiceImpl<DaPointDao, DaPointEntity> implements DaPointService {
 
-    @Resource
+    @Autowired
     private DaMeasurePointService daMeasurePointService;
 
-    @Resource
+    @Autowired
     private DaMathPointService daMathPointService;
 
-    @Resource
+    @Autowired
+    private DaCumulatePointService daCumulatePointService;
+
+    @Autowired
     private DaSequenceNumService daSequenceNumService;
     
     @Resource
     private DaPointDao daPointDao;
 
+    @Resource
+    private ChannelSourceService channelSourceService;
+
+    @Resource
+    private DaPointCollectStatusService daPointCollectStatusService;
+
+    private static Map<String, ApiPointDTO> pointIdMap = new ConcurrentHashMap<>();
+
+    private static Map<String, DaPointDTO> pointNoMap = new ConcurrentHashMap<>();
 
     @Override
-    public PageResult<DaPointEntity> queryPage(DaPointPageReqVO reqVO) {
-        return daPointDao.selectPage(reqVO);
+    public PageResult<DaPointDTO> queryPage(DaPointPageReqVO reqVO) {
+        IPage<DaPointDTO> page = daPointDao.selectPageList(reqVO);
+        return new PageResult<>(page.getRecords(), page.getTotal());
+    }
+
+    private void clearCache() {
+        pointIdMap.clear();
+        pointNoMap.clear();
     }
 
     @Override
     public DaPointDTO info(String id) {
         DaPointEntity entity = daPointDao.selectById(id);
         DaPointDTO result = ConvertUtils.sourceToTarget(entity, DaPointDTO.class);
-        if (PointTypeEnum.MEASURE_POINT.getCode().equals(result.getPointType())) {
-            DaMeasurePointDTO measurePoint = daMeasurePointService.getByPoint(id);
-            result.setMeasurePoint(measurePoint);
-            List<String> sourceOption = new ArrayList<>();
-            sourceOption.add(measurePoint.getSourceType());
-            sourceOption.add(measurePoint.getSourceId());
-            sourceOption.add(measurePoint.getTagNo());
-            result.setSourceOption(sourceOption);
-        } else if (PointTypeEnum.CALCULATE_POINT.getCode().equals(result.getPointType())) {
-            result.setMathPoint(daMathPointService.getByPoint(id));
+        result.setMeasurePoint(new DaMeasurePointDTO());
+        result.setMathPoint(new DaMathPointDTO());
+        result.setCumulatePoint(new DaCumulatePointDTO());
+        switch (PointTypeEnum.getEumByCode(result.getPointType())) {
+            case MEASURE_POINT:
+                DaMeasurePointDTO measurePoint = daMeasurePointService.getByPoint(id);
+                result.setMeasurePoint(measurePoint);
+                List<String> sourceOption = new ArrayList<>();
+                sourceOption.add(measurePoint.getSourceType());
+                sourceOption.add(measurePoint.getSourceId());
+                sourceOption.add(measurePoint.getTagNo());
+                result.setSourceOption(sourceOption);
+                break;
+            case CALCULATE_POINT:
+                result.setMathPoint(daMathPointService.getByPoint(id));
+                break;
+            case CUMULATE:
+                result.setCumulatePoint(daCumulatePointService.getByPoint(id));
+                break;
+            default:
+                break;
         }
         return result;
     }
 
     @Override
+    public ApiPointDTO getSimpleInfoById(String id) {
+        if (pointIdMap.containsKey(id)) {
+            return pointIdMap.get(id);
+        }
+        ApiPointDTO dto = ConvertUtils.sourceToTarget(daPointDao.selectById(id), ApiPointDTO.class);
+        if (dto == null) {
+            return null;
+        }
+        pointIdMap.put(id, dto);
+        return dto;
+    }
+
+    @Override
+    public DaPointDTO getSimpleInfoByNo(String no) {
+        QueryWrapper<DaPointEntity> queryWrapper = new QueryWrapper<>();
+        queryWrapper.eq("point_no", no);
+        return ConvertUtils.sourceToTarget(daPointDao.selectOne(queryWrapper), DaPointDTO.class);
+    }
+
+    @Override
     public List<DaPointDTO> list(Map<String, Object> params) {
-        String page = (String) params.get("page");
-        String limit = (String) params.get("limit");
-        String pointType = (String)params.get("pointType");
+        Object pointType = params.get("pointType");
         List<String> pointNos = new ArrayList<>();
         if (params.get("pointNos") != null) {
             pointNos = JSONArray.parseArray(JSONArray.toJSONString(params.get("pointNos")), String.class);
         }
-        String pointNoLike = (String)params.get("pointNoLike");
+        List<String> pointTypes = new ArrayList<>();
+        if (params.get("pointTypes") != null) {
+            pointTypes = Arrays.asList(params.get("pointTypes").toString().split(","));
+        }
+
+        Object pointNoLike = params.get("pointNoLike");
         QueryWrapper<DaPointEntity> queryWrapper = new QueryWrapper();
-        queryWrapper.eq(StringUtils.isNotBlank(pointType), "point_type", pointType)
-        .in(!CollectionUtils.isEmpty(pointNos),"point_no", pointNos)
-        .like(StringUtils.isNotBlank(pointNoLike), "point_no", pointNoLike)
-                .last("limit 1, 10");
+        queryWrapper.eq(!ObjectUtils.isEmpty(pointType), "point_type", pointType);
+        queryWrapper.in(pointNos.size() != 0,"point_no", pointNos);
+        queryWrapper.like(!ObjectUtils.isEmpty(pointNoLike), "point_no", pointNoLike);
+        queryWrapper.in(pointTypes.size() != 0,"point_type", pointTypes);
         List<DaPointEntity> list = daPointDao.selectList(queryWrapper);
         return ConvertUtils.sourceToTarget(list, DaPointDTO.class);
     }
@@ -94,22 +159,36 @@
     public void add(DaPointDTO dataPoint) {
         DaPointEntity daPointEntity = ConvertUtils.sourceToTarget(dataPoint, DaPointEntity.class);
         daPointEntity.setId(UUID.randomUUID().toString());
-        if (PointTypeEnum.MEASURE_POINT.getName().equals(dataPoint.getPointType())) {
-            DaMeasurePointDTO measurePoint = new DaMeasurePointDTO();
-            measurePoint.setSourceType(dataPoint.getSourceOption().get(0));
-            measurePoint.setSourceId(dataPoint.getSourceOption().get(1));
-            measurePoint.setTagNo(dataPoint.getSourceOption().get(2));
-            daMeasurePointService.add(measurePoint, daPointEntity.getId());
-            daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_M.name()));
-        } else if (PointTypeEnum.CALCULATE_POINT.getName().equals(dataPoint.getPointType())) {
-            daMathPointService.add(dataPoint.getMathPoint(), daPointEntity.getId());
-            daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_C.name()));
-        } else if (PointTypeEnum.CONSTANT.getName().equals(dataPoint.getPointType())) {
-            daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_F.name()));
+        switch (PointTypeEnum.getEumByCode(dataPoint.getPointType())) {
+            case MEASURE_POINT:
+                DaMeasurePointDTO measurePoint = new DaMeasurePointDTO();
+                measurePoint.setSourceType(dataPoint.getSourceOption().get(0));
+                measurePoint.setSourceId(dataPoint.getSourceOption().get(1));
+                measurePoint.setTagNo(dataPoint.getSourceOption().get(2));
+                daMeasurePointService.add(measurePoint, daPointEntity.getId());
+                daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_M.name()));
+                break;
+            case CALCULATE_POINT:
+                daMathPointService.add(dataPoint.getMathPoint(), daPointEntity.getId());
+                daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_C.name()));
+                break;
+            case CONSTANT:
+                daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_F.name()));
+                break;
+            case CUMULATE:
+                daCumulatePointService.add(dataPoint.getCumulatePoint(), daPointEntity.getId());
+                daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_L.name()));
+                break;
+            default:
+                break;
         }
         daPointEntity.setIsEnable(CommonConstant.IS_ENABLE);
         daPointEntity.setCreateTime(new Date());
         daPointDao.insert(daPointEntity);
+
+        daPointCollectStatusService.add(daPointEntity.getPointNo());
+        // 清空缓存
+        clearCache();
     }
 
     @Override
@@ -118,15 +197,25 @@
         DaPointEntity daPointEntity = ConvertUtils.sourceToTarget(dataPoint, DaPointEntity.class);
         daPointEntity.setUpdateTime(new Date());
         daPointDao.updateById(daPointEntity);
-        if (PointTypeEnum.MEASURE_POINT.getName().equals(dataPoint.getPointType())) {
-            DaMeasurePointDTO measurePoint = dataPoint.getMeasurePoint();
-            measurePoint.setSourceType(dataPoint.getSourceOption().get(0));
-            measurePoint.setSourceId(dataPoint.getSourceOption().get(1));
-            measurePoint.setTagNo(dataPoint.getSourceOption().get(2));
-            daMeasurePointService.update(measurePoint);
-        } else if (PointTypeEnum.CALCULATE_POINT.equals(dataPoint.getPointType())) {
-            daMathPointService.update(dataPoint.getMathPoint());
+        switch (PointTypeEnum.getEumByCode(dataPoint.getPointType())) {
+            case MEASURE_POINT:
+                DaMeasurePointDTO measurePoint = dataPoint.getMeasurePoint();
+                measurePoint.setSourceType(dataPoint.getSourceOption().get(0));
+                measurePoint.setSourceId(dataPoint.getSourceOption().get(1));
+                measurePoint.setTagNo(dataPoint.getSourceOption().get(2));
+                daMeasurePointService.update(measurePoint);
+                break;
+            case CALCULATE_POINT:
+                daMathPointService.update(dataPoint.getMathPoint());
+                break;
+            case CUMULATE:
+                daCumulatePointService.update(dataPoint.getCumulatePoint());
+                break;
+            default:
+                break;
         }
+        // 清空缓存
+        clearCache();
     }
 
     @Override
@@ -135,6 +224,18 @@
         daPointDao.deleteBatchIds(Arrays.asList(ids));
         daMeasurePointService.deleteByPoint(ids);
         daMathPointService.deleteByPoint(ids);
+        daCumulatePointService.deleteByPoint(ids);
+        // 清空缓存
+        clearCache();
+    }
+
+    @Override
+    public List<DaPointDTO> getConstantPoint(DaPointPageReqVO reqVO) {
+        Map<String, Object> params = new HashMap<>();
+        params.put("pointType", PointTypeEnum.CONSTANT.getCode());
+        params.put("pointNo", reqVO.getPointNo());
+        params.put("pointName", reqVO.getPointName());
+        return daPointDao.getConstantPoint(params);
     }
 
     @Override
@@ -153,6 +254,16 @@
         params.put("isEnable", CommonConstant.IS_ENABLE);
         params.put("pointNos", pointNos);
         return daPointDao.getConstantPoint(params);
+    }
+
+    @Override
+    public List<DaPointDTO> getMeasurePoint(DaPointPageReqVO reqVO) {
+        Map<String, Object> params = new HashMap<>();
+        params.put("pointType", PointTypeEnum.MEASURE_POINT.getCode());
+        params.put("pointNo", reqVO.getPointNo());
+        params.put("pointName", reqVO.getPointName());
+        params.put("sourceName", reqVO.getSourceName());
+        return daPointDao.getMeasurePoint(params);
     }
 
     @Override
@@ -186,6 +297,15 @@
     }
 
     @Override
+    public List<DaPointDTO> getMathPoint(DaPointPageReqVO reqVO) {
+        Map<String, Object> params = new HashMap<>();
+        params.put("pointType", PointTypeEnum.CALCULATE_POINT.getCode());
+        params.put("pointNo", reqVO.getPointNo());
+        params.put("pointName", reqVO.getPointName());
+        return daPointDao.getMathPoint(params);
+    }
+
+    @Override
     public List<DaPointDTO> getMathPoint(String freq) {
         Map<String, Object> params = new HashMap<>();
         params.put("pointType", PointTypeEnum.CALCULATE_POINT.getCode());
@@ -204,11 +324,34 @@
     }
 
     @Override
+    public List<DaPointDTO> getCumulatePoint(String freq) {
+        Map<String, Object> params = new HashMap<>(3);
+        params.put("pointType", PointTypeEnum.CUMULATE.getCode());
+        params.put("isEnable", CommonConstant.IS_ENABLE);
+        params.put("minfreqid", freq);
+        return daPointDao.getCumulatePoint(params);
+    }
+
+    @Override
+    public List<DaPointDTO> getCumulatePoint(DaPointPageReqVO reqVO) {
+        Map<String, Object> params = new HashMap<>(3);
+        params.put("pointType", PointTypeEnum.CUMULATE.getCode());
+        params.put("pointNo", reqVO.getPointNo());
+        params.put("pointName", reqVO.getPointName());
+        return daPointDao.getCumulatePoint(params);
+    }
+
+    @Override
     public DaPointDTO getByNo(String pointNo) {
+        if (pointNoMap.containsKey(pointNo)) {
+            return pointNoMap.get(pointNo);
+        }
         QueryWrapper<DaPointEntity> wrapper = new QueryWrapper<>();
         wrapper.eq("point_no", pointNo);
         DaPointEntity entity = daPointDao.selectOne(wrapper);
-        return ConvertUtils.sourceToTarget(entity, DaPointDTO.class);
+        DaPointDTO dto = ConvertUtils.sourceToTarget(entity, DaPointDTO.class);
+        pointNoMap.put(pointNo, dto);
+        return dto;
     }
 
     @Override
@@ -229,6 +372,87 @@
     }
 
     @Override
+    @DSTransactional(rollbackFor = Exception.class) // 添加事务,异常则回滚所有导入
+    public PointImportRespVO importPointList(List<PointImportExcelVO> importPoints, boolean isUpdateSupport) {
+        // 1.1 参数校验
+        if (CollUtil.isEmpty(importPoints)) {
+            throw exception(POINT_IMPORT_LIST_IS_EMPTY);
+        }
+
+        Map<String, Map<String, String>> sourcesIdMap = channelSourceService.getSourcesId();
+        // 2. 遍历,逐个创建 or 更新
+        PointImportRespVO respVO = PointImportRespVO.builder().createPointnames(new ArrayList<>())
+                .updatePointnames(new ArrayList<>()).failurePointnames(new LinkedHashMap<>()).build();
+        importPoints.forEach(importPoint -> {
+
+            // 判断如果不存在,再进行插入
+            DaPointEntity existPoint = baseMapper.selectByPointName(importPoint.getPointName());
+            if (existPoint == null) {
+                DaPointEntity daPointEntity = ConvertUtils.sourceToTarget(importPoint, DaPointEntity.class);
+                daPointEntity.setId(UUID.randomUUID().toString());
+                daPointEntity.setIsEnable(CommonConstant.IS_ENABLE);
+                daPointEntity.setCreateTime(new Date());
+                switch (PointTypeEnum.getEumByCode(daPointEntity.getPointType())) {
+                    case MEASURE_POINT:
+                        DaMeasurePointDTO measurePoint = new DaMeasurePointDTO();
+                        measurePoint.setSourceType(importPoint.getSourceType());
+                        measurePoint.setSourceId(sourcesIdMap.get(importPoint.getSourceType()).get(importPoint.getSourceName()));
+                        measurePoint.setTagNo(importPoint.getTagNo());
+                        measurePoint.setValueType(importPoint.getValueType());
+                        measurePoint.setDimension(importPoint.getDimension());
+                        daMeasurePointService.add(measurePoint, daPointEntity.getId());
+                        daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_M.name()));
+                        break;
+                    case CALCULATE_POINT:
+                        daMathPointService.add(importPoint.getExpression(), daPointEntity.getId());
+                        daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_C.name()));
+                        break;
+                    case CONSTANT:
+                        daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_F.name()));
+                        break;
+                    case CUMULATE:
+                        DaCumulatePointDTO cumulatePoint = new DaCumulatePointDTO();
+                        cumulatePoint.setMomentPoint(importPoint.getMomentPoint());
+                        cumulatePoint.setLength(importPoint.getLength());
+                        cumulatePoint.setDivisor(importPoint.getDivisor());
+                        daCumulatePointService.add(cumulatePoint, daPointEntity.getId());
+                        daPointEntity.setPointNo(daSequenceNumService.getAndIncreaseByCode(IncreaseCodeEnum.POINT_L.name()));
+                    default:
+                        break;
+                }
+
+                daPointDao.insert(daPointEntity);
+                daPointCollectStatusService.add(daPointEntity.getPointNo());
+                respVO.getCreatePointnames().add(importPoint.getPointName());
+                return;
+            }
+
+            // 如果存在,判断是否允许更新
+            if (!isUpdateSupport) {
+                respVO.getFailurePointnames().put(importPoint.getPointName(), POINT_EXISTS.getMsg());
+                return;
+            }
+
+            DaPointEntity updatePoint = BeanUtils.toBean(importPoint, DaPointEntity.class);
+            updatePoint.setId(existPoint.getId());
+            baseMapper.updateById(updatePoint);
+            DaMeasurePointEntity measurePoint = new DaMeasurePointEntity();
+            measurePoint.setSourceType(importPoint.getSourceType());
+            measurePoint.setSourceId(sourcesIdMap.get(importPoint.getSourceType()).get(importPoint.getSourceName()));
+            measurePoint.setTagNo(importPoint.getTagNo());
+            daMeasurePointService.update(measurePoint, new QueryWrapper<DaMeasurePointEntity>().eq("point_id",updatePoint.getId()));
+            respVO.getUpdatePointnames().add(importPoint.getPointName());
+        });
+        return respVO;
+    }
+
+    @Override
+    public List<DaPointDTO> getList(DaPointPageReqVO exportReqVO) {
+        return daPointDao.getList(exportReqVO);
+    }
+
+    @Override
+    @DSTransactional(rollbackFor = Exception.class)
     public void enableByIds(String[] ids) {
         if (CollectionUtils.isEmpty(Arrays.asList(ids))) {
             return;
@@ -236,12 +460,14 @@
         Arrays.asList(ids).forEach(item -> {
             DaPointEntity entity = new DaPointEntity();
             entity.setId(item);
-            entity.setIsEnable(IsEnableEnum.ENABLE.value());
+            entity.setIsEnable(IsEnableEnum.ENABLE.getCode());
+            entity.setUpdateTime(new Date());
             daPointDao.updateById(entity);
         });
     }
 
     @Override
+    @DSTransactional(rollbackFor = Exception.class)
     public void disableByIds(String[] ids) {
         if (CollectionUtils.isEmpty(Arrays.asList(ids))) {
             return;
@@ -249,7 +475,8 @@
         Arrays.asList(ids).forEach(item -> {
             DaPointEntity entity = new DaPointEntity();
             entity.setId(item);
-            entity.setIsEnable(IsEnableEnum.DISABLE.value());
+            entity.setIsEnable(IsEnableEnum.DISABLE.getCode());
+            entity.setUpdateTime(new Date());
             daPointDao.updateById(entity);
         });
     }

--
Gitblit v1.9.3