From 9dde95485ec15c50c84f54a5bb41fb98e29bc1c9 Mon Sep 17 00:00:00 2001 From: cxh Date: Wed, 12 Nov 2025 17:47:56 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E7=AE=A1=E7=BA=BF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/rtmp_manager.cpp | 72 +++++++++++++++++++++----------------------- 1 file changed, 35 insertions(+), 37 deletions(-) diff --git a/src/rtmp_manager.cpp b/src/rtmp_manager.cpp index 390410a..5aff5b1 100644 --- a/src/rtmp_manager.cpp +++ b/src/rtmp_manager.cpp @@ -1,4 +1,4 @@ -// rtsp_manager.cpp +// rtmp_manager.cpp #include "rtmp_manager.hpp" #include @@ -12,14 +12,14 @@ #include // ========== 工具函数 ========== -static bool device_exists(const std::string &path) +static bool device_exists(const std::string& path) { struct stat st; return (stat(path.c_str(), &st) == 0); } // 动态获取指定网卡 IPv4 地址 -static std::string get_ip_address(const std::string &ifname) +static std::string get_ip_address(const std::string& ifname) { struct ifaddrs *ifaddr, *ifa; char ip[INET_ADDRSTRLEN] = {0}; @@ -36,7 +36,7 @@ static std::string get_ip_address(const std::string &ifname) if (ifa->ifa_addr->sa_family == AF_INET && ifname == ifa->ifa_name) { - void *addr = &((struct sockaddr_in *)ifa->ifa_addr)->sin_addr; + void* addr = &((struct sockaddr_in*)ifa->ifa_addr)->sin_addr; if (inet_ntop(AF_INET, addr, ip, sizeof(ip))) { freeifaddrs(ifaddr); @@ -60,10 +60,10 @@ void RTMPManager::init() LOG_INFO("[RTMP] GStreamer initialized."); } -std::string RTMPManager::make_key(const std::string &name) { return name + "_main"; } +std::string RTMPManager::make_key(const std::string& name) { return name + "_main"; } // ========== 创建推流管线 ========== -GstElement *RTMPManager::create_pipeline(const Camera &cam) +GstElement* RTMPManager::create_pipeline(const Camera& cam) { const int width = cam.width; const int height = cam.height; @@ -74,28 +74,26 @@ GstElement *RTMPManager::create_pipeline(const Camera &cam) const std::string app = "camera"; const std::string location = "rtmp://127.0.0.1:1935/" + app + "/" + stream_name + "?vhost=live"; - // 给关键元素起名字：src, vc, enc, par, mux, sink, tee std::string pipeline_str = "v4l2src name=src device=" + cam.device + - " ! video/x-raw,width=" + std::to_string(width) + ",height=" + std::to_string(height) + + " ! " + "video/x-raw,format=NV12,width=" + + std::to_string(width) + ",height=" + std::to_string(height) + ",framerate=" + std::to_string(fps) + - "/1 " - "! videoconvert name=vc ! video/x-raw,format=NV12 " - "! tee name=t " - "t. ! queue " - "! mpph264enc name=enc bps=" + + "/1 ! " + "tee name=t " + "t. ! queue max-size-buffers=5 leaky=downstream ! " + "mpph264enc bps=" + std::to_string(bitrate) + " gop=" + std::to_string(fps) + - " " - "! h264parse name=par " - "! flvmux name=mux streamable=true " + " rc-mode=cbr ! " + "h264parse ! flvmux streamable=true name=mux " "audiotestsrc wave=silence ! audioconvert ! audioresample ! voaacenc ! aacparse ! mux. " - "mux. ! rtmpsink name=sink location=\"" + + "mux. ! rtmpsink location=\"" + location + - "\" sync=false " - // 预留第二路 tee 分支（以后加叠加/探测都行），先丢掉 + "\" sync=false async=false " "t. ! queue ! fakesink sync=false"; - GError *error = nullptr; - GstElement *pipeline = gst_parse_launch(pipeline_str.c_str(), &error); + GError* error = nullptr; + GstElement* pipeline = gst_parse_launch(pipeline_str.c_str(), &error); if (error) { LOG_ERROR(std::string("[RTMP] Pipeline creation failed: ") + error->message); @@ -106,7 +104,7 @@ GstElement *RTMPManager::create_pipeline(const Camera &cam) } // ========== 主推流循环 ========== -void RTMPManager::stream_loop(Camera cam, StreamContext *ctx) +void RTMPManager::stream_loop(Camera cam, StreamContext* ctx) { const std::string key = make_key(cam.name); while (ctx->running) @@ -123,7 +121,7 @@ void RTMPManager::stream_loop(Camera cam, StreamContext *ctx) } // 创建管线 - GstElement *pipeline = create_pipeline(cam); + GstElement* pipeline = create_pipeline(cam); if (!pipeline) { ctx->status.running = false; @@ -132,22 +130,22 @@ void RTMPManager::stream_loop(Camera cam, StreamContext *ctx) continue; } gst_element_set_name(pipeline, key.c_str()); - GstBus *bus = gst_element_get_bus(pipeline); + GstBus* bus = gst_element_get_bus(pipeline); // 在 videoconvert 的 src pad 上挂 probe 判断是否有帧流过 bool got_frame = false; { - GstElement *vc = gst_bin_get_by_name(GST_BIN(pipeline), "vc"); + GstElement* vc = gst_bin_get_by_name(GST_BIN(pipeline), "vc"); if (vc) { - GstPad *pad = gst_element_get_static_pad(vc, "src"); + GstPad* pad = gst_element_get_static_pad(vc, "src"); if (pad) { gst_pad_add_probe( pad, GST_PAD_PROBE_TYPE_BUFFER, - [](GstPad *, GstPadProbeInfo *, gpointer user_data) -> GstPadProbeReturn + [](GstPad*, GstPadProbeInfo*, gpointer user_data) -> GstPadProbeReturn { - *static_cast(user_data) = true; + *static_cast(user_data) = true; return GST_PAD_PROBE_OK; }, &got_frame, nullptr); @@ -228,7 +226,7 @@ void RTMPManager::stream_loop(Camera cam, StreamContext *ctx) } // 监听错误/EOS - GstMessage *msg = gst_bus_timed_pop_filtered(bus, 200 * GST_MSECOND, + GstMessage* msg = gst_bus_timed_pop_filtered(bus, 200 * GST_MSECOND, (GstMessageType)(GST_MESSAGE_ERROR | GST_MESSAGE_EOS)); if (!msg) continue; @@ -237,7 +235,7 @@ void RTMPManager::stream_loop(Camera cam, StreamContext *ctx) { case GST_MESSAGE_ERROR: { - GError *err = nullptr; + GError* err = nullptr; gst_message_parse_error(msg, &err, nullptr); ctx->status.running = false; ctx->status.last_error = err ? err->message : "GStreamer error"; @@ -283,7 +281,7 @@ void RTMPManager::start_all() std::lock_guard lock(streams_mutex); int delay_ms = 0; - for (auto &cam : g_app_config.cameras) + for (auto& cam : g_app_config.cameras) { auto key = make_key(cam.name); if (streams.find(key) != streams.end()) @@ -310,20 +308,20 @@ void RTMPManager::start_all() void RTMPManager::stop_all() { std::lock_guard lock(streams_mutex); - for (auto &kv : streams) kv.second->running.store(false); - for (auto &kv : streams) + for (auto& kv : streams) kv.second->running.store(false); + for (auto& kv : streams) if (kv.second->thread.joinable()) kv.second->thread.join(); streams.clear(); } -bool RTMPManager::is_streaming(const std::string &cam_name) +bool RTMPManager::is_streaming(const std::string& cam_name) { std::lock_guard lock(streams_mutex); auto it = streams.find(make_key(cam_name)); return (it != streams.end() && it->second->status.running); } -std::string RTMPManager::get_stream_url(const std::string &cam_name) +std::string RTMPManager::get_stream_url(const std::string& cam_name) { std::string ip = get_ip_address("enP2p33s0"); if (ip.empty()) ip = "127.0.0.1"; @@ -337,7 +335,7 @@ std::vector RTMPManager::get_all_channels_status() for (size_t i = 0; i < g_app_config.cameras.size(); ++i) { - const auto &cam = g_app_config.cameras[i]; + const auto& cam = g_app_config.cameras[i]; auto key = make_key(cam.name); ChannelInfo ch; @@ -349,7 +347,7 @@ std::vector RTMPManager::get_all_channels_status() auto it = streams.find(key); if (it != streams.end()) { - auto &status = it->second->status; + auto& status = it->second->status; ch.running = status.running; if (status.running) {