From bcecd69be3fb0d64e0a692ce328ca7b1f44dedf3 Mon Sep 17 00:00:00 2001
From: liusuyi <13324259@qq.com>
Date: 星期五, 02 六月 2023 19:08:34 +0800
Subject: [PATCH] 优化

---
 ard-work/src/main/java/com/ruoyi/inspect/service/impl/ArdVideoInspectTaskServiceImpl.java |  249 +++++++++++++++++++++++++++++++++++--------------
 1 files changed, 175 insertions(+), 74 deletions(-)

diff --git a/ard-work/src/main/java/com/ruoyi/inspect/service/impl/ArdVideoInspectTaskServiceImpl.java b/ard-work/src/main/java/com/ruoyi/inspect/service/impl/ArdVideoInspectTaskServiceImpl.java
index 85d6bb1..c35cc97 100644
--- a/ard-work/src/main/java/com/ruoyi/inspect/service/impl/ArdVideoInspectTaskServiceImpl.java
+++ b/ard-work/src/main/java/com/ruoyi/inspect/service/impl/ArdVideoInspectTaskServiceImpl.java
@@ -5,22 +5,16 @@
 
 import com.ruoyi.alarmpoints.well.domain.ArdAlarmpointsWell;
 import com.ruoyi.alarmpoints.well.mapper.ArdAlarmpointsWellMapper;
-import com.ruoyi.alarmpoints.well.service.IArdAlarmpointsWellService;
 import com.ruoyi.common.utils.DateUtils;
-
 import com.ruoyi.device.camera.domain.ArdCameras;
 import com.ruoyi.device.camera.domain.CameraCmd;
 import com.ruoyi.device.camera.mapper.ArdCamerasMapper;
-import com.ruoyi.device.camera.service.IArdCamerasService;
 import com.ruoyi.device.hiksdk.config.MinioClientSingleton;
 import com.ruoyi.device.hiksdk.service.IHikClientService;
 import com.ruoyi.inspect.domain.ArdVideoInspectRecord;
 import com.ruoyi.inspect.mapper.ArdVideoInspectRecordMapper;
-import com.ruoyi.inspect.service.IArdVideoInspectRecordService;
 import lombok.extern.slf4j.Slf4j;
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.SecurityUtils;
 import org.springframework.transaction.annotation.Transactional;
@@ -29,6 +23,7 @@
 import com.ruoyi.inspect.domain.ArdVideoInspectTask;
 import com.ruoyi.inspect.service.IArdVideoInspectTaskService;
 
+import javax.annotation.PostConstruct;
 import javax.annotation.Resource;
 
 /**
@@ -50,6 +45,18 @@
     private ArdCamerasMapper ardCamerasMapper;
     @Resource
     private IHikClientService hikClientService;
+
+    @PostConstruct
+    public void initTask() {
+        List<ArdVideoInspectTask> ardVideoInspectTasks = ardVideoInspectTaskMapper.selectArdVideoInspectTaskList(new ArdVideoInspectTask());
+        /*鍚姩鍚庡皢鎵�鏈夋墜鍔ㄤ换鍔$疆涓哄叧闂�*/
+        for (ArdVideoInspectTask ardVideoInspectTask : ardVideoInspectTasks) {
+            if (ardVideoInspectTask.getInspectMode().equals("鎵嬪姩")) {
+                ardVideoInspectTask.setMenualSwitch("鍏�");
+                ardVideoInspectTaskMapper.updateArdVideoInspectTask(ardVideoInspectTask);
+            }
+        }
+    }
 
     /**
      * 鏌ヨ瑙嗛宸℃浠诲姟
@@ -101,9 +108,22 @@
     public int updateArdVideoInspectTask(ArdVideoInspectTask ardVideoInspectTask) {
         ardVideoInspectTask.setUpdateBy(SecurityUtils.getUsername());
         ardVideoInspectTask.setUpdateTime(DateUtils.getNowDate());
-        ardVideoInspectTaskMapper.deleteArdVideoInspectTaskStepByTaskId(ardVideoInspectTask.getId())
-        ;
+        ardVideoInspectTaskMapper.deleteArdVideoInspectTaskStepByTaskId(ardVideoInspectTask.getId());
         insertArdVideoInspectTaskStep(ardVideoInspectTask);
+        return ardVideoInspectTaskMapper.updateArdVideoInspectTask(ardVideoInspectTask);
+    }
+
+    /**
+     * 淇敼瑙嗛宸℃浠诲姟,bu鏇存柊姝ラ
+     *
+     * @param ardVideoInspectTask 瑙嗛宸℃浠诲姟
+     * @return 缁撴灉
+     */
+    @Transactional
+    @Override
+    public int updateArdVideoInspectTaskSingle(ArdVideoInspectTask ardVideoInspectTask) {
+        ardVideoInspectTask.setUpdateBy(SecurityUtils.getUsername());
+        ardVideoInspectTask.setUpdateTime(DateUtils.getNowDate());
         return ardVideoInspectTaskMapper.updateArdVideoInspectTask(ardVideoInspectTask);
     }
 
