package com.ard.sample.manage.service.impl;
|
|
import com.ard.common.core.redis.RedisConst;
|
import com.ard.common.core.redis.RedisOpsUtils;
|
import com.ard.sample.common.model.ResponseResult;
|
import com.ard.sample.component.websocket.enums.MessageType;
|
import com.ard.sample.component.websocket.handler.DockWebSocketHandler;
|
import com.ard.sample.component.websocket.handler.LiveStreamHandler;
|
import com.ard.sample.manage.model.dto.DeviceDTO;
|
import com.ard.sample.manage.model.dto.LiveDTO;
|
import com.ard.sample.manage.model.dto.LiveTypeDTO;
|
import com.ard.sample.manage.service.IDeviceService;
|
import com.ard.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.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Component;
|
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.TimeUnit;
|
|
@Slf4j
|
@Component
|
public class DockOnlineServiceImpl {
|
|
@Autowired
|
private IDeviceService _deviceService;
|
|
@Autowired
|
private DockWebSocketHandler _dockWebSocketHandler;
|
|
@Autowired
|
protected ObjectMapper _objectMapper;
|
|
@Autowired
|
private LiveStreamHandler _liveStreamHandler;
|
|
@Autowired
|
private ILiveStreamService _liveStreamService;
|
|
/**
|
* 定时任务，每2秒取一次机场设备的在线状态，如已离线更新状态
|
* */
|
@Scheduled(cron = "0/2 * * * * ?")
|
public void cronOnlineTask() {
|
List<DeviceDTO> devicesList = _deviceService.getDevicesTopoForWeb();
|
for (DeviceDTO device : devicesList) {
|
//String key = "online:" + device.getDeviceSn();
|
//System.out.println("在线状态："+RedisOpsUtils.checkExist(key));
|
// 更新机场设备状态："3"-"设备已离线"，
|
if (!_deviceService.checkDeviceOnline(device.getDeviceSn())) {
|
_deviceService.updateWorkType(device.getDeviceSn(), 3L);
|
RedisOpsUtils.set(RedisConst.OSD_STATUS_PREFIX + device.getDeviceSn(), 3);
|
RedisOpsUtils.del(RedisConst.WEBSOCKET_VIDEO_ADDRESS_PREFIX + device.getDeviceSn());
|
RedisOpsUtils.del(RedisConst.COVER_STATE_PREFIX + device.getDeviceSn());
|
clearLiveRetry(device.getDeviceSn());
|
} else {
|
tryStartLiveStreaming(device.getDeviceSn());
|
}
|
|
// 机场1：未绑定飞行器时 childDeviceSn 为空，跳过，避免无效的在线判断和起流
|
String childDeviceSn = device.getChildDeviceSn();
|
if (childDeviceSn == null || childDeviceSn.isEmpty()) {
|
continue;
|
}
|
if (!_deviceService.checkDeviceOnline(childDeviceSn)) {
|
_deviceService.updateWorkType(childDeviceSn, 6L);
|
RedisOpsUtils.del(RedisConst.WEBSOCKET_VIDEO_ADDRESS_PREFIX + childDeviceSn);
|
RedisOpsUtils.del(RedisConst.COVER_STATE_PREFIX + childDeviceSn);
|
clearLiveRetry(childDeviceSn);
|
} else if (isDronePoweredOff(childDeviceSn)) {
|
// 飞行器舱内关机（6）或开机中（7）时相机尚未就绪，起流必然失败（如 513006），直接跳过
|
clearLiveRetry(childDeviceSn);
|
} else {
|
tryStartLiveStreaming(childDeviceSn);
|
}
|
}
|
}
|
|
/**
|
* 按退避策略尝试开启直播。
|
* 原实现每 2 秒对未就绪设备执行一次“停-开”，开机阶段会刷出大量 513006/613011 日志；
|
* 这里失败后按 2s、4s、8s、16s、30s 递增间隔重试。
|
*/
|
private void tryStartLiveStreaming(String sn) {
|
if (!canRetryLive(sn)) {
|
return;
|
}
|
markLiveRetry(sn, StartLiveStreaming(sn));
|
}
|
|
/**
|
* 飞行器是否处于关机或开机中状态。
|
* 6-舱内关机（机场 OSD 写入），7-开机中（下发 drone_open 时写入）。
|
*/
|
private boolean isDronePoweredOff(String childDeviceSn) {
|
Object status = RedisOpsUtils.get(RedisConst.OSD_STATUS_PREFIX + childDeviceSn);
|
if (status == null) {
|
return false;
|
}
|
String value = String.valueOf(status);
|
return "6".equals(value) || "7".equals(value);
|
}
|
|
/**
|
* 连续失败次数与下次允许重试的时间戳。
|
*/
|
private static class LiveRetryState {
|
private int failures;
|
private long nextRetryTime;
|
}
|
|
private static final long LIVE_RETRY_BASE_DELAY_MS = 2000L;
|
private static final long LIVE_RETRY_MAX_DELAY_MS = 30000L;
|
|
private final Map<String, LiveRetryState> liveRetryStates = new ConcurrentHashMap<>();
|
|
private boolean canRetryLive(String sn) {
|
LiveRetryState state = liveRetryStates.get(sn);
|
return state == null || System.currentTimeMillis() >= state.nextRetryTime;
|
}
|
|
private void markLiveRetry(String sn, boolean success) {
|
if (success) {
|
liveRetryStates.remove(sn);
|
return;
|
}
|
LiveRetryState state = liveRetryStates.computeIfAbsent(sn, k -> new LiveRetryState());
|
state.failures++;
|
long delay = Math.min(LIVE_RETRY_BASE_DELAY_MS << Math.min(state.failures - 1, 4), LIVE_RETRY_MAX_DELAY_MS);
|
state.nextRetryTime = System.currentTimeMillis() + delay;
|
log.info("SN: {}-直播启动失败（第 {} 次），{} 秒后重试", sn, state.failures, delay / 1000);
|
}
|
|
private void clearLiveRetry(String sn) {
|
liveRetryStates.remove(sn);
|
}
|
|
/**
|
* 定时任务，每1秒读一次redis的机场设备的状态，并发送到websocket客户端
|
* */
|
//@Scheduled(cron = "0/1 * * * * ?")
|
@Scheduled(initialDelay = 2, fixedRate = 3, timeUnit = TimeUnit.SECONDS)
|
public void sendDeviceStatus(){
|
// 测试 机场oss读取文件
|
// URL url = _ossService.getObjectUrl("kmz","/2025/11/25/新建航点航线(1)_20251125161322A001.kmz");
|
// log.info("机场任务航线文件url地址：{}", url.toString());
|
// 发送设备状态到websocket客户端
|
List<DeviceDTO> devicesList = _deviceService.getDevicesTopoForWeb();
|
for (DeviceDTO device : devicesList) {
|
String dockStatus = "";
|
String droneStatus = "";
|
// 判断redis中是否存在设备状态数据
|
if(RedisOpsUtils.checkExist(RedisConst.OSD_STATUS_PREFIX + device.getDeviceSn())){
|
dockStatus = String.valueOf(RedisOpsUtils.get(RedisConst.OSD_STATUS_PREFIX + device.getDeviceSn()));
|
}
|
if(RedisOpsUtils.checkExist(RedisConst.OSD_STATUS_PREFIX + device.getChildDeviceSn())){
|
droneStatus = String.valueOf(RedisOpsUtils.get(RedisConst.OSD_STATUS_PREFIX + device.getChildDeviceSn()));
|
}
|
String json = getJSONString(MessageType.STATE.getType(), device.getDeviceSn(), dockStatus, droneStatus);
|
|
// 发送给websocket客户端
|
_dockWebSocketHandler.sendMessageToSubscribers(device.getDeviceSn(), MessageType.STATE.getType(), json, "");
|
// log.info("发送设备状态{}到websocket客户端：{}", MessageType.STATE.getType(), json);
|
}
|
}
|
|
public String getJSONString(String messageType, String dockSn, String dockStatus, String droneStatus){
|
Map<String, Object> json = new HashMap<>();
|
json.put("code", 200);
|
json.put("gateway", dockSn);
|
json.put("type", messageType);
|
json.put("message", "success");
|
|
Map<String, String> data = new HashMap<>();
|
data.put("dockstate", dockStatus);
|
data.put("dronestate", droneStatus);
|
|
json.put("data", data);
|
|
// 将Map转换为JSON字符串
|
String jsonString = "";
|
try{
|
jsonString = _objectMapper.writeValueAsString(json);
|
}
|
catch (Exception e){
|
e.printStackTrace();
|
}
|
|
return jsonString;
|
}
|
|
/**
|
* 开启直播（后台定时任务调用）。
|
*
|
* @return true-已存在可用地址或起流成功；false-起流失败
|
*/
|
public boolean StartLiveStreaming(String sn) {
|
//调用开始直播
|
ResponseResult rlive = null;
|
String resultLive = "";
|
try {
|
// 获取直播参数
|
LiveTypeDTO param = _liveStreamHandler.getLiveParam(sn);
|
// 判断param是否为空，不为空则返回正确的直播参数，为null则未获取直播参数，不为空则继续操作
|
if (param == null) {
|
log.info("SN: {}-未获取到直播参数param为null，无法开启直播！", sn);
|
return false;
|
}
|
// 如果存在直播则读取直播地址返回
|
if (RedisOpsUtils.checkExist(RedisConst.WEBSOCKET_VIDEO_ADDRESS_PREFIX + sn)) {
|
resultLive = String.valueOf(RedisOpsUtils.get(RedisConst.WEBSOCKET_VIDEO_ADDRESS_PREFIX + sn));
|
// log.info("SN: {}-redis中存在相机的播放地址信息，直接返回地址！", sn);
|
return true;
|
} else {
|
// log.info("SN: {}-redis中未找到相机的播放地址信息！停止直播后再开启直播", sn);
|
_liveStreamHandler.StopLiveStreaming(sn);
|
// 如果不存在直播，则开始直播，并返回直播地址
|
rlive = _liveStreamService.liveStart(param);
|
if (rlive.getCode() == 0) {
|
// 直播成功后，在redis中存一个标记，将videoid保留，留给停止直播使用
|
log.info("SN: {}-开启直播成功！", sn);
|
rlive.setGateway(sn);
|
rlive.setType(MessageType.LIVESTREAM.getType());
|
resultLive = _objectMapper.writeValueAsString(rlive);
|
RedisOpsUtils.set(RedisConst.WEBSOCKET_VIDEO_ADDRESS_PREFIX + sn, resultLive);
|
LiveDTO live = (LiveDTO) rlive.getData();
|
Boolean isWeb = _deviceService.updateWebrtcUrl(sn,live.getWebrtcUrl());
|
// if (isWeb) {
|
// log.info("SN: {}-更新webrtc地址成功！", sn);
|
// }
|
// 定时任务是后台自动起流，前端并没有发订阅，必须主动把地址推给已订阅该设备的页面；
|
// 否则飞行器起飞后相机就绪、后台起流成功，页面却收不到地址，只能刷新后才出画面
|
_dockWebSocketHandler.sendMessageToSubscribers(sn, MessageType.LIVESTREAM.getType(), resultLive, "");
|
return true;
|
}
|
log.info("SN: {}-开启直播失败！返回值：{}", sn, rlive);
|
}
|
} catch (Exception e) {
|
log.info("订时任务推流-SN: {}-开启直播失败！原因：{}", sn, e.getMessage());
|
e.printStackTrace();
|
}
|
return false;
|
}
|
}
|
