From 37f8d15dd8954e04ad09445184c04a58773af17a Mon Sep 17 00:00:00 2001
From: ‘liusuyi’ <1951119284@qq.com>
Date: 星期三, 14 六月 2023 09:13:37 +0800
Subject: [PATCH] 增加websocket报警点数量推送

---
 ard-work/src/main/java/com/ruoyi/alarm/globalAlarm/service/impl/GlobalAlarmServiceImpl.java |   53 +++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 47 insertions(+), 6 deletions(-)

diff --git a/ard-work/src/main/java/com/ruoyi/alarm/globalAlarm/service/impl/GlobalAlarmServiceImpl.java b/ard-work/src/main/java/com/ruoyi/alarm/globalAlarm/service/impl/GlobalAlarmServiceImpl.java
index 55ace11..82b68b3 100644
--- a/ard-work/src/main/java/com/ruoyi/alarm/globalAlarm/service/impl/GlobalAlarmServiceImpl.java
+++ b/ard-work/src/main/java/com/ruoyi/alarm/globalAlarm/service/impl/GlobalAlarmServiceImpl.java
@@ -26,9 +26,7 @@
 
 import javax.annotation.Resource;
 import java.text.SimpleDateFormat;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
 import java.util.stream.Collectors;
 
 import static jdk.nashorn.internal.runtime.regexp.joni.Config.log;
@@ -46,6 +44,8 @@
     @Resource
     ArdAlarmStealelecMapper ardAlarmStealelecMapper;
     @Resource
+    ArdAlarmTubeMapper ardAlarmTubeMapper;
+    @Resource
     IArdAlarmTubeService ardAlarmTubeService;
     @Resource
     ArdTubesMapper ardTubesMapper;
@@ -59,7 +59,7 @@
         String refreshTime = configService.selectConfigByKey("refreshTime");
         switch (condition.getCommand()) {
             case 1001:
-                List<ArdAlarmStealelec> ardAlarmStealelecs = ardAlarmStealelecMapper.selectAllByCommandArdAlarmStealelecList(refreshTime);
+                List<ArdAlarmStealelec> ardAlarmStealelecs = ardAlarmStealelecMapper.selectListAllByCommand(refreshTime);
                 return ardAlarmStealelecs.stream()
                         .map(ardAlarmStealelec -> {
                             GlobalAlarmData globalAlarmData = new GlobalAlarmData()
@@ -74,6 +74,22 @@
                             return globalAlarmData;
                         })
                         .collect(Collectors.toList());
+            case 1014:
+                List<ArdAlarmTube> ardAlarmTubes = ardAlarmTubeMapper.selectListAllByCommand(refreshTime);
+                return ardAlarmTubes.stream()
+                        .map(ardAlarmTube -> {
+                            GlobalAlarmData globalAlarmData = new GlobalAlarmData()
+                                    .setId(ardAlarmTube.getId())
+                                    .setName(ardAlarmTube.getTubeName())
+                                    .setAlarmTime(ardAlarmTube.getAlarmTime())
+                                    .setLongitude(ardAlarmTube.getLongitude())
+                                    .setLatitude(ardAlarmTube.getLatitude())
+                                    .setAltitude(ardAlarmTube.getAltitude())
+                                    .setCount(ardAlarmTube.getCount())
+                                    .setTotal(ardAlarmTube.getTotal());
+                            return globalAlarmData;
+                        })
+                        .collect(Collectors.toList());
             default:
                 return null;
         }
@@ -84,13 +100,20 @@
         SimpleDateFormat fmt = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
         switch (condition.getCommand()) {
             case 1001:
-                String id = condition.getId();
-                ArdAlarmStealelec ardAlarmStealelec = ardAlarmStealelecMapper.selectByPrimaryKey(id);
+                ArdAlarmStealelec ardAlarmStealelec = ardAlarmStealelecMapper.selectByPrimaryKey(condition.getId());
                 if(StringUtils.isNotNull(ardAlarmStealelec)) {
                     String describe = ardAlarmStealelec.getDescribe();
                     String startTime = fmt.format(ardAlarmStealelec.getStartTime());
                     int i = ardAlarmStealelecMapper.updateViewTimeByDescribe(describe, startTime, DateUtils.getTime());
                     return ardAlarmStealelec;
+                }
+            case 1014:
+                ArdAlarmTube ardAlarmTube = ardAlarmTubeMapper.selectArdAlarmTubeById(condition.getId());
+                if(StringUtils.isNotNull(ardAlarmTube)) {
+                    String tubeId = ardAlarmTube.getTubeId();
+                    String alarmTime = fmt.format(ardAlarmTube.getAlarmTime());
+                    int i = ardAlarmTubeMapper.updateViewTimeByTubeId(tubeId, alarmTime, DateUtils.getTime());
+                    return ardAlarmTube;
                 }
             default: return null;
         }
@@ -127,4 +150,22 @@
                 break;
         }
     }
+    /**
+     * @鎻忚堪 鏌ヨ鎵�鏈夋姤璀︾殑褰撳墠鏁伴噺
+     * @鍙傛暟 []
+     * @杩斿洖鍊� java.util.Map<java.lang.String,java.lang.Integer>
+     * @鍒涘缓浜� 鍒樿嫃涔�
+     * @鍒涘缓鏃堕棿 2023/6/13 16:30
+     * @淇敼浜哄拰鍏跺畠淇℃伅
+     */
+    @Override
+    public Map<String, Integer> selectAlarmLogsCount() {
+        Map<String, Integer> countMap=new HashMap<>();
+        String refreshTime = configService.selectConfigByKey("refreshTime");
+        int count1001 = ardAlarmStealelecMapper.selectCountByStartTime(refreshTime);
+        countMap.put("1001",count1001);
+        int count1014 = ardAlarmTubeMapper.selectCountByStartTime(refreshTime);
+        countMap.put("1014 ",count1014);
+        return countMap;
+    }
 }

--
Gitblit v1.9.3