@@ -158,41 +178,81 @@
         }
     }
 
-
     /**
      * 鎵嬪姩宸℃浠诲姟
      */
     @Override
     public void manualTaskRun(String TaskId) {
-        ArdVideoInspectTask videoInspectTask = ardVideoInspectTaskMapper.selectArdVideoInspectTaskById(TaskId);
-        if (videoInspectTask.getArdVideoInspectTaskStepList().size() == 0) {
-            return;
-        }
-        String currentStepId = videoInspectTask.getCurrentStepId();
-        String currentStepStartTime = videoInspectTask.getCurrentStepStartTime();
-        if (StringUtils.isNull(currentStepId)||StringUtils.isEmpty(currentStepId)) {
-            //浠�1鍙锋楠ゅ紑濮�
-            videoInspectTask.setCurrentStepId(videoInspectTask.getArdVideoInspectTaskStepList().get(0).getId());
-            if(InspectionTaskManager.taskMap.containsKey(TaskId)) {
-                startRunStep(videoInspectTask);
+        try {
+            ArdVideoInspectTask videoInspectTask = ardVideoInspectTaskMapper.selectArdVideoInspectTaskById(TaskId);
+            if (videoInspectTask.getArdVideoInspectTaskStepList().size() == 0) {
+                return;
             }
-        } else /*褰撳墠浠诲姟宸茬粡鎵ц锛屽垽鏂槸鍚﹀埌鏈�*/ {
-            if (StringUtils.isNull(currentStepStartTime)||StringUtils.isEmpty(currentStepStartTime)) {
-                //褰撳墠浠诲姟鍚姩鏃堕棿涓簄ull锛屽垯璇存槑褰撳墠姝ラ琚腑鏂紝鐩存帴鍚姩褰撳墠姝ラ
-                startRunStep(videoInspectTask);
-            } else {
-                boolean expird = isExpirdStep(videoInspectTask); /*鍒ゆ柇褰撳墠姝ラ鏃堕棿鏄惁杩囨湡*/
-                if (expird) {
-                    //鍋滄褰曞儚
-                    stopRunStep(videoInspectTask);
-                    //鍒囨崲姝ラ
-                    String nextStepId = changeNextStep(videoInspectTask);
-                    //寮�濮嬩笅涓�姝ラ
-                    videoInspectTask.setCurrentStepId(nextStepId);
+            String currentStepId = videoInspectTask.getCurrentStepId();
+            String currentStepStartTime = videoInspectTask.getCurrentStepStartTime();
+            if (StringUtils.isNull(currentStepId) || StringUtils.isEmpty(currentStepId)) {
+                //浠�1鍙锋楠ゅ紑濮�
+                videoInspectTask.setCurrentStepId(videoInspectTask.getArdVideoInspectTaskStepList().get(0).getId());
+                if (InspectionTaskManager.taskMap.containsKey(TaskId)) {
                     startRunStep(videoInspectTask);
+                }
+            } else /*褰撳墠浠诲姟宸茬粡鎵ц锛屽垽鏂槸鍚﹀埌鏈�*/ {
+                if (StringUtils.isNull(currentStepStartTime) || StringUtils.isEmpty(currentStepStartTime)) {
+                    //褰撳墠浠诲姟鍚姩鏃堕棿涓簄ull锛屽垯璇存槑褰撳墠姝ラ琚腑鏂紝鐩存帴鍚姩褰撳墠姝ラ
+                    startRunStep(videoInspectTask);
+                } else {
+                    boolean expird = isExpirdStep(videoInspectTask); /*鍒ゆ柇褰撳墠姝ラ鏃堕棿鏄惁杩囨湡*/
+                    if (expird) {
+                        //鍋滄褰曞儚
+                        stopRunStep(videoInspectTask);
+                        //鍒囨崲姝ラ
+                        String nextStepId = changeNextStep(videoInspectTask);
+                        //寮�濮嬩笅涓�姝ラ
+                        videoInspectTask.setCurrentStepId(nextStepId);
+                        startRunStep(videoInspectTask);
+                    }
+                }
+            }
+        } catch (Exception ex) {
+            log.error("鎵嬪姩宸℃寮傚父锛�" + ex.getMessage());
+        }
+    }
+
+    /**
+     * 鍒ゆ柇褰撳墠鎵嬪姩浠诲姟鑳藉惁寮�鍚�
+     */
+    @Override
+    public boolean isEnablemanualTask(String TaskId) {
+        /*鎵弿褰撳墠鎵嬪姩寮�鐨勪换鍔★紝鑾峰彇鎵嬪姩鎺у埗鐨勭浉鏈篿dList*/
+        List<String> menualCameraId = new ArrayList<>();
+        ArdVideoInspectTask menualTask = new ArdVideoInspectTask();
+        menualTask.setInspectMode("鎵嬪姩");
+        menualTask.setMenualSwitch("寮�");
+        List<ArdVideoInspectTask> menualTasks = ardVideoInspectTaskMapper.selectArdVideoInspectTaskList(menualTask);
+        if (menualTasks.size() > 0) {
+            for (ArdVideoInspectTask task : menualTasks) {
+                if (StringUtils.isNull(task.getCameraId()) || StringUtils.isEmpty(task.getCameraId())) {
+                    continue;
+                }
+                if (!menualCameraId.contains(task.getCameraId())) {
+                    menualCameraId.add(task.getCameraId());
                 }
             }
         }
+        /*鍒ゆ柇鐩告満鏄惁宸茬粡琚汉鎵嬪姩鎺у埗*/
+        ArdVideoInspectTask videoInspectTask = ardVideoInspectTaskMapper.selectArdVideoInspectTaskById(TaskId);
+        if (StringUtils.isNull(videoInspectTask)) {
+            return false;
+        }
+        String cameraId = videoInspectTask.getCameraId();
+        if (StringUtils.isEmpty(cameraId)) {
+            return false;
+        }
+        if (menualCameraId.contains(cameraId)) {
+            /*濡傛灉褰撳墠浠诲姟鍏宠仈鐨勭浉鏈哄凡缁忔湁浜烘墜鍔ㄦ帶鍒讹紝鍒欎笉鍏佽鎺у埗*/
+            return false;
+        }
+        return true;
     }
 
     /**
@@ -200,49 +260,86 @@
      */
     @Override
     public void autoTaskRun() {
-        /*鎵弿鎵�鏈夊彲鎵ц浠诲姟1-鏃堕棿婊¤冻2-鑷姩*/
-        ArdVideoInspectTask ardVideoInspectTask = new ArdVideoInspectTask();
-        ardVideoInspectTask.setInspectMode("鑷姩");
-        List<ArdVideoInspectTask> ardVideoInspectTasks = ardVideoInspectTaskMapper.selectArdVideoInspectTaskList(ardVideoInspectTask);
-        for (ArdVideoInspectTask videoInspectTask : ardVideoInspectTasks) {
-            /*閬嶅巻鎵�鏈夋椂闂存弧瓒崇殑鑷姩浠诲姟*/
-            boolean timeCompare = DateUtils.TimeCompare(videoInspectTask.getStartTime(), videoInspectTask.getEndTime());
-            if (timeCompare) {
-                log.info("鑷姩浠诲姟鍚姩锛�" + videoInspectTask.getId());
-                /*鑾峰彇褰撳墠浠诲姟鐨勬墍鏈夊贰妫�姝ラ*/
-                List<ArdVideoInspectTaskStep> ardVideoInspectTaskStepList = ardVideoInspectTaskMapper.selectArdVideoInspectTaskById(videoInspectTask.getId()).getArdVideoInspectTaskStepList();
-                if (ardVideoInspectTaskStepList.size() == 0) {
-                    continue;
-                }
-                videoInspectTask.setArdVideoInspectTaskStepList(ardVideoInspectTaskStepList);
-                String currentStepId = videoInspectTask.getCurrentStepId();
-                if (StringUtils.isNull(currentStepId)) {
-                    videoInspectTask.setCurrentStepId(ardVideoInspectTaskStepList.get(0).getId());
-                    startRunStep(videoInspectTask);//寮�濮嬪綋鍓嶄换鍔$殑绗竴涓楠�
-                } else /*褰撳墠浠诲姟宸茬粡鎵ц锛屽垽鏂槸鍚﹀埌鏈�*/ {
-                    if (StringUtils.isNull(videoInspectTask.getCurrentStepStartTime())) {
-                        //褰撳墠浠诲姟鍚姩鏃堕棿涓簄ull锛屽垯璇存槑褰撳墠姝ラ琚腑鏂紝鐩存帴鍚姩褰撳墠姝ラ
-                        startRunStep(videoInspectTask);
-                    } else {
-                        boolean expird = isExpirdStep(videoInspectTask); /*鍒ゆ柇褰撳墠姝ラ鏃堕棿鏄惁杩囨湡*/
-                        if (expird) {
-                            //鍋滄褰曞儚
-                            stopRunStep(videoInspectTask);
-                            //鍒囨崲姝ラ
-                            String nextStepId = changeNextStep(videoInspectTask);
-                            //寮�濮嬩笅涓�姝ラ
-                            videoInspectTask.setCurrentStepId(nextStepId);
-                            startRunStep(videoInspectTask);
-                        }
+        try {
+            /*鎵弿褰撳墠鎵嬪姩寮�鐨勪换鍔★紝鑾峰彇鎵嬪姩鎺у埗鐨勭浉鏈篿dList*/
+            List<String> menualCameraId = new ArrayList<>();
+            ArdVideoInspectTask menualTask = new ArdVideoInspectTask();
+            menualTask.setInspectMode("鎵嬪姩");
+            menualTask.setMenualSwitch("寮�");
+            List<ArdVideoInspectTask> menualTasks = ardVideoInspectTaskMapper.selectArdVideoInspectTaskList(menualTask);
+            if (menualTasks.size() > 0) {
+                for (ArdVideoInspectTask task : menualTasks) {
+                    if (StringUtils.isNull(task.getCameraId()) || StringUtils.isEmpty(task.getCameraId())) {
+                        continue;
+                    }
+                    if (!menualCameraId.contains(task.getCameraId())) {
+                        menualCameraId.add(task.getCameraId());
                     }
                 }
             }
+
+            /*鎵弿鎵�鏈夎嚜鍔ㄤ换鍔�*/
+            ArdVideoInspectTask autoTask = new ArdVideoInspectTask();
+            autoTask.setInspectMode("鑷姩");
+            List<ArdVideoInspectTask> ardVideoInspectTasks = ardVideoInspectTaskMapper.selectArdVideoInspectTaskList(autoTask);
+            for (ArdVideoInspectTask videoInspectTask : ardVideoInspectTasks) {
+                /*閬嶅巻鎵�鏈夋椂闂存弧瓒崇殑鑷姩浠诲姟*/
+                boolean timeCompare = DateUtils.TimeCompare(videoInspectTask.getStartTime(), videoInspectTask.getEndTime());
+                if (timeCompare) {
+                    if (menualCameraId.contains(videoInspectTask.getCameraId())) {
+                        /*鑻ュ綋鍓嶇浉鏈烘鍦ㄦ湁浠诲姟鎵嬪姩鎵ц锛屽垯涓嶆墽琛�*/
+                        continue;
+                    }
+                    /*鑾峰彇褰撳墠浠诲姟鐨勬墍鏈夊贰妫�姝ラ*/
+                    List<ArdVideoInspectTaskStep> ardVideoInspectTaskStepList = ardVideoInspectTaskMapper.selectArdVideoInspectTaskById(videoInspectTask.getId()).getArdVideoInspectTaskStepList();
+                    if (ardVideoInspectTaskStepList.size() == 0) {
+                        continue;
+                    }
+                    videoInspectTask.setArdVideoInspectTaskStepList(ardVideoInspectTaskStepList);
+                    String currentStepId = videoInspectTask.getCurrentStepId();
+                    String currentStepStartTime = videoInspectTask.getCurrentStepStartTime();
+                    if (StringUtils.isNull(currentStepId) || StringUtils.isEmpty(currentStepId)) {
+                        //寮�濮嬪綋鍓嶄换鍔$殑绗竴涓楠�
+                        videoInspectTask.setCurrentStepId(ardVideoInspectTaskStepList.get(0).getId());
+                        startRunStep(videoInspectTask);
+                    } else /*褰撳墠浠诲姟宸茬粡鎵ц锛屽垽鏂槸鍚﹀埌鏈�*/ {
+                        if (StringUtils.isNull(currentStepStartTime) || StringUtils.isEmpty(currentStepStartTime)) {
+                            //褰撳墠浠诲姟鍚姩鏃堕棿涓虹┖锛屽垯璇存槑褰撳墠姝ラ琚腑鏂紝鐩存帴鍚姩褰撳墠姝ラ
+                            startRunStep(videoInspectTask);
+                        } else {
+                            //鍒ゆ柇褰撳墠姝ラ鏃堕棿鏄惁杩囨湡
+                            boolean expird = isExpirdStep(videoInspectTask);
+                            if (expird) {
+                                //鍋滄褰曞儚
+                                stopRunStep(videoInspectTask);
+                                //鍒囨崲姝ラ
+                                String nextStepId = changeNextStep(videoInspectTask);
+                                //寮�濮嬩笅涓�姝ラ
+                                videoInspectTask.setCurrentStepId(nextStepId);
+                                startRunStep(videoInspectTask);
+                            }
+                        }
+                    }
+                } else {
+                    if (StringUtils.isEmpty(videoInspectTask.getCurrentStepId()) || StringUtils.isEmpty(videoInspectTask.getCurrentStepStartTime())) {
+                        continue;
+                    }
+                    /*杩囨湡鐨勪换鍔℃竻绌哄綋鍓嶆楠や俊鎭�*/
+                    videoInspectTask.setCurrentStepId("");
+                    videoInspectTask.setCurrentStepStartTime("");
+                    ardVideoInspectTaskMapper.updateArdVideoInspectTask(videoInspectTask);
+                }
+            }
+        } catch (Exception ex) {
+            log.error("鑷姩宸℃浠诲姟寮傚父锛�" + ex.getMessage());
         }
     }
 
+    /**
+     * 鑷姩鎵嬪姩鍚堝苟锛屾壂搴撴柟寮忥紝鐩墠涓嶄娇鐢�
+     */
     @Override
     public void taskRun() {
-        /*鎵弿鎵�鏈夊彲鎵ц浠诲姟1-鏃堕棿婊¤冻2-鑷姩*/
         ArdVideoInspectTask ardVideoInspectTask = new ArdVideoInspectTask();
         List<ArdVideoInspectTask> ardVideoInspectTasks = ardVideoInspectTaskMapper.selectArdVideoInspectTaskList(ardVideoInspectTask);
         for (ArdVideoInspectTask videoInspectTask : ardVideoInspectTasks) {
@@ -262,13 +359,15 @@
                 continue;
             }
             videoInspectTask.setArdVideoInspectTaskStepList(ardVideoInspectTaskStepList);
-            String currentStepId = videoInspectTask.getCurrentStepId();
-            if (StringUtils.isNull(currentStepId)) {
+            String currentStepId = videoInspectTask.getCurrentStepId();//褰撳墠姝ラid
+            String currentStepStartTime = videoInspectTask.getCurrentStepStartTime();//褰撳墠姝ラ鍚姩鏃堕棿
+            if (StringUtils.isNull(currentStepId) || StringUtils.isEmpty(currentStepId)) {
+                //寮�濮嬬涓�涓楠�
                 videoInspectTask.setCurrentStepId(ardVideoInspectTaskStepList.get(0).getId());
-                startRunStep(videoInspectTask);//寮�濮嬪綋鍓嶄换鍔$殑绗竴涓楠�
-            } else /*褰撳墠浠诲姟宸茬粡鎵ц锛屽垽鏂槸鍚﹀埌鏈�*/ {
-                if (StringUtils.isNull(videoInspectTask.getCurrentStepStartTime())) {
-                    //褰撳墠浠诲姟鍚姩鏃堕棿涓簄ull锛屽垯璇存槑褰撳墠姝ラ琚腑鏂紝鐩存帴鍚姩褰撳墠姝ラ
+                startRunStep(videoInspectTask);
+            } else {
+                if (StringUtils.isNull(currentStepStartTime) || StringUtils.isEmpty(currentStepStartTime)) {
+                    //褰撳墠浠诲姟鍚姩鏃堕棿涓虹┖锛屽垯璇存槑褰撳墠姝ラ琚腑鏂紝鐩存帴鍚姩褰撳墠姝ラ
                     startRunStep(videoInspectTask);
                 } else {
                     boolean expird = isExpirdStep(videoInspectTask); /*鍒ゆ柇褰撳墠姝ラ鏃堕棿鏄惁杩囨湡*/
@@ -337,6 +436,7 @@
                         ardVideoInspectTaskMapper.updateArdVideoInspectTask(avit);
                     } else {
                         cmd.setEnable(false);//鍋滄褰曞儚
+                        cmd.setUploadMinio(true);
                         hikClientService.recordToMinio(cmd);
                         /*鎺у埗澶辫触,褰撳墠姝ラ鍚姩鏃堕棿缃畁ull*/
                         ArdVideoInspectTask avit = new ArdVideoInspectTask();
@@ -380,6 +480,7 @@
                 String recordName = cameraId + "/" + time + "/" + uuid + ".mp4";
                 cmd.setRecordBucketName("record");
                 cmd.setRecordObjectName(recordName);
+                cmd.setUploadMinio(true);
                 hikClientService.recordToMinio(cmd);
                 /*鎻掑叆宸℃璁板綍*/
                 ArdVideoInspectRecord ardVideoInspectRecord = new ArdVideoInspectRecord();

--
Gitblit v1.9.3