This commit is contained in:
cxh 2025-10-15 16:21:25 +08:00
parent ecc8840688
commit 6b8b3e6987

View File

@ -85,6 +85,9 @@ RTMPManager::StreamResultInfo RTMPManager::start_camera(const Camera &cam, Strea
res.url = get_stream_url(cam.name, type); res.url = get_stream_url(cam.name, type);
std::string key = make_stream_key(cam.name, type); std::string key = make_stream_key(cam.name, type);
try
{
auto ctx = std::make_unique<StreamContext>(); auto ctx = std::make_unique<StreamContext>();
StreamContext *ctx_ptr = ctx.get(); StreamContext *ctx_ptr = ctx.get();
@ -114,6 +117,15 @@ RTMPManager::StreamResultInfo RTMPManager::start_camera(const Camera &cam, Strea
{ {
res.result = 1; res.result = 1;
res.reason = "Start timeout"; res.reason = "Start timeout";
// 超时时停止线程
stop_camera(cam.name, type);
}
}
catch (const std::exception &e)
{
LOG_ERROR("[RTMP] Exception in start_camera for " + key + ": " + e.what());
res.result = 1;
res.reason = "Start exception: " + std::string(e.what());
} }
return res; return res;
@ -165,46 +177,75 @@ void RTMPManager::stream_loop(Camera cam, StreamType type, StreamContext *ctx)
std::string key = make_stream_key(cam.name, type); std::string key = make_stream_key(cam.name, type);
LOG_INFO("[RTMP] Stream loop started for " + key); LOG_INFO("[RTMP] Stream loop started for " + key);
GstElement *pipeline = create_pipeline(cam, type); GstElement *pipeline = nullptr;
if (!pipeline) GstBus *bus = nullptr;
{
res.result = 1;
res.reason = "Failed to create pipeline";
ctx->start_result.set_value(res);
return;
}
GstBus *bus = gst_element_get_bus(pipeline);
gst_element_set_state(pipeline, GST_STATE_PLAYING);
bool first_frame_received = false;
bool start_result_set = false; bool start_result_set = false;
try
{
// 确保在异常情况下也能设置结果
auto set_start_result = [&](const StreamResultInfo &r) auto set_start_result = [&](const StreamResultInfo &r)
{ {
if (!start_result_set) if (!start_result_set)
{
try
{ {
ctx->start_result.set_value(r); ctx->start_result.set_value(r);
start_result_set = true; start_result_set = true;
} }
catch (const std::future_error &e)
{
LOG_ERROR("[RTMP] Failed to set start result: " + std::string(e.what()));
}
}
}; };
// 创建管道
pipeline = create_pipeline(cam, type);
if (!pipeline)
{
res.result = 1;
res.reason = "Failed to create pipeline";
set_start_result(res);
return;
}
bus = gst_element_get_bus(pipeline);
if (!bus)
{
res.result = 1;
res.reason = "Failed to get pipeline bus";
set_start_result(res);
return;
}
// 设置管道状态
GstStateChangeReturn state_ret = gst_element_set_state(pipeline, GST_STATE_PLAYING);
if (state_ret == GST_STATE_CHANGE_FAILURE)
{
res.result = 1;
res.reason = "Failed to set pipeline to playing state";
set_start_result(res);
return;
}
bool first_frame_received = false;
auto start_time = std::chrono::steady_clock::now(); auto start_time = std::chrono::steady_clock::now();
while (true) while (true)
{ {
// 检查是否应该停止
{ {
std::lock_guard<std::mutex> lock(streams_mutex); std::lock_guard<std::mutex> lock(streams_mutex);
auto it = streams.find(key); auto it = streams.find(key);
if (it == streams.end() || !it->second->running.load()) break; if (it == streams.end() || !it->second->running.load())
{
LOG_INFO("[RTMP] Stream loop stopping for " + key);
break;
}
} }
GstMessage *msg = // 检查第一帧超时
gst_bus_timed_pop_filtered(bus, 100 * GST_MSECOND,
static_cast<GstMessageType>(GST_MESSAGE_ERROR | GST_MESSAGE_EOS |
GST_MESSAGE_STATE_CHANGED | GST_MESSAGE_ELEMENT));
// 超时未收到第一帧
if (!first_frame_received) if (!first_frame_received)
{ {
auto elapsed = std::chrono::steady_clock::now() - start_time; auto elapsed = std::chrono::steady_clock::now() - start_time;
@ -213,10 +254,17 @@ void RTMPManager::stream_loop(Camera cam, StreamType type, StreamContext *ctx)
res.result = 1; res.result = 1;
res.reason = "No frames received within timeout"; res.reason = "No frames received within timeout";
set_start_result(res); set_start_result(res);
LOG_WARN("[RTMP] First frame timeout for " + key);
break; break;
} }
} }
// 等待GStreamer消息
GstMessage *msg = gst_bus_timed_pop_filtered(
bus, 100 * GST_MSECOND,
static_cast<GstMessageType>(GST_MESSAGE_ERROR | GST_MESSAGE_EOS | GST_MESSAGE_STATE_CHANGED |
GST_MESSAGE_ELEMENT));
if (!msg) continue; if (!msg) continue;
switch (GST_MESSAGE_TYPE(msg)) switch (GST_MESSAGE_TYPE(msg))
@ -230,17 +278,21 @@ void RTMPManager::stream_loop(Camera cam, StreamType type, StreamContext *ctx)
res.result = 1; res.result = 1;
res.reason = "Pipeline error: " + err_msg; res.reason = "Pipeline error: " + err_msg;
set_start_result(res); set_start_result(res);
LOG_ERROR("[RTMP] Pipeline error for " + key + ": " + err_msg);
if (err) g_error_free(err); if (err) g_error_free(err);
if (debug) g_free(debug); if (debug) g_free(debug);
break; gst_message_unref(msg);
goto cleanup;
} }
case GST_MESSAGE_EOS: case GST_MESSAGE_EOS:
res.result = 1; res.result = 1;
res.reason = "EOS received"; res.reason = "EOS received";
set_start_result(res); set_start_result(res);
break; LOG_INFO("[RTMP] EOS received for " + key);
gst_message_unref(msg);
goto cleanup;
case GST_MESSAGE_STATE_CHANGED: case GST_MESSAGE_STATE_CHANGED:
// pipeline 播放状态变化,不算成功 // 可以添加状态变化的日志用于调试
break; break;
case GST_MESSAGE_ELEMENT: case GST_MESSAGE_ELEMENT:
if (!first_frame_received) if (!first_frame_received)
@ -249,6 +301,7 @@ void RTMPManager::stream_loop(Camera cam, StreamType type, StreamContext *ctx)
res.result = 0; res.result = 0;
res.reason = "First frame received, started OK"; res.reason = "First frame received, started OK";
set_start_result(res); set_start_result(res);
LOG_INFO("[RTMP] First frame received for " + key);
} }
break; break;
default: default:
@ -257,12 +310,72 @@ void RTMPManager::stream_loop(Camera cam, StreamType type, StreamContext *ctx)
gst_message_unref(msg); gst_message_unref(msg);
} }
}
catch (const std::exception &e)
{
LOG_ERROR("[RTMP] Exception in stream_loop for " + key + ": " + e.what());
res.result = 1;
res.reason = "Exception: " + std::string(e.what());
if (!start_result_set)
{
try
{
ctx->start_result.set_value(res);
start_result_set = true;
}
catch (const std::future_error &)
{
// 忽略future错误
}
}
}
catch (...)
{
LOG_ERROR("[RTMP] Unknown exception in stream_loop for " + key);
res.result = 1;
res.reason = "Unknown exception";
if (!start_result_set)
{
try
{
ctx->start_result.set_value(res);
start_result_set = true;
}
catch (const std::future_error &)
{
// 忽略future错误
}
}
}
cleanup:
// 清理GStreamer资源
if (pipeline)
{
gst_element_set_state(pipeline, GST_STATE_NULL); gst_element_set_state(pipeline, GST_STATE_NULL);
if (bus) gst_object_unref(bus);
gst_object_unref(pipeline); gst_object_unref(pipeline);
}
if (bus)
{
gst_object_unref(bus);
}
// 清理 streams // 确保start_result被设置
if (!start_result_set)
{
try
{
res.result = 1;
res.reason = "Stream loop exited unexpectedly";
ctx->start_result.set_value(res);
}
catch (const std::future_error &)
{
// 忽略future错误
}
}
// 从streams中移除
{ {
std::lock_guard<std::mutex> lock(streams_mutex); std::lock_guard<std::mutex> lock(streams_mutex);
streams.erase(key); streams.erase(key);