forked from uav/ard-uav-server
ard-sample/src/main/java/com/ard/sample/manage/service/impl/DockOnlineServiceImpl.java
@@ -1,22 +1,29 @@ package com.ard.sample.manage.service.impl; import com.alibaba.fastjson2.JSON; 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.ArrayList; 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 { @@ -29,28 +36,119 @@ @Autowired protected ObjectMapper _objectMapper; @Autowired private LiveStreamHandler _liveStreamHandler; @Autowired private ILiveStreamService _liveStreamService; /** * 定时任务，每5秒取一次机场设备的在线状态，如已离线更新状态 * 定时任务，每2秒取一次机场设备的在线状态，如已离线更新状态 * */ @Scheduled(cron = "0/5 * * * * ?") @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())){ if (!_deviceService.checkDeviceOnline(device.getDeviceSn())) { _deviceService.updateWorkType(device.getDeviceSn(), 3L); RedisOpsUtils.setWithExpire(RedisConst.OSD_STATUS_PREFIX + device.getDeviceSn(), 3, RedisConst.DEVICE_ALIVE_SECOND); 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); } } } /** * 定时任务，每5秒读一次redis的机场设备的状态，并发送到websocket客户端 * 按退避策略尝试开启直播。 * 原实现每 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/5 * * * * ?") //@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) { @@ -61,12 +159,13 @@ 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()));; 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); _dockWebSocketHandler.sendMessageToSubscribers(device.getDeviceSn(), MessageType.STATE.getType(), json, ""); // log.info("发送设备状态{}到websocket客户端：{}", MessageType.STATE.getType(), json); } } @@ -94,4 +193,57 @@ 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; } }
