From e945a560b26437d610754cea5abfd992cd63fd7b Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Wed, 21 Aug 2024 13:39:14 +0800
Subject: [PATCH] 媒体文件处理,国土对接证书替换

---
 src/main/java/com/dji/sample/manage/controller/LiveStreamController.java |   71 ++++++++++++++++++++++++++++-------
 1 files changed, 57 insertions(+), 14 deletions(-)

diff --git a/src/main/java/com/dji/sample/manage/controller/LiveStreamController.java b/src/main/java/com/dji/sample/manage/controller/LiveStreamController.java
index c8bd144..c3a72cf 100644
--- a/src/main/java/com/dji/sample/manage/controller/LiveStreamController.java
+++ b/src/main/java/com/dji/sample/manage/controller/LiveStreamController.java
@@ -1,8 +1,8 @@
 package com.dji.sample.manage.controller;
 
-import com.dji.sample.common.model.CustomClaim;
 import com.dji.sample.common.model.ResponseResult;
 import com.dji.sample.component.mqtt.model.ChannelName;
+import com.dji.sample.manage.model.dto.CapacityCameraDTO;
 import com.dji.sample.manage.model.dto.CapacityDeviceDTO;
 import com.dji.sample.manage.model.dto.LiveTypeDTO;
 import com.dji.sample.manage.model.receiver.LiveCapacityReceiver;
@@ -14,10 +14,8 @@
 import org.springframework.messaging.MessageHeaders;
 import org.springframework.web.bind.annotation.*;
 
-import javax.servlet.http.HttpServletRequest;
+import java.io.IOException;
 import java.util.List;
-
-import static com.dji.sample.component.AuthInterceptor.TOKEN_CLAIM;
 
 /**
  * @author sean.zhou
@@ -46,23 +44,62 @@
         liveStreamService.saveLiveCapacity(liveCapacity, headers.getTimestamp());
     }
 
+
+    /**
+     * 获取直播地址
+     * @param workspaceId 项目id
+     * @param sn 设备号
+     * @return
+     */
+    @GetMapping("/getLiveUrl/{workspace_id}")
+    public ResponseResult getLiveUrl(@PathVariable("workspace_id") String workspaceId, String sn) {
+        List<CapacityDeviceDTO> liveCapacity = liveStreamService.getLiveCapacity(workspaceId,sn);
+
+        if (null == liveCapacity || liveCapacity.isEmpty()) {
+            return ResponseResult.error(-1,"暂无直播或者设备未开机");
+        }
+        CapacityDeviceDTO deviceDTO = liveCapacity.get(0);
+
+        if (null == deviceDTO.getCamerasList() || deviceDTO.getCamerasList().isEmpty()) {
+            return ResponseResult.error(-1,"暂无直播地址");
+        }
+
+        List<CapacityCameraDTO> camerasList = deviceDTO.getCamerasList();
+        if (null == camerasList || camerasList.isEmpty()) {
+            return ResponseResult.error(-1,"获取相机信息失败");
+        }
+
+        CapacityCameraDTO cameraDTO = camerasList.get(0);
+
+        String videoIndex = cameraDTO.getVideosList().get(0).getIndex();
+
+        LiveTypeDTO liveParam = new LiveTypeDTO();
+        String videoId = deviceDTO.getSn() + "-" + cameraDTO.getIndex() + "-" + videoIndex;
+        liveParam.setUrl("rtmp://www.ainfo.top:735/uav/" + videoId);
+        liveParam.setUrlType(1);
+        liveParam.setVideoId(deviceDTO.getSn() + "/" + cameraDTO.getIndex() + "/" + videoIndex);
+        liveParam.setVideoQuality(0);
+        return liveStreamService.liveStart(liveParam);
+    }
+
     /**
      * Get live capability data of all drones in the current user's workspace from the database.
-     * @param request
+     * 从数据库中获取当前工作区中所有无人机的实时性能数据。
+     * @param workspaceId
      * @return  live capability
      */
-    @GetMapping("/capacity")
-    public ResponseResult<List<CapacityDeviceDTO>> getLiveCapacity(HttpServletRequest request) {
-        // Get information about the current user.
-        CustomClaim customClaim = (CustomClaim)request.getAttribute(TOKEN_CLAIM);
+    @GetMapping("/capacity/{workspace_id}")
+    public ResponseResult<List<CapacityDeviceDTO>> getLiveCapacity(@PathVariable("workspace_id") String workspaceId,String sn) {
+        // Get information about the current user. 获取当前登录用户的信息
+//        CustomClaim customClaim = (CustomClaim)request.getAttribute(TOKEN_CLAIM);
 
-        List<CapacityDeviceDTO> liveCapacity = liveStreamService.getLiveCapacity(customClaim.getWorkspaceId());
-
+        List<CapacityDeviceDTO> liveCapacity = liveStreamService.getLiveCapacity(workspaceId,sn);
         return ResponseResult.success(liveCapacity);
     }
 
     /**
      * Live streaming according to the parameters passed in from the web side.
+     * 根据从web端传入的参数进行直播。
      * @param liveParam Live streaming parameters.
      * @return
      */
@@ -71,8 +108,14 @@
         return liveStreamService.liveStart(liveParam);
     }
 
+    @PostMapping("/streams/address")
+    public ResponseResult liveAddress(@RequestParam String deviceSn,@RequestParam String deviceName) throws IOException {
+        return liveStreamService.liveAddress(deviceSn,deviceName);
+    }
+
     /**
      * Stop live streaming according to the parameters passed in from the web side.
+     * 根据从web端传入的参数停止直播。
      * @param liveParam Live streaming parameters.
      * @return
      */
@@ -82,8 +125,8 @@
     }
 
     /**
-     * Set the quality of the live streaming according to the parameters passed in from the web side.
-     * @param liveParam Live streaming parameters.
+     * 根据从web端传入的参数设置直播的质量。
+     * @param liveParam 直播参数
      * @return
      */
     @PostMapping("/streams/update")
@@ -96,4 +139,4 @@
         return liveStreamService.liveLensChange(liveParam);
     }
 
-}
\ No newline at end of file
+}

--
Gitblit v1.9.3