‘liusuyi’
2023-09-06 1f7962d0545de446e96ccaa81b33bbf71a69b41b
ruoyi-quartz/src/main/java/com/ruoyi/quartz/task/PatrolInspectionTask.java
@@ -1,26 +1,10 @@
package com.ruoyi.quartz.task;
import com.ruoyi.alarmpoints.well.domain.ArdAlarmpointsWell;
import com.ruoyi.alarmpoints.well.service.IArdAlarmpointsWellService;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.device.camera.domain.ArdCameras;
import com.ruoyi.device.camera.domain.CameraCmd;
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.domain.ArdVideoInspectTask;
import com.ruoyi.inspect.domain.ArdVideoInspectTaskStep;
import com.ruoyi.inspect.service.IArdVideoInspectRecordService;
import com.ruoyi.inspect.service.IArdVideoInspectTaskService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.*;
/**
 * @Description: 巡检任务
@@ -30,163 +14,17 @@
 * @Version: 1.0
 **/
@Component("PatrolInspectionTask")
@Slf4j(topic = "PatrolInspectionTask")
public class PatrolInspectionTask {
    @Resource
    IArdVideoInspectTaskService ardVideoInspectTaskService;
    @Resource
    IArdAlarmpointsWellService ardAlarmpointsWellService;
    @Resource
    IArdVideoInspectRecordService ardVideoInspectRecordService;
    @Resource
    IArdCamerasService ardCamerasService;
    @Resource
    IHikClientService hikClientService;
    public void scanRun(){
        /*扫描所有可执行任务1-时间满足2-自动*/
        log.info("开始扫描所有可执行任务");
        ArdVideoInspectTask ardVideoInspectTask = new ArdVideoInspectTask();
        ardVideoInspectTask.setInspectMode("自动");
        List<ArdVideoInspectTask> ardVideoInspectTasks = ardVideoInspectTaskService.selectArdVideoInspectTaskList(ardVideoInspectTask);
        for (ArdVideoInspectTask videoInspectTask : ardVideoInspectTasks) {
            /*遍历所有时间满足的自动任务*/
            boolean timeCompare = DateUtils.TimeCompare(videoInspectTask.getStartTime(), videoInspectTask.getEndTime());
            if (timeCompare) {
                log.info("满足时间");
                /*当前任务的巡检步骤*/
                List<ArdVideoInspectTaskStep> ardVideoInspectTaskStepList = ardVideoInspectTaskService.selectArdVideoInspectTaskById(videoInspectTask.getId()).getArdVideoInspectTaskStepList();
                Map<Integer, Object> sortMap = new HashMap<>();//排序map
                Map<String, Object> stepMap = new HashMap<>();//IDmap
                for (ArdVideoInspectTaskStep step : ardVideoInspectTaskStepList) {
                    sortMap.put(step.getOrderNumber(), step);
                    stepMap.put(step.getId(), step);
                }
                /*获取当前任务正在执行的步骤和当前步骤开始的时间*/
                String currentStepId = videoInspectTask.getCurrentStepId();
                String cameraId = videoInspectTask.getCameraId();
                Integer channel = videoInspectTask.getChannel();
                String currentStepStartTime = videoInspectTask.getCurrentStepStartTime();
                /*如果当前步骤未找到,则从1开始*/
                if (StringUtils.isNull(currentStepId)) {
                    ArdVideoInspectTaskStep step = (ArdVideoInspectTaskStep) sortMap.get(1);
                    String wellId = step.getWellId();//获取第一个步骤的井号
                    if (!StringUtils.isNull(wellId)) {
                        /*获取井坐标*/
                        ArdAlarmpointsWell ardAlarmpointsWell = ardAlarmpointsWellService.selectArdAlarmpointsWellById(wellId);
                        double[] targetPositon = new double[3];
                        targetPositon[0] = ardAlarmpointsWell.getLongitude();
                        targetPositon[1] = ardAlarmpointsWell.getLatitude();
                        targetPositon[2] = ardAlarmpointsWell.getAltitude();
                        /*获取相机坐标*/
                        ArdCameras cameras = ardCamerasService.selectArdCamerasById(cameraId);
                        double[] cameraPositon = new double[3];
                        cameraPositon[0] = cameras.getLongitude();
                        cameraPositon[1] = cameras.getLatitude();
                        cameraPositon[2] = cameras.getAltitude();
                        /*控制相机巡检*/
                        CameraCmd cmd = new CameraCmd();
                        cmd.setCameraId(cameraId);
                        cmd.setChannelNum(channel);
                        cmd.setCamPosition(cameraPositon);
                        cmd.setTargetPosition(targetPositon);
                        cmd.setOperator("sys_patrol_inspect");
                        cmd.setExpired(step.getRecordingTime());
                        boolean setTargetPosition = hikClientService.setTargetPosition(cmd);
                        if (setTargetPosition) {
                            /*控制相机巡检成功,开始录像*/
                            cmd.setEnable(true);//启动录像
                            hikClientService.recordToMinio(cmd);
                            /*更新任务当前步骤id和步骤启动时间*/
                            ArdVideoInspectTask avit = new ArdVideoInspectTask();
                            avit.setId(videoInspectTask.getId());
                            avit.setCurrentStepId(ardVideoInspectTaskStepList.get(0).getId());
                            avit.setCurrentStepStartTime(DateUtils.getTime());
                            ardVideoInspectTaskService.updateArdVideoInspectTaskNoUpdater(avit);
                        }
                    }
                } else /*当前任务已经执行,判断是否到期*/ {
                    if (stepMap.containsKey(currentStepId)) {
                        /*获取当前步骤信息*/
                        ArdVideoInspectTaskStep currentStep = (ArdVideoInspectTaskStep) stepMap.get(currentStepId);
                        /*获取到当前步骤的开始和结束时间*/
                        Date currentStepStartDate = DateUtils.dateTime(DateUtils.YYYY_MM_DD_HH_MM_SS, currentStepStartTime);
                        Date currentStepStopDate = DateUtils.addMinutes(currentStepStartDate, currentStep.getRecordingTime());
                        /*判断当前步骤时间是否过期*/
                        if (!DateUtils.TimeCompare(currentStepStartDate, currentStepStopDate)) {
                            /*已过期,停止录像*/
                            CameraCmd cmd = new CameraCmd();
                            cmd.setCameraId(cameraId);
                            cmd.setChannelNum(channel);
                            cmd.setOperator("sys_patrol_inspect");
                            cmd.setEnable(false);//停止录像
                            String uuid = UUID.randomUUID().toString().replace("-", "");
                            String time = new SimpleDateFormat("yyyyMMdd").format(new Date());
                            String recordName = cameraId + "/" + time + "/" + uuid + ".mp4";
                            cmd.setRecordBucketName("record");
                            cmd.setRecordObjectName(recordName);
                            String url= MinioClientSingleton.domainUrl+ "/" +cmd.getRecordBucketName()+ "/"+recordName;
                            hikClientService.recordToMinio(cmd);
                            /*更新巡检记录*/
                            ArdVideoInspectRecord ardVideoInspectRecord = new ArdVideoInspectRecord();
                            ardVideoInspectRecord.setStepId(currentStepId);
                            ardVideoInspectRecord.setStartTime(DateUtils.parseDateToStr(DateUtils.YYYY_MM_DD_HH_MM_SS, currentStepStartDate));
                            ardVideoInspectRecord.setEndTime(DateUtils.parseDateToStr(DateUtils.YYYY_MM_DD_HH_MM_SS, currentStepStopDate));
                            ardVideoInspectRecord.setRecordFilePath(url);
                            ardVideoInspectRecordService.insertArdVideoInspectRecord(ardVideoInspectRecord);
                            /*切换下一步骤*/
                            Integer orderNumber = currentStep.getOrderNumber();
                            /*判断当前步骤序号是否小于步骤总数*/
                            if (orderNumber < stepMap.size()) {
                                /*小于则执行下一步骤*/
                                orderNumber++;
                            } else {
                                /*否则从1开始执行*/
                                orderNumber = 1;
                            }
                            ArdVideoInspectTaskStep step = (ArdVideoInspectTaskStep) sortMap.get(orderNumber);
                            String wellId = step.getWellId();//获取第一个步骤的井号
                            if (!StringUtils.isNull(wellId)) {
                                /*获取井坐标*/
                                ArdAlarmpointsWell ardAlarmpointsWell = ardAlarmpointsWellService.selectArdAlarmpointsWellById(wellId);
                                double[] targetPositon = new double[3];
                                targetPositon[0] = ardAlarmpointsWell.getLongitude();
                                targetPositon[1] = ardAlarmpointsWell.getLatitude();
                                targetPositon[2] = ardAlarmpointsWell.getAltitude();
                                /*获取相机坐标*/
                                ArdCameras cameras = ardCamerasService.selectArdCamerasById(cameraId);
                                double[] cameraPositon = new double[3];
                                cameraPositon[0] = cameras.getLongitude();
                                cameraPositon[1] = cameras.getLatitude();
                                cameraPositon[2] = cameras.getAltitude();
                                /*控制相机巡检*/
                                log.info("指向" + ardAlarmpointsWell.getWellId() + "坐标:" + targetPositon);
                                cmd = new CameraCmd();
                                cmd.setCameraId(cameraId);
                                cmd.setChannelNum(channel);
                                cmd.setCamPosition(cameraPositon);
                                cmd.setTargetPosition(targetPositon);
                                cmd.setOperator("sys_patrol_inspect");
                                cmd.setExpired(step.getRecordingTime());
                                boolean setTargetPosition = hikClientService.setTargetPosition(cmd);
                                if (setTargetPosition) {
                                    /*控制相机巡检成功,开始录像*/
                                    cmd.setEnable(true);//启动录像
                                    hikClientService.recordToMinio(cmd);
                                    /*更新任务当前步骤id和步骤启动时间*/
                                    ArdVideoInspectTask avit = new ArdVideoInspectTask();
                                    avit.setId(videoInspectTask.getId());
                                    avit.setCurrentStepId(step.getId());
                                    avit.setCurrentStepStartTime(DateUtils.getTime());
                                    ardVideoInspectTaskService.updateArdVideoInspectTaskNoUpdater(avit);
                                }
                            }
                        }
                    }
                }
            }
        }
    /**
     * 定时巡检任务
     * 刘苏义
     * 2023/8/9 9:18:42
     */
    public void scanRun() {
      ardVideoInspectTaskService.autoTaskRun();
    }
}