From c19666de02a21d62fb74c993a481141466111484 Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期三, 28 八月 2024 21:52:25 +0800
Subject: [PATCH] 工单获取图片ok

---
 ycl-server/src/main/java/com/ycl/task/WorkOrderImgTask.java                      |    8 +++
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |   77 +++++++++++++++++++-------------------
 ycl-server/src/main/java/com/ycl/platform/service/WorkOrderService.java          |    3 +
 3 files changed, 47 insertions(+), 41 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/WorkOrderService.java b/ycl-server/src/main/java/com/ycl/platform/service/WorkOrderService.java
index edd2a15..6acfa37 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/WorkOrderService.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/WorkOrderService.java
@@ -9,6 +9,7 @@
 import com.ycl.platform.domain.vo.screen.WorkOrderRegionVO;
 import com.ycl.platform.domain.vo.screen.WorkOrderTotalVO;
 import com.ycl.system.Result;
+import org.bytedeco.javacv.FFmpegFrameGrabber;
 
 import java.util.List;
 import java.util.Map;
@@ -184,7 +185,7 @@
      * @param channelId 璁惧閫氶亾缂栧彿
      * @return 鏁版嵁
      */
-    String getFrameImgByDevice(String deviceId, String channelId);
+    String getFrameImgByDevice(String deviceId, String channelId) throws FFmpegFrameGrabber.Exception;
 
     /**
      * 鏌ヨ宸ュ崟锛氬凡涓嬪彂銆佸凡澶勭悊銆佸凡瀹屾垚鐨勫伐鍗�
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 d944fb8..a389e0d 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,6 +1,7 @@
 package com.ycl.platform.service.impl;
 
 import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
@@ -662,60 +663,58 @@
 
 
     @Override
-    public String getFrameImgByDevice(String deviceId, String channelId) {
+    public String getFrameImgByDevice(String deviceId, String channelId) throws FFmpegFrameGrabber.Exception {
         String url = String.format(this.rtspServer + "/api/play/start/%s/%s", deviceId, channelId);
         String result = HttpUtils.sendGet(url);
-        log.error("鎷垮埌鍙栨祦鍝嶅簲缁撴灉锛�" + result);
+// 鍋囪WVPResult鏄纭В鏋愬搷搴旂殑绫�
         WVPResult wvpResult = JSON.parseObject(result, WVPResult.class);
-        if (0 == wvpResult.getCode()) {
-            log.error("璇锋眰鎴愬姛");
-            StreamContent data = (StreamContent) wvpResult.getData();
-            String rtspUrl = data.getRtsp();
-            log.error("鐩爣鍦板潃锛�" + rtspUrl);
+        if (wvpResult.getCode() == 0) {
+            System.out.println("璇锋眰鎴愬姛");
+            JSONObject data = (JSONObject) wvpResult.getData();
+            String rtspUrl = data.getString("fmp4"); // 娉ㄦ剰锛氶�氬父RTSP URL涓嶄細浠�"fmp4"缁撳熬锛岃繖閲屽彲鑳芥槸鐗瑰畾API鐨勮繑鍥炴牸寮�
             if (StringUtils.hasText(rtspUrl)) {
+                System.out.println("鐩爣鍦板潃锛�" + rtspUrl);
+
+                FFmpegFrameGrabber grabber = null;
                 try {
-                    FFmpegFrameGrabber grabber = FFmpegFrameGrabber.createDefault(rtspUrl);
-                    grabber.setOption("rtsp_transport", "tcp"); // 浣跨敤tcp鐨勬柟寮忥紝涓嶇劧浼氫涪鍖呭緢涓ラ噸
-                    grabber.setImageWidth(960);
-                    grabber.setImageHeight(540);
+                    grabber = FFmpegFrameGrabber.createDefault(rtspUrl);
                     grabber.start();
-                    CanvasFrame canvasFrame = new CanvasFrame("姝h寕");// 鍒涘缓绐楀彛
-                    canvasFrame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);// 璁剧疆绐楀彛鍏抽棴绋嬪簭鏄惁涔熻窡闅忓叧闂�
-                    canvasFrame.setAlwaysOnTop(true);
 
-                    OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
-                    int ex = 0;
-                    String imgPath = null;
-                    while (true) {
-                        if (ex > 1) {
-                            break;
-                        }
-                        Frame frame = grabber.grabImage();
-                        canvasFrame.showImage(frame);
-                        //绋嬪簭鍒拌繖閲屽叾瀹炲凡缁忓疄鐜颁簡棰勮鐨勫姛鑳戒簡锛屼笅闈㈢殑鏂规硶灏辨槸灏嗘祦淇濆瓨鎴愬浘鐗�
+                    Frame frame = grabber.grabImage(); // 鐩存帴鎹曡幏涓�甯�
+                    if (frame != null) {
+                        System.out.println("鎴愬姛鎹曡幏涓�甯�");
 
-                        //opencv_core.Mat
+                        // 灏咶rame杞崲涓篗at
+                        OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
                         Mat mat = converter.convertToMat(frame);
-                        imgPath = serverConfig.getUrl() + PlatformConfig.getUploadPath() + deviceId + IdUtils.fastSimpleUUID() + ".png";
-                        opencv_imgcodecs.imwrite(imgPath, mat);
 
-                        ex++;
-                        try {
-                            Thread.sleep(1000);
-                        }
-                        catch (InterruptedException e)
-                        {
-                            e.printStackTrace();
-                        } // 1000姣鍒锋柊涓�娆″湒鍍�
+                        // 鐢熸垚鍥剧墖璺緞
+                        String imgPath = PlatformConfig.getUploadPath() + deviceId + IdUtils.fastSimpleUUID() + ".png";
+                        System.out.println("鍥剧墖鍦板潃锛�" + imgPath);
+
+                        // 淇濆瓨鍥剧墖
+                        opencv_imgcodecs.imwrite(imgPath, mat);
+                    } else {
+                        System.out.println("鏈崟鑾峰埌甯�");
                     }
-                    grabber.stop(); // 鍋滄鎹曡幏
-                    grabber.release(); // 閲婃斁璧勬簮
-                    return imgPath;
-                } catch (IOException e) {
+                } catch (FrameGrabber.Exception e) {
                     e.printStackTrace();
+                } finally {
+                    if (grabber != null) {
+                        try {
+                            grabber.stop(); // 鍋滄鎹曡幏
+                        } catch (FrameGrabber.Exception e) {
+                            e.printStackTrace();
+                        }
+                        // 閫氬父涓嶉渶瑕佽皟鐢╮elease()锛屽洜涓簊top()浼氬鐞嗚祫婧愰噴鏀�
+                        // grabber.release(); // 閲婃斁璧勬簮
+                    }
                 }
             }
+        } else {
+            System.out.println("璇锋眰澶辫触锛岄敊璇爜锛�" + wvpResult.getCode());
         }
+
         return null;
     }
 
diff --git a/ycl-server/src/main/java/com/ycl/task/WorkOrderImgTask.java b/ycl-server/src/main/java/com/ycl/task/WorkOrderImgTask.java
index 2464693..85a2aec 100644
--- a/ycl-server/src/main/java/com/ycl/task/WorkOrderImgTask.java
+++ b/ycl-server/src/main/java/com/ycl/task/WorkOrderImgTask.java
@@ -7,6 +7,7 @@
 import com.ycl.platform.service.WorkOrderService;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.bytedeco.javacv.FFmpegFrameGrabber;
 import org.springframework.stereotype.Component;
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
@@ -55,7 +56,12 @@
         for (DeviceInfoVO deviceInfo : deviceList) {
             executorService.submit(() -> {
                 // 鍥芥爣璁惧鐨勭紪鐮佸氨鏄彇瑙嗛娴佺殑璁惧缂栫爜锛屽浗鏍囪澶囧氨涓�涓�傚浗鏍囪澶囩殑姣忎竴涓�氶亾浠h〃涓�涓憚鍍忓ご锛屼篃灏辨槸璁惧id鏄彇娴佺殑閫氶亾id
-                String frameImg = workOrderService.getFrameImgByDevice(gbDevices.get(0).getDeviceId(), deviceInfo.getDeviceId());
+                String frameImg = null;
+                try {
+                    frameImg = workOrderService.getFrameImgByDevice(gbDevices.get(0).getDeviceId(), deviceInfo.getDeviceId());
+                } catch (FFmpegFrameGrabber.Exception e) {
+                    e.printStackTrace();
+                }
                 if (StringUtils.hasText(frameImg)) {
                     workOrderService.updateImgById(deviceInfo.getWorkOrderId(), frameImg);
                 }

--
Gitblit v1.8.0