package com.dji.sample.manage.controller;
|
|
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;
|
import com.dji.sample.manage.service.ILiveStreamService;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.integration.annotation.ServiceActivator;
|
import org.springframework.messaging.MessageHeaders;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.io.IOException;
|
import java.util.List;
|
|
/**
|
* @author sean.zhou
|
* @version 0.1
|
* @date 2021/11/19
|
*/
|
|
@RestController
|
@Slf4j
|
@RequestMapping("${url.manage.prefix}${url.manage.version}/live")
|
public class LiveStreamController {
|
|
@Autowired
|
private ILiveStreamService liveStreamService;
|
|
@Autowired
|
private ObjectMapper mapper;
|
|
/**
|
* Analyze the live streaming capabilities of drones.
|
* This data is necessary if drones are required for live streaming.
|
* @param liveCapacity the capacity of drone and dock
|
*/
|
@ServiceActivator(inputChannel = ChannelName.INBOUND_STATE_CAPACITY)
|
public void stateCapacity(LiveCapacityReceiver liveCapacity, MessageHeaders headers) {
|
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 workspaceId
|
* @return live capability
|
*/
|
@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(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
|
*/
|
@PostMapping("/streams/start")
|
public ResponseResult liveStart(@RequestBody LiveTypeDTO liveParam) {
|
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
|
*/
|
@PostMapping("/streams/stop")
|
public ResponseResult liveStop(@RequestBody LiveTypeDTO liveParam) {
|
return liveStreamService.liveStop(liveParam.getVideoId());
|
}
|
|
/**
|
* 根据从web端传入的参数设置直播的质量。
|
* @param liveParam 直播参数
|
* @return
|
*/
|
@PostMapping("/streams/update")
|
public ResponseResult liveSetQuality(@RequestBody LiveTypeDTO liveParam) {
|
return liveStreamService.liveSetQuality(liveParam);
|
}
|
|
@PostMapping("/streams/switch")
|
public ResponseResult liveLensChange(@RequestBody LiveTypeDTO liveParam) {
|
return liveStreamService.liveLensChange(liveParam);
|
}
|
|
}
|
