From 99647c423596c86a314cb43edb5a6fb70817cca6 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Wed, 01 Apr 2026 11:17:57 +0800
Subject: [PATCH] 优化
---
ard-work/src/main/java/com/ard/work/component/websocket/handler/UavStreamManager.java | 21 +++++++++++++++++++++
1 files changed, 21 insertions(+), 0 deletions(-)
diff --git a/ard-work/src/main/java/com/ard/work/component/websocket/handler/UavStreamManager.java b/ard-work/src/main/java/com/ard/work/component/websocket/handler/UavStreamManager.java
index 40b795b..0cc0861 100644
--- a/ard-work/src/main/java/com/ard/work/component/websocket/handler/UavStreamManager.java
+++ b/ard-work/src/main/java/com/ard/work/component/websocket/handler/UavStreamManager.java
@@ -2,6 +2,7 @@
import com.alibaba.fastjson2.JSON;
+import com.ard.vtdu.config.MediamtxSixHeConfig;
import com.ard.vtdu.domain.ArdVtdu;
import com.ard.work.component.websocket.utils.WebSocketUtil;
import com.ard.work.domain.message.BaseMessage;
@@ -27,6 +28,8 @@
@Resource
private ITyService tyService;
+ @Resource
+ private MediamtxSixHeConfig mediamtxSixHeConfig;
// 盒子拉流地址缓存 key:boxSn value: webrtcUrl
public static final ConcurrentMap<String, String> BOX_STREAM_PULL_MAP = new ConcurrentHashMap<>();
@@ -123,6 +126,24 @@
if (boxLiveUrl == null) {
throw new RuntimeException("未获取到盒子直播地址：" + boxSn);
}
+ // 2. 获取原始 URL 字符串
+ String originalUrl = boxLiveUrl.getRtc();
+ if (mediamtxSixHeConfig.getEnable() && originalUrl != null && !originalUrl.isEmpty()) {
+ // --- 开始替换逻辑 ---
+ // 配置你的外网地址
+ String targetHost = mediamtxSixHeConfig.getPlayHost() + ":" + mediamtxSixHeConfig.getWebrtcPort();
+ // 正则解释：
+ // (^https?://) -> 捕获组1：匹配 http:// 或 https://
+ // ([^/:]+) -> 捕获组2：匹配 IP 地址或域名 (非 / 和 : 的字符)
+ // (:\d+)? -> 捕获组3：匹配 :端口号 (可选)
+ // (/.*) -> 捕获组4：匹配后面的路径参数
+ String regex = "(^https?://)([^/:]+)(:\\d+)?(/.*)";
+ // 替换为：协议 + 外网IP:端口 + 路径
+ // $1 代表协议, $4 代表路径
+ String newUrl = originalUrl.replaceAll(regex, "$1" + targetHost + "$4");
+ // 将新 URL 设置回去
+ boxLiveUrl.setRtc(newUrl);
+ }
boxLiveUrl.setBoxSn(boxSn);
return boxLiveUrl;
}
--
Gitblit v1.9.3