From 6d1540258545a41fcbaa50c43b3480a22896d03e Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期一, 30 九月 2024 18:23:20 +0800
Subject: [PATCH] 点位在线定时任务卡死修复

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |   85 +++++++++++++++++++++---------------------
 1 files changed, 42 insertions(+), 43 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 c8b9498..27f6298 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
@@ -56,6 +56,11 @@
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
 
+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;
@@ -758,69 +763,63 @@
 
 
     @Override
-    public synchronized String getFrameImgByDevice(String deviceId, String channelId, String workOrderNo) {
+    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);
         String result = HttpUtils.sendGet(url);
+        log.error("鎷垮埌鍙栨祦鍝嶅簲缁撴灉锛�" + result);
         WVPResult wvpResult = JSON.parseObject(result, WVPResult.class);
         String imgUrl = null;
-        if (wvpResult.getCode() == 0) {
+        if (0 == wvpResult.getCode()) {
+            log.error("璇锋眰鎴愬姛");
             JSONObject data = (JSONObject) wvpResult.getData();
-            String rtspUrl = data.getString("fmp4"); // 鍙杕p4鍦板潃
+            String rtspUrl = data.getString("rtsp"); // 鍙栧湴鍧�
+            log.error("鐩爣鍦板潃锛�" + rtspUrl);
+            FFmpegFrameGrabber grabber = FFmpegFrameGrabber.createDefault(rtspUrl);
             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"); // 璁剧疆鎹曡幏鍒嗘瀽鐨勬渶澶у瓧鑺�
-                    //璁剧疆10s瓒呮椂
-                    grabber.setTimeout(10000);
+                    grabber.setImageWidth(1920);
+                    grabber.setImageHeight(1080);
+                    grabber.setFrameRate(25);
                     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();
+                    Frame frame = grabber.grabFrame();
+                    if (frame != null) {
+                        //杞崲鍥惧儚
+                        Java2DFrameConverter converter = new Java2DFrameConverter();
+                        BufferedImage srcImage = converter.getBufferedImage(frame);
+                        if (srcImage != null) {
+                            imgUrl = workOrderNo + "_" + IdUtils.fastSimpleUUID() + ".png";
+                            // 鐢熸垚鍥剧墖璺緞
+                            String imgPath = PlatformConfig.getProfile() + "/" + imgUrl;
+                            System.out.println("鍥剧墖淇濆瓨鍦板潃锛�" + imgPath);
+                            imgUrl = Constants.RESOURCE_PREFIX + "/" + imgUrl;
+                            //鍒涘缓鏂囦欢
+                            File file = new File(imgPath);
+                            //杈撳嚭鏂囦欢
+                            ImageIO.write(srcImage, "png", file);
                         }
                     }
+                } catch (IOException e) {
+                    e.printStackTrace();
+                } finally {
+                    grabber.stop(); // 鍋滄鎹曡幏
+                    grabber.close(); // 閲婃斁璧勬簮
                 }
             }
-        } 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);
         String imgUrl = null;
         if (wvpResult.getCode() == 0) {
             JSONObject data = (JSONObject) wvpResult.getData();
-            String rtspUrl = data.getString("fmp4"); // 鍙杕p4鍦板潃
+            String rtspUrl = data.getString("rtsp"); // 鍙杕p4鍦板潃
             if (StringUtils.hasText(rtspUrl)) {
                 System.out.println("鐩爣鍦板潃锛�" + rtspUrl);
                 FFmpegFrameGrabber grabber = null;
@@ -874,13 +873,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)
@@ -888,7 +887,7 @@
     }
 
     @Override
-    public Result processImg(String workOrderNo) {
+    public Result processImg (String workOrderNo){
         WorkOrder workOrder = new LambdaQueryChainWrapper<>(baseMapper)
                 .eq(WorkOrder::getWorkOrderNo, workOrderNo)
                 .one();
@@ -929,7 +928,7 @@
     }
 
     @Override
-    public Result detailByNo(String workOrderNo) {
+    public Result detailByNo (String workOrderNo){
         WorkOrderDetailVO workOrder = baseMapper.detailByNo(workOrderNo);
         // 鏄惁鎶ュ
         boolean hasReport = new LambdaQueryChainWrapper<>(reportMapper)
@@ -951,7 +950,7 @@
     }
 
     @Override
-    public List<WorkOrderVO> export(WorkOrderExportQuery query) {
+    public List<WorkOrderVO> export (WorkOrderExportQuery query){
         query.setUnitId(SecurityUtils.getUnitId());
         SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
         if (query.getStart() == null) {

--
Gitblit v1.8.0