From e9af9f5cfeddbe5c0b33a3060b8ea6364c51e744 Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期五, 15 十一月 2024 11:37:10 +0800
Subject: [PATCH] 白名单故障类型

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |  456 +++++++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 335 insertions(+), 121 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
index dad7b84..96e8e1b 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
@@ -1,17 +1,23 @@
 package com.ycl.platform.service.impl;
 
-import com.alibaba.excel.annotation.format.DateTimeFormat;
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
+import com.alibaba.excel.write.metadata.WriteSheet;
 import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONArray;
 import com.alibaba.fastjson2.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ycl.config.PlatformConfig;
-import com.ycl.config.ServerConfig;
 import com.ycl.exception.ServiceException;
 import com.ycl.platform.domain.entity.*;
+import com.ycl.platform.domain.excel.ErrorExport;
+import com.ycl.platform.domain.excel.PointExport;
+import com.ycl.platform.domain.excel.WorkOrderWhiteExport;
 import com.ycl.platform.domain.form.*;
 import com.ycl.platform.domain.query.*;
 import com.ycl.platform.domain.vo.*;
@@ -23,13 +29,11 @@
 import com.ycl.platform.service.WorkOrderAuditingRecordService;
 import com.ycl.platform.service.WorkOrderService;
 import com.ycl.platform.service.YwPointService;
-import com.ycl.platform.wvp.StreamContent;
 import com.ycl.platform.wvp.WVPResult;
 import com.ycl.system.Result;
 import com.ycl.system.domain.SysConfig;
 import com.ycl.system.entity.SysDictData;
 import com.ycl.system.mapper.SysConfigMapper;
-import com.ycl.system.mapper.SysDictDataMapper;
 import com.ycl.system.model.LoginUser;
 import com.ycl.system.page.PageUtil;
 import com.ycl.utils.DateUtils;
@@ -39,38 +43,37 @@
 import com.ycl.utils.uuid.IdUtils;
 import constant.Constants;
 import constant.RedisConstant;
+import enumeration.ErrorType;
 import enumeration.general.*;
+import jakarta.servlet.http.HttpServletResponse;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.ObjectUtils;
-import org.bytedeco.javacv.*;
+import org.bytedeco.javacv.FFmpegFrameGrabber;
+import org.bytedeco.javacv.Frame;
+import org.bytedeco.javacv.FrameGrabber;
+import org.bytedeco.javacv.OpenCVFrameConverter;
 import org.bytedeco.opencv.global.opencv_imgcodecs;
 import org.bytedeco.opencv.opencv_core.Mat;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
-import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.context.ApplicationContext;
+import org.springframework.scheduling.annotation.Async;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.Assert;
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
+import pojo.CascadeOption;
 
-import javax.imageio.ImageIO;
-import javax.swing.*;
-import java.awt.image.BufferedImage;
-import java.io.File;
 import java.io.IOException;
-import java.net.HttpURLConnection;
-import java.net.URL;
-import java.net.URLDecoder;
-import java.net.URLEncoder;
-import java.nio.charset.StandardCharsets;
-import java.text.SimpleDateFormat;
+import java.io.OutputStream;
 import java.time.LocalDateTime;
 import java.time.ZoneId;
 import java.util.*;
 import java.util.concurrent.TimeUnit;
+import java.util.function.Function;
 import java.util.stream.Collectors;
 
 /**
@@ -91,9 +94,13 @@
     private final NotifyService notifyService;
     private final WorkOrderDistributeRecordMapper workOrderDistributeRecordMapper;
     private final WorkOrderErrorTypeServiceImpl workOrderErrorTypeService;
+    private final WorkOrderErrorTypeMapper workOrderErrorTypeMapper;
+    private final WorkOrderWhiteMapper workOrderWhiteMapper;
     private final SysConfigMapper configMapper;
     private final ReportMapper reportMapper;
     private final WorkOrderCheckImgMapper workOrderCheckImgMapper;
+    private final DeviceInfoMapper deviceInfoMapper;
+    private final ApplicationContext applicationContext;
     @Value("${rtsp.server:http://127.0.0.1:7788}")
     private String rtspServer;
 
@@ -104,15 +111,19 @@
     private RedisCache redisCache;
 
     @Override
-    @Transactional(rollbackFor = Exception.class)
     public synchronized Boolean innerAddWorkOrder(List<WorkOrder> workOrderList) {
+        //閬垮厤鍧�(浜嬪姟鍔犻攣浼氬け鏁堛�佹柟娉曞唴閮ㄨ皟鐢ㄤ簨鍔′細澶辨晥)
+        WorkOrderServiceImpl self = applicationContext.getBean(WorkOrderServiceImpl.class);
+        return self.batchAddWorkOrder(workOrderList);
+    }
+
+    @Transactional(rollbackFor = Exception.class)
+    public Boolean batchAddWorkOrder(List<WorkOrder> workOrderList) {
         int total = workOrderList.size();
-        workOrderList.stream().filter(item -> {
+        //娓呯悊鏁版嵁
+        workOrderList = workOrderList.stream().filter(item -> {
             return StringUtils.hasText(item.getSerialNumber()) && Objects.nonNull(item.getStatus()) && !CollectionUtils.isEmpty(item.getErrorTypeList());
-        });
-        if (CollectionUtils.isEmpty(workOrderList)) {
-            return Boolean.TRUE;
-        }
+        }).collect(Collectors.toList());
         // 鏍规嵁鍥芥爣鐮佸幓閲�
         workOrderList = workOrderList.stream()
                 .collect(Collectors.toMap(
@@ -120,6 +131,12 @@
                         p -> p,
                         (existing, replacement) -> existing // 鍐茬獊鏃朵繚鐣欑涓�涓�
                 )).values().stream().collect(Collectors.toList());
+        //妫�鏌ョ櫧鍚嶅崟
+        workOrderList = checkWhite(workOrderList);
+
+        if (CollectionUtils.isEmpty(workOrderList)) {
+            return Boolean.TRUE;
+        }
         List<String> serialNumberList = workOrderList.stream().map(WorkOrder::getSerialNumber).collect(Collectors.toList());
         // 鏌ュ嚭鏁版嵁搴撲腑鍥芥爣鐮佸搴旂殑鏈畬鎴愮殑宸ュ崟
         List<WorkOrder> inDatabaseWorkOrderList = baseMapper.getNotFinishedWorkOrders(serialNumberList);
@@ -161,7 +178,10 @@
                             willAddErrorType.add(workOrderErrorType);
                         }
                     }
-                    databaseWorkOrder.setStatus(WorkOrderStatusEnum.DISTRIBUTED);
+                    //閬垮厤浠d笅鍙戠殑宸ュ崟鐩存帴鍙樻垚涓嬪彂
+                    if (!databaseWorkOrder.getStatus().equals(WorkOrderStatusEnum.WAIT_DISTRIBUTE)) {
+                        databaseWorkOrder.setStatus(WorkOrderStatusEnum.DISTRIBUTED);
+                    }
                     databaseWorkOrder.setUpdateTime(now);
                     willUpdateStatusWorkOrderList.add(databaseWorkOrder);
                     updateNum++;
@@ -199,9 +219,12 @@
             return Boolean.TRUE;
         }
         List<String> willAddSerialNumber = waitAddList.stream().map(WorkOrder::getSerialNumber).collect(Collectors.toList());
+        //鍙敓鎴愯�冩牳璁惧銆佷笖鏈夎繍缁村崟浣嶇殑宸ュ崟
         List<YwPoint> pointList = new LambdaQueryChainWrapper<>(ywPointService.getBaseMapper())
                 .select(YwPoint::getUnitId, YwPoint::getSerialNumber, YwPoint::getImportantTag, YwPoint::getImportantTag, YwPoint::getProvinceTag, YwPoint::getImportantCommandImageTag)
                 .in(YwPoint::getSerialNumber, willAddSerialNumber)
+                .eq(YwPoint::getExamineStatus, Boolean.TRUE)
+                .isNotNull(YwPoint::getUnitId)
                 .list();
         Map<String, YwPoint> pointMapping = pointList.stream().collect(Collectors.toMap(YwPoint::getSerialNumber, point -> point));
         // 鏌ュ嚭閲嶇偣鐐逛綅銆佹櫘閫氱偣浣嶇殑澶勭悊鏃堕棿
@@ -211,6 +234,25 @@
         List<WorkOrder> notAddList = new ArrayList<>();
         //鏌edis浠婃棩宸ュ崟鏁伴噺
         int workOrderNum = 0;
+        for (WorkOrder workOrder : waitAddList) {
+            YwPoint point = pointMapping.get(workOrder.getSerialNumber());
+            if (Objects.isNull(point)) {
+                notAddList.add(workOrder);
+                continue;
+            }
+            if (Objects.nonNull(point.getUnitId())) {
+                workOrder.setUnitId(Math.toIntExact(point.getUnitId()));
+            }
+            if (point.getImportantTag() || point.getImportantCommandImageTag()) {
+                workOrder.setProcessingPeriod(Integer.valueOf(important.getConfigValue()));
+            } else {
+                workOrder.setProcessingPeriod(Integer.valueOf(normal.getConfigValue()));
+            }
+        }
+        waitAddList.removeAll(notAddList);
+        if (CollectionUtils.isEmpty(waitAddList)) {
+            return Boolean.TRUE;
+        }
         //UUID浣滀负value锛屼繚璇佷笂閿佺殑绾跨▼鑷繁瑙i攣
         String requestId = IdUtils.fastSimpleUUID();
         try {
@@ -234,23 +276,11 @@
             log.error("鑾峰彇閿佸紓甯�");
             return Boolean.FALSE;
         }
+        //璁板綍宸ュ崟鏁�,琛ュ厖宸ュ崟鍙�
         for (WorkOrder workOrder : waitAddList) {
-            YwPoint point = pointMapping.get(workOrder.getSerialNumber());
-            if (Objects.isNull(point)) {
-                notAddList.add(workOrder);
-                continue;
-            }
             //鏁板瓧鍓嶉潰琛�0
             workOrderNum++;
             workOrder.setWorkOrderNo(IdUtils.workOrderNO(now, String.format("%05d", workOrderNum)));
-            if (Objects.nonNull(point.getUnitId())) {
-                workOrder.setUnitId(Math.toIntExact(point.getUnitId()));
-            }
-            if (point.getImportantTag() || point.getImportantCommandImageTag()) {
-                workOrder.setProcessingPeriod(Integer.valueOf(important.getConfigValue()));
-            } else {
-                workOrder.setProcessingPeriod(Integer.valueOf(normal.getConfigValue()));
-            }
             // 淇濆瓨閿欒绫诲瀷
             for (String errorType : workOrder.getErrorTypeList()) {
                 WorkOrderErrorType workOrderErrorType = new WorkOrderErrorType();
@@ -260,12 +290,8 @@
                 workOrderErrorType.setErrorName(errorType);
                 waitAddErrorTypeList.add(workOrderErrorType);
             }
+            log.info("鍗冲皢鐢熸垚鐨勫伐鍗�:{}", workOrder);
         }
-        waitAddList.removeAll(notAddList);
-        if (CollectionUtils.isEmpty(waitAddList)) {
-            return Boolean.TRUE;
-        }
-        //璁板綍宸ュ崟鏁�
         redisCache.setCacheObject(RedisConstant.WORKORDER_NUM, workOrderNum);
         redisCache.releaseLock(RedisConstant.WORKORDER_NUM_LOCK, requestId);
         // 淇濆瓨宸ュ崟鍜屾晠闅滅被鍨�
@@ -294,6 +320,7 @@
         return Boolean.TRUE;
     }
 
+
     /**
      * 娣诲姞
      *
@@ -302,6 +329,16 @@
      */
     @Override
     public Result add(WorkOrderForm form) {
+        //妫�鏌ョ櫧鍚嶅崟
+        List<WorkOrder> data = new ArrayList<>();
+        WorkOrder workOrder = WorkOrderForm.getEntityByForm(form, null);
+        data.add(workOrder);
+        data = checkWhite(data);
+        if (CollectionUtils.isEmpty(data)) {
+            return Result.error("璇ヨ澶囩殑鏁呴殰绫诲瀷鍦ㄧ櫧鍚嶅崟涓紝涓嶈兘娣诲姞宸ュ崟");
+        }
+        WorkOrder entity = data.get(0);
+
         //鏌edis浠婃棩宸ュ崟鏁伴噺
         int workOrderNum = 0;
         //UUID浣滀负value锛屼繚璇佷笂閿佺殑绾跨▼鑷繁瑙i攣
@@ -317,7 +354,7 @@
         } else {
             return Result.error("宸ュ崟姝e湪娣诲姞锛岃绋嶅悗鍐嶈瘯");
         }
-        WorkOrder entity = WorkOrderForm.getEntityByForm(form, null);
+
         entity.setCreateTime(DateUtils.getNowDate());
         entity.setStatus(WorkOrderStatusEnum.WAIT_DISTRIBUTE);
         Date now = new Date();
@@ -332,6 +369,7 @@
         }
         return Result.error("娣诲姞澶辫触");
     }
+
 
     /**
      * 淇敼
@@ -385,14 +423,10 @@
     @Override
     @Transactional(rollbackFor = Exception.class)
     public Result batchAuditing(WorkOrderBatchAuditingForm form) {
-        // 鏍规嵁鏁呴殰绫诲瀷鑾峰彇鍒楄〃
-        List<WorkOrder> list = new LambdaQueryChainWrapper<>(baseMapper)
-                .in(WorkOrder::getStatus, WorkOrderStatusEnum.YW_HANDLE.getValue())
-                .in(WorkOrder::getErrorType, form.getErrorTypes())
-                .select(WorkOrder::getId, WorkOrder::getUnitId, WorkOrder::getWorkOrderNo, WorkOrder::getSerialNumber)
-                .list();
+        //鏍规嵁宸ュ崟缂栧彿鑾峰彇宸ュ崟
+        List<WorkOrder> list = baseMapper.selectByNos(form.getWorkOrderNumbers());
         if (list.isEmpty()) {
-            return Result.error("娌℃湁宸ュ崟鍙互瀹℃牳");
+            return Result.error("娌℃湁寰呭鏍稿伐鍗�");
         }
         List<String> workOrderNoList = list.stream().map(WorkOrder::getWorkOrderNo).collect(Collectors.toList());
         List<String> serialNumbers = list.stream().map(WorkOrder::getSerialNumber).toList();
@@ -433,6 +467,7 @@
     @Override
     @Transactional(rollbackFor = Exception.class)
     public Result ywCondition(WorkOrderYWConditionForm form) {
+
         WorkOrder workOrder = baseMapper.selectById(form.getId());
         if (Objects.isNull(workOrder)) {
             throw new ServiceException("宸ュ崟涓嶅瓨鍦�");
@@ -449,13 +484,45 @@
         workOrderYwConditionRecord.setYwProofMaterials(form.getYwProofMaterials());
         workOrderYwConditionRecord.setSysMsg(Boolean.FALSE);
         workOrderYwConditionRecordMapper.insert(workOrderYwConditionRecord);
+        //寮傛鑾峰彇鍥剧墖
+        WorkOrderServiceImpl self = applicationContext.getBean(WorkOrderServiceImpl.class);
+        self.getImage(workOrder);
         return Result.ok("鎿嶄綔鎴愬姛");
+    }
+
+    @Async("threadPoolTaskExecutor")
+    public void getImage(WorkOrder workOrder) {
+        List<DeviceInfo> gbDevices = new LambdaQueryChainWrapper<>(deviceInfoMapper)
+                .orderByDesc(DeviceInfo::getUpdateTime)
+                .last("limit 1")
+                .list();
+        if (CollectionUtils.isEmpty(gbDevices)) {
+            return;
+        }
+        // 鍥芥爣璁惧鐨勭紪鐮佸氨鏄彇瑙嗛娴佺殑璁惧缂栫爜锛屽浗鏍囪澶囧氨涓�涓�傚浗鏍囪澶囩殑姣忎竴涓�氶亾浠h〃涓�涓憚鍍忓ご锛屼篃灏辨槸璁惧id鏄彇娴佺殑閫氶亾id
+        String frameImg = null;
+        try {
+            log.info("鍥芥爣骞冲彴:{},璁惧缂栫爜:{},宸ュ崟鍙�:{}", gbDevices.get(0).getDeviceId(), workOrder.getSerialNumber(), workOrder.getWorkOrderNo());
+            frameImg = this.getFrameImgByDevice(gbDevices.get(0).getDeviceId(), workOrder.getSerialNumber(), workOrder.getWorkOrderNo());
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        if (StringUtils.hasText(frameImg)) {
+            WorkOrderCheckImg img = new WorkOrderCheckImg();
+            img.setWorkOrderNo(workOrder.getWorkOrderNo());
+            img.setImgUrl(frameImg);
+            img.setCreateTime(new Date());
+            workOrderCheckImgMapper.insert(img);
+        }
     }
 
     @Override
     public List<WorkOrderYwConditionRecordVO> selectYwConditionByYwId(String workOrderNo) {
         List<WorkOrderYwConditionRecordVO> ywConditionList = workOrderYwConditionRecordMapper.selectYwConditionByYwId(workOrderNo);
         ywConditionList.stream().forEach(item -> {
+            if (item.getUserId().equals(1)) {
+                item.setUnitName("绠$悊鍛�");
+            }
             if (Objects.nonNull(item.getSysMsg()) && item.getSysMsg()) {
                 item.setUnitName("绯荤粺娑堟伅");
             }
@@ -503,6 +570,19 @@
      */
     @Override
     public Result removeById(String id) {
+        WorkOrder workOrder = baseMapper.selectById(id);
+        String workOrderNo = workOrder.getWorkOrderNo();
+        //鍒犻櫎宸ュ崟瀹℃牳璁板綍
+        workOrderAuditingRecordMapper.delete(new QueryWrapper<WorkOrderAuditingRecord>().eq("work_order_no", workOrderNo));
+        //鍒犻櫎宸ュ崟鍥剧墖璁板綍
+        workOrderCheckImgMapper.delete(new QueryWrapper<WorkOrderCheckImg>().eq("work_order_no", workOrderNo));
+        //鍒犻櫎宸ュ崟涓嬪彂璁板綍
+        workOrderDistributeRecordMapper.delete(new QueryWrapper<WorkOrderDistributeRecord>().eq("work_order_no", workOrderNo));
+        //鍒犻櫎宸ュ崟鏁呴殰绫诲瀷
+        workOrderErrorTypeMapper.delete(new QueryWrapper<WorkOrderErrorType>().eq("work_order_no", workOrderNo));
+        //鍒犻櫎宸ュ崟鎯呭喌璁板綍
+        workOrderYwConditionRecordMapper.delete(new QueryWrapper<WorkOrderYwConditionRecord>().eq("work_order_no", workOrderNo));
+
         if (baseMapper.deleteById(id) > 0) {
             return Result.ok("鍒犻櫎鎴愬姛");
         }
@@ -518,14 +598,27 @@
     @Override
     public Result page(WorkOrderQuery query) {
         IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class);
-        query.setUnitId(SecurityUtils.getUnitId());
-        query.setStart(DateUtils.getDayStart(query.getStart()));
-        query.setEnd(DateUtils.getDayEnd(query.getEnd()));
+        if (query.getUnitId() == null) {
+            query.setUnitId(SecurityUtils.getUnitId());
+        }
+        if (query.getStart() != null) query.setStart(DateUtils.getDayStart(query.getStart()));
+        if (query.getEnd() != null) query.setEnd(DateUtils.getDayEnd(query.getEnd()));
         baseMapper.page(page, query);
         if (!CollectionUtils.isEmpty(page.getRecords())) {
             page.getRecords().stream().forEach(item -> {
                 if (StringUtils.hasText(item.getErrorType())) {
-                    item.setErrorTypeList(List.of(item.getErrorType().split(",")));
+                    List<String> errorTypeList = new ArrayList<>(List.of(item.getErrorType().split(",")));
+                    int index = errorTypeList.indexOf(query.getErrorTypeLabel());
+                    if (index != -1) {
+                        // 淇濆瓨瑕佺Щ鍔ㄧ殑鍏冪礌
+                        String firstOccurrence = errorTypeList.get(index);
+                        // 浠庡垪琛ㄤ腑绉婚櫎璇ュ厓绱狅紙绗竴娆″嚭鐜扮殑浣嶇疆锛�
+                        errorTypeList.remove(index);
+                        // 灏嗚鍏冪礌娣诲姞鍒板垪琛ㄧ殑棣栦綅
+                        errorTypeList.add(0, firstOccurrence);
+                    }
+                    // 璁剧疆鏇存柊鍚庣殑鍒楄〃
+                    item.setErrorTypeList(errorTypeList);
                 }
                 if (StringUtils.hasText(item.getImgListStr())) {
                     item.setImgList(List.of(item.getImgListStr().split(",")));
@@ -538,6 +631,8 @@
     @Override
     public Result distributePage(DistributeWorkOrderQuery query) {
         IPage<WorkOrderVO> page = PageUtil.getPage(query, WorkOrderVO.class);
+        if (query.getStart() != null) query.setStart(DateUtils.getDayStart(query.getStart()));
+        if (query.getEnd() != null) query.setEnd(DateUtils.getDayEnd(query.getEnd()));
         baseMapper.distributePage(page, query);
         return Result.ok().data(page).total(page.getTotal());
     }
@@ -763,61 +858,70 @@
 
 
     @Override
-    public synchronized String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) throws Exception {
-        String url = String.format(this.rtspServer + "/api/play/start/%s/%s", deviceId, channelId);
+    public String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) throws Exception {
+        String url = String.format(this.rtspServer + "/api/play/start/img/%s/%s", deviceId, channelId);
+        log.info("璁块棶璺緞{}", url);
         String result = HttpUtils.sendGet(url);
-        log.info("鎷垮埌鍙栨祦鍝嶅簲缁撴灉锛�" + result);
+        log.info("鎷垮埌鍙栨祦鍥剧墖鍝嶅簲缁撴灉锛�" + result);
         WVPResult wvpResult = JSON.parseObject(result, WVPResult.class);
-        String imgUrl = null;
         if (wvpResult.getCode() == 0) {
-            JSONObject data = (JSONObject) wvpResult.getData();
-            String rtspUrl = data.getString("fmp4"); // 鍙杕p4鍦板潃
-            if (StringUtils.hasText(rtspUrl)) {
-                System.out.println("鐩爣鍦板潃锛�" + rtspUrl);
-                FFmpegFrameGrabber grabber = null;
-                try {
-                    grabber = new FFmpegFrameGrabber(rtspUrl);
-//                    grabber.setOption("rtsp_transport", "tcp"); // 浣跨敤tcp鐨勬柟寮忥紝涓嶇劧浼氫涪鍖呭緢涓ラ噸
-//                    grabber.setVideoOption("probesize", "10000"); // 璁剧疆鎹曡幏鍒嗘瀽鐨勬渶澶у瓧鑺�
-                    grabber.start();
-                    Frame frame = grabber.grabImage(); // 鐩存帴鎹曡幏涓�甯�
-                    if (frame != null) {
-                        System.out.println("鎴愬姛鎹曡幏涓�甯�");
-                        // 灏咶rame杞崲涓篗at
-                        OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
-                        Mat mat = converter.convertToMat(frame);
-
-                        imgUrl = workOrderNo + "_" + IdUtils.fastSimpleUUID() + ".png";
-                        // 鐢熸垚鍥剧墖璺緞
-                        String imgPath = PlatformConfig.getProfile() + "/" + imgUrl;
-                        System.out.println("鍥剧墖淇濆瓨鍦板潃锛�" + imgPath);
-                        imgUrl = Constants.RESOURCE_PREFIX + "/" + imgUrl;
-                        // 淇濆瓨鍥剧墖
-                        opencv_imgcodecs.imwrite(imgPath, mat);
-                    } else {
-                        System.out.println("鏈崟鑾峰埌甯�");
-                    }
-                } catch (FrameGrabber.Exception e) {
-                    e.printStackTrace();
-                } finally {
-                    if (grabber != null) {
-                        try {
-                            grabber.stop(); // 鍋滄鎹曡幏
-                        } catch (FrameGrabber.Exception e) {
-                            e.printStackTrace();
-                        }
-                    }
-                }
-            }
-        } else {
-            System.out.println("璇锋眰澶辫触锛岄敊璇爜锛�" + wvpResult.getCode() + "--" + wvpResult.getMsg());
+            Object imgUrl = (Object) wvpResult.getData();
+            imgUrl = Objects.isNull(imgUrl) ? "" : imgUrl;
+            log.info("鎷垮埌鍙栨祦鍥剧墖锛�" + imgUrl);
+            return (String) imgUrl;
         }
-        System.out.println("鍥剧墖URL锛�" + imgUrl);
-        return imgUrl;
+        return null;
+//        WVPResult wvpResult = JSON.parseObject(result, WVPResult.class);
+//        String imgUrl = null;
+//        if (wvpResult.getCode() == 0) {
+//            JSONObject data = (JSONObject) wvpResult.getData();
+//            String rtspUrl = data.getString("fmp4"); // 鍙杕p4鍦板潃
+//            if (StringUtils.hasText(rtspUrl)) {
+//                System.out.println("鐩爣鍦板潃锛�" + rtspUrl);
+//                FFmpegFrameGrabber grabber = null;
+//                try {
+//                    grabber = new FFmpegFrameGrabber(rtspUrl);
+////                    grabber.setOption("rtsp_transport", "tcp"); // 浣跨敤tcp鐨勬柟寮忥紝涓嶇劧浼氫涪鍖呭緢涓ラ噸
+////                    grabber.setVideoOption("probesize", "10000"); // 璁剧疆鎹曡幏鍒嗘瀽鐨勬渶澶у瓧鑺�
+//                    grabber.start();
+//                    Frame frame = grabber.grabImage(); // 鐩存帴鎹曡幏涓�甯�
+//                    if (frame != null) {
+//                        System.out.println("鎴愬姛鎹曡幏涓�甯�");
+//                        // 灏咶rame杞崲涓篗at
+//                        OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
+//                        Mat mat = converter.convertToMat(frame);
+//
+//                        imgUrl = workOrderNo + "_" + IdUtils.fastSimpleUUID() + ".png";
+//                        // 鐢熸垚鍥剧墖璺緞
+//                        String imgPath = PlatformConfig.getProfile() + "/" + imgUrl;
+//                        System.out.println("鍥剧墖淇濆瓨鍦板潃锛�" + imgPath);
+//                        imgUrl = Constants.RESOURCE_PREFIX + "/" + imgUrl;
+//                        // 淇濆瓨鍥剧墖
+//                        opencv_imgcodecs.imwrite(imgPath, mat);
+//                    } else {
+//                        System.out.println("鏈崟鑾峰埌甯�");
+//                    }
+//                } catch (FrameGrabber.Exception e) {
+//                    e.printStackTrace();
+//                } finally {
+//                    if (grabber != null) {
+//                        try {
+//                            grabber.stop(); // 鍋滄鎹曡幏
+//                        } catch (FrameGrabber.Exception e) {
+//                            e.printStackTrace();
+//                        }
+//                    }
+//                }
+//            }
+//        } else {
+//            System.out.println("璇锋眰澶辫触锛岄敊璇爜锛�" + wvpResult.getCode() + "--" + wvpResult.getMsg());
+//        }
+//        System.out.println("鍥剧墖URL锛�" + imgUrl);
+//        return imgUrl;
     }
 
     @Override
-    public void saveFrameImgByDevice (String deviceId, String channelId, String workOrderNo){
+    public void saveFrameImgByDevice(String deviceId, String channelId, String workOrderNo) {
         String url = String.format(this.rtspServer + "/api/play/start/%s/%s", deviceId, channelId);
         String result = HttpUtils.sendGet(url);
         WVPResult wvpResult = JSON.parseObject(result, WVPResult.class);
@@ -878,13 +982,13 @@
     }
 
     @Override
-    public List<DeviceInfoVO> hasErrorWorkOrderList (Date start, Date end){
+    public List<DeviceInfoVO> hasErrorWorkOrderList(Date start, Date end) {
         List<DeviceInfoVO> list = baseMapper.hasErrorWorkOrderList(start, end);
         return list;
     }
 
     @Override
-    public void updateImgById (Integer workOrderId, String imgPath){
+    public void updateImgById(Integer workOrderId, String imgPath) {
         new LambdaUpdateChainWrapper<>(baseMapper)
                 .eq(WorkOrder::getId, workOrderId)
                 .set(WorkOrder::getYwCheckResult, imgPath)
@@ -892,7 +996,7 @@
     }
 
     @Override
-    public Result processImg (String workOrderNo){
+    public Result processImg(String workOrderNo) {
         WorkOrder workOrder = new LambdaQueryChainWrapper<>(baseMapper)
                 .eq(WorkOrder::getWorkOrderNo, workOrderNo)
                 .one();
@@ -933,7 +1037,7 @@
     }
 
     @Override
-    public Result detailByNo (String workOrderNo){
+    public Result detailByNo(String workOrderNo) {
         WorkOrderDetailVO workOrder = baseMapper.detailByNo(workOrderNo);
         // 鏄惁鎶ュ
         boolean hasReport = new LambdaQueryChainWrapper<>(reportMapper)
@@ -955,23 +1059,133 @@
     }
 
     @Override
-    public List<WorkOrderVO> export (WorkOrderExportQuery query){
-        query.setUnitId(SecurityUtils.getUnitId());
-        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-        if (query.getStart() == null) {
-            query.setStart(format.format(DateUtils.getDayStart(new Date())));
-        } else {
-            query.setStart(query.getStart() + " 00:00:00");
+    public List<WorkOrderVO> export(WorkOrderExportQuery query) {
+        if (query.getUnitId() == null) {
+            query.setUnitId(SecurityUtils.getUnitId());
         }
-        if (query.getEnd() == null) {
-            query.setStart(format.format(DateUtils.getDayEnd(new Date())));
-        } else {
-            query.setEnd(query.getEnd() + " 23:59:59");
-        }
+        if (query.getStart() != null) query.setStart(query.getStart() + " 00:00:00");
+        if (query.getEnd() != null) query.setEnd(query.getEnd() + " 23:59:59");
 
-        List<WorkOrderVO> export = baseMapper.export(query);
-
-        System.out.println(export);
-        return export;
+        return baseMapper.export(query);
     }
+
+    /**
+     * 妫�娴嬪伐鍗曟寜閽�
+     *
+     * @param workOrderNo
+     * @return
+     */
+    @Override
+    public Result checkImage(String workOrderNo, String serialNumber) {
+        // 鏌ュ嚭鍥芥爣璁惧锛屽氨涓�鏉℃暟鎹�
+        List<DeviceInfo> gbDevices = new LambdaQueryChainWrapper<>(deviceInfoMapper)
+                .orderByDesc(DeviceInfo::getUpdateTime)
+                .last("limit 1")
+                .list();
+        if (!CollectionUtils.isEmpty(gbDevices)) {
+            try {
+                String imgUrl = this.getFrameImgByDevice(gbDevices.get(0).getDeviceId(), serialNumber, workOrderNo);
+                if (StringUtils.hasText(imgUrl)) {
+                    WorkOrderCheckImg img = new WorkOrderCheckImg();
+                    img.setWorkOrderNo(workOrderNo);
+                    img.setImgUrl(imgUrl);
+                    img.setCreateTime(new Date());
+                    workOrderCheckImgMapper.insert(img);
+                }
+                return Result.ok().data(imgUrl);
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        }
+        return Result.ok();
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public Result batchDeleteWorkOrder(List<String> workOrderNos) {
+        if (!CollectionUtils.isEmpty(workOrderNos)) {
+            //鍒犻櫎宸ュ崟瀹℃牳璁板綍
+            workOrderAuditingRecordMapper.delete(new QueryWrapper<WorkOrderAuditingRecord>().in("work_order_no", workOrderNos));
+            //鍒犻櫎宸ュ崟鍥剧墖璁板綍
+            workOrderCheckImgMapper.delete(new QueryWrapper<WorkOrderCheckImg>().in("work_order_no", workOrderNos));
+            //鍒犻櫎宸ュ崟涓嬪彂璁板綍
+            workOrderDistributeRecordMapper.delete(new QueryWrapper<WorkOrderDistributeRecord>().in("work_order_no", workOrderNos));
+            //鍒犻櫎宸ュ崟鏁呴殰绫诲瀷
+            workOrderErrorTypeMapper.delete(new QueryWrapper<WorkOrderErrorType>().in("work_order_no", workOrderNos));
+            //鍒犻櫎宸ュ崟鎯呭喌璁板綍
+            workOrderYwConditionRecordMapper.delete(new QueryWrapper<WorkOrderYwConditionRecord>().in("work_order_no", workOrderNos));
+            //鍒犻櫎宸ュ崟
+            this.baseMapper.delete(new QueryWrapper<WorkOrder>().in("work_order_no", workOrderNos));
+        }
+        return Result.ok();
+    }
+
+    @Override
+    public Result errorAll() {
+        List<CascadeOption> results = new ArrayList<>();
+        CascadeOption video = new CascadeOption();
+        video.setLabel("瑙嗛");
+        video.setValue("VIDEO");
+        List<ErrorType> videoErr = ErrorType.getErrorTypesByCategory("VIDEO");
+        List<ErrorType> videoErr1 = ErrorType.getErrorTypesByCategory("COMMON");
+        videoErr.addAll(videoErr1);
+        List<CascadeOption> videoChildren = videoErr.stream().map(item -> CascadeOption.builder()
+                .value(item.getValue())
+                .label(item.getDesc())
+                .build()).collect(Collectors.toList());
+        video.setChildren(videoChildren);
+
+        CascadeOption car = new CascadeOption();
+        car.setLabel("杞﹁締");
+        car.setValue("CAR");
+        List<ErrorType> carErr = ErrorType.getErrorTypesByCategory("CAR");
+        List<ErrorType> carErr1 = ErrorType.getErrorTypesByCategory("COMMON");
+        List<ErrorType> carErr2 = ErrorType.getErrorTypesByCategory("CARORFACE");
+        carErr.addAll(carErr1);
+        carErr.addAll(carErr2);
+        List<CascadeOption> carChildren = carErr.stream().map(item -> CascadeOption.builder()
+                .value(item.getValue())
+                .label(item.getDesc())
+                .build()).collect(Collectors.toList());
+        car.setChildren(carChildren);
+
+        CascadeOption face = new CascadeOption();
+        face.setLabel("浜鸿劯");
+        face.setValue("FACE");
+        List<ErrorType> faceErr = ErrorType.getErrorTypesByCategory("FACE");
+        List<ErrorType> faceErr1 = ErrorType.getErrorTypesByCategory("COMMON");
+        List<ErrorType> faceErr2 = ErrorType.getErrorTypesByCategory("CARORFACE");
+        faceErr.addAll(faceErr1);
+        faceErr.addAll(faceErr2);
+        List<CascadeOption> faceChildren = faceErr.stream().map(item -> CascadeOption.builder()
+                .value(item.getValue())
+                .label(item.getDesc())
+                .build()).collect(Collectors.toList());
+        face.setChildren(faceChildren);
+
+        results.add(video);
+        results.add(car);
+        results.add(face);
+        return Result.ok().data(results);
+    }
+
+    //鐧藉悕鍗曟牎楠�
+    private List<WorkOrder> checkWhite(List<WorkOrder> workOrderList) {
+        // 鏌ヨ鍑虹櫧鍚嶅崟鍒楄〃
+        Map<String, WorkOrderWhite> whiteMap = workOrderWhiteMapper.selectList().stream().collect(Collectors.toMap(WorkOrderWhite::getSerialNumber, Function.identity()));
+        workOrderList = workOrderList.stream()
+                .filter(workOrder -> {
+                    List<String> errorTypeList = workOrder.getErrorTypeList();
+                    WorkOrderWhite white = whiteMap.get(workOrder.getSerialNumber());
+                    if (white != null) {
+                        List<String> whiteErrorListSet = Arrays.asList(white.getErrorType().split(","));
+                        errorTypeList.removeIf(whiteErrorListSet::contains);
+                    }
+                    return !errorTypeList.isEmpty(); // 鍙繚鐣� errorTypeList 闈炵┖鐨� WorkOrder
+                })
+                .collect(Collectors.toList());
+        return workOrderList;
+    }
+
+
 }

--
Gitblit v1.8.0