zrj
2024-08-17 5577a7280cd97315fbf9d1ef1b52aa0f569c0081
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,11 +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
@@ -47,6 +44,44 @@
        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.
     * 从数据库中获取当前工作区中所有无人机的实时性能数据。