From 22af8b1c29515317280099df02470c68856f9021 Mon Sep 17 00:00:00 2001
From: aix <vip_xiaobin810@163.com>
Date: Tue, 02 Jul 2024 21:23:08 +0800
Subject: [PATCH] 航线飞行

---
 src/main/java/com/dji/sample/manage/service/impl/LiveStreamServiceImpl.java |   72 ++++++++++++++++++++++++++++++++++-
 1 files changed, 69 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/dji/sample/manage/service/impl/LiveStreamServiceImpl.java b/src/main/java/com/dji/sample/manage/service/impl/LiveStreamServiceImpl.java
index 7404fdb..6f630a2 100644
--- a/src/main/java/com/dji/sample/manage/service/impl/LiveStreamServiceImpl.java
+++ b/src/main/java/com/dji/sample/manage/service/impl/LiveStreamServiceImpl.java
@@ -1,5 +1,6 @@
 package com.dji.sample.manage.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.dji.sample.common.error.LiveErrorEnum;
 import com.dji.sample.common.model.ResponseResult;
 import com.dji.sample.component.mqtt.model.CommonTopicResponse;
@@ -7,7 +8,9 @@
 import com.dji.sample.component.mqtt.service.IMessageSenderService;
 import com.dji.sample.component.redis.RedisConst;
 import com.dji.sample.component.redis.RedisOpsUtils;
+import com.dji.sample.manage.dao.IDeviceSetMapper;
 import com.dji.sample.manage.model.dto.*;
+import com.dji.sample.manage.model.entity.DeviceSetEntity;
 import com.dji.sample.manage.model.enums.DeviceDomainEnum;
 import com.dji.sample.manage.model.enums.LiveStreamMethodEnum;
 import com.dji.sample.manage.model.enums.LiveUrlTypeEnum;
@@ -16,6 +19,8 @@
 import com.dji.sample.manage.model.receiver.CapacityDeviceReceiver;
 import com.dji.sample.manage.model.receiver.LiveCapacityReceiver;
 import com.dji.sample.manage.service.*;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -24,6 +29,8 @@
 import java.lang.reflect.Field;
 import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 import java.util.stream.Collectors;
 
 import static com.dji.sample.component.mqtt.model.TopicConst.*;
@@ -42,6 +49,8 @@
 
     @Autowired
     private IDeviceService deviceService;
+    @Autowired
+    private IDeviceSetMapper deviceSetMapper;
 
     @Autowired
     private IWorkspaceService workspaceService;
@@ -51,7 +60,6 @@
 
     @Autowired
     private IDeviceRedisService deviceRedisService;
-
     @Override
     public List<CapacityDeviceDTO> getLiveCapacity(String workspaceId,String sn) {
 
@@ -172,7 +180,63 @@
         }
 
         return ResponseResult.success(live);
+
     }
+
+    @Override
+    public ResponseResult liveAddress(String deviceSn,String deviceId) {
+       DeviceSetEntity deviceSet= deviceSetMapper.selectOne(new LambdaQueryWrapper<DeviceSetEntity>()
+                .eq(DeviceSetEntity::getDeviceSn,deviceSn)
+                .eq(DeviceSetEntity::getDeviceId,deviceId)
+        );
+        String workspaceId=getIdBySn(deviceSn);
+        List<CapacityDeviceDTO> dto=getLiveCapacity(workspaceId,deviceSn);
+        String vedioId=deviceSn+"/165-0-7/normal-0";
+        String url="https://"+deviceSet.getServerIp()+"/zb/rtp/"+deviceSet.getAgentId()+"_"+deviceSet.getChannel()+".live.flv";
+        LiveTypeDTO liveParam=new LiveTypeDTO();
+        liveParam.setUrl(url);
+        liveParam.setUrlType(3);
+        liveParam.setVideoId(vedioId);
+        liveParam.setVideoQuality(0);
+        ResponseResult responseResult = this.checkBeforeLive(liveParam.getVideoId());
+        if (ResponseResult.CODE_SUCCESS != responseResult.getCode()) {
+            return responseResult;
+        }
+
+        DeviceDTO data = (DeviceDTO)responseResult.getData();
+        String respTopic = THING_MODEL_PRE + PRODUCT +
+                data.getDeviceSn() + SERVICES_SUF;
+
+        //获取返回结果
+        ServiceReply receiveReply = this.publishLiveStart(respTopic, liveParam);
+        System.out.println(receiveReply.getResult());
+        LiveDTO live = new LiveDTO();
+        live.setUrl(url);
+        //相机已经在直播中,请勿重复开启直播
+        if(receiveReply.getResult() == 513003) {
+            return ResponseResult.success(live);
+        }
+        return ResponseResult.success(live);
+    }
+
+
+        public static String getVedioId(String data,String sn) {
+            return findDeviceBySn(data, sn);
+        }
+
+        public static String findDeviceBySn(String dtoListString, String snToFind) {
+            String regex = "CapacityDeviceDTO\\(sn=" + snToFind + ", .*? index=([\\w\\-]+).*? index=([\\w\\-]+).*?\\)";
+            Pattern pattern = Pattern.compile(regex);
+            Matcher matcher = pattern.matcher(dtoListString);
+
+            if (matcher.find()) {
+                String index1 = matcher.group(1);
+                String index2 = matcher.group(2);
+                return snToFind + "/" + index1 + "/" + index2;
+            }
+
+            return null; // 如果未找到匹配的sn,则返回null或者适当的默认值
+        }
 
     @Override
     public ResponseResult liveStop(String videoId) {
@@ -180,10 +244,8 @@
         if (responseResult.getCode() != 0) {
             return responseResult;
         }
-
         //thing/product/{gateway_sn}/services
         String respTopic = THING_MODEL_PRE + PRODUCT + responseResult.getData().getDeviceSn() + SERVICES_SUF;
-
         videoId = videoId.replace("_","/");
 
         ServiceReply receiveReply = this.publishLiveStop(respTopic, videoId);
@@ -351,6 +413,10 @@
         return messageSender.publishWithReply(ServiceReply.class, topic, response);
     }
 
+    private String getIdBySn(String dockSn){
+        DeviceSetEntity entity=deviceSetMapper.selectOne(new LambdaQueryWrapper<DeviceSetEntity>().eq(DeviceSetEntity::getDeviceSn,dockSn));
+        return entity.getWorkspaceId();
+    }
     /**
      * Send a message to the pilot via mqtt to set quality.
      * @param respTopic

--
Gitblit v1.9.3