From b0ef86224a8da5d802dbbfa0117bb737ee4bb138 Mon Sep 17 00:00:00 2001
From: cxh <zxwl@56.com>
Date: Wed, 10 Sep 2025 13:03:16 +0800
Subject: [PATCH] first commit

---
 src/main.cpp | 108 +++++++++++++++++++++++++++++++++------------------
 1 file changed, 71 insertions(+), 37 deletions(-)

diff --git a/src/main.cpp b/src/main.cpp
index 942697e..786fb29 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -9,6 +9,10 @@
 #include <unistd.h> // write, STDOUT_FILENO
 #include <chrono>
 
+// 可通过这些开关快速启用/禁用线程进行调试
+constexpr bool ENABLE_RTSP_THREAD = true;  // 设置为 false 禁用 RTSP 线程
+constexpr bool ENABLE_MQTT_THREAD = false; // 设置为 false 禁用 MQTT 线程
+
 std::atomic<bool> g_running(true);
 
 static void minimal_signal_handler(int signum)
@@ -55,15 +59,33 @@ int main()
     std::thread rtsp_thread;
     std::thread mqtt_thread;
 
-    std::thread rtsp_thread([&]()
-                            {
-        RTSPManager::start(g_app_config.cameras);
-        rtsp_thread_exited.store(true, std::memory_order_relaxed); });
+    // 启动 RTSP 线程（如果启用）
+    if (ENABLE_RTSP_THREAD)
+    {
+        rtsp_thread = std::thread([&]()
+                                  {
+            RTSPManager::start(g_app_config.cameras);
+            rtsp_thread_exited.store(true, std::memory_order_relaxed); });
+        LOG_INFO("[MAIN] RTSP thread started");
+    }
+    else
+    {
+        LOG_INFO("[MAIN] RTSP thread disabled by build-time toggle");
+    }
 
-    // std::thread mqtt_thread([&]()
-    //                         {
-    //     mqtt_client_thread_func();
-    //     mqtt_thread_exited.store(true, std::memory_order_relaxed); });
+    // 启动 MQTT 线程（如果启用）
+    if (ENABLE_MQTT_THREAD)
+    {
+        mqtt_thread = std::thread([&]()
+                                  {
+            mqtt_client_thread_func();
+            mqtt_thread_exited.store(true, std::memory_order_relaxed); });
+        LOG_INFO("[MAIN] MQTT thread started");
+    }
+    else
+    {
+        LOG_INFO("[MAIN] MQTT thread disabled by build-time toggle");
+    }
 
     // 等待退出信号
     while (g_running.load(std::memory_order_relaxed))
@@ -76,54 +98,66 @@ int main()
     const auto poll_interval = std::chrono::milliseconds(100);
     auto deadline = std::chrono::steady_clock::now() + max_wait;
 
-    // 等 RTSP 线程退出
-    while (!rtsp_thread_exited.load(std::memory_order_relaxed) &&
-           std::chrono::steady_clock::now() < deadline)
+    if (ENABLE_RTSP_THREAD)
     {
-        std::this_thread::sleep_for(poll_interval);
-    }
-    if (rtsp_thread.joinable())
-    {
-        if (rtsp_thread_exited.load(std::memory_order_relaxed))
+        while (!rtsp_thread_exited.load(std::memory_order_relaxed) &&
+               std::chrono::steady_clock::now() < deadline)
         {
-            rtsp_thread.join();
-            LOG_INFO("[MAIN] RTSP thread finished and joined.");
+            std::this_thread::sleep_for(poll_interval);
         }
-        else
+
+        if (rtsp_thread.joinable())
         {
-            LOG_WARN("[MAIN] RTSP thread did not exit within timeout.");
+            if (rtsp_thread_exited.load(std::memory_order_relaxed))
+            {
+                rtsp_thread.join();
+                LOG_INFO("[MAIN] RTSP thread finished and joined.");
+            }
+            else
+            {
+                LOG_WARN("[MAIN] RTSP thread did not exit within timeout.");
+            }
         }
     }
 
-    // 等 MQTT 线程退出
+    // 重置 MQTT 线程等待的截止时间
     deadline = std::chrono::steady_clock::now() + max_wait;
-    while (!mqtt_thread_exited.load(std::memory_order_relaxed) &&
-           std::chrono::steady_clock::now() < deadline)
+
+    if (ENABLE_MQTT_THREAD)
     {
-        std::this_thread::sleep_for(poll_interval);
-    }
-    if (mqtt_thread.joinable())
-    {
-        if (mqtt_thread_exited.load(std::memory_order_relaxed))
+        while (!mqtt_thread_exited.load(std::memory_order_relaxed) &&
+               std::chrono::steady_clock::now() < deadline)
         {
-            mqtt_thread.join();
-            LOG_INFO("[MAIN] MQTT thread finished and joined.");
+            std::this_thread::sleep_for(poll_interval);
         }
-        else
+
+        if (mqtt_thread.joinable())
         {
-            LOG_WARN("[MAIN] MQTT thread did not exit within timeout.");
+            if (mqtt_thread_exited.load(std::memory_order_relaxed))
+            {
+                mqtt_thread.join();
+                LOG_INFO("[MAIN] MQTT thread finished and joined.");
+            }
+            else
+            {
+                LOG_WARN("[MAIN] MQTT thread did not exit within timeout.");
+            }
         }
     }
 
-    // 如果某些线程仍未退出，强制结束进程（不会跑全局析构）
-    if ((rtsp_thread.joinable() && !rtsp_thread_exited.load(std::memory_order_relaxed)) ||
-        (mqtt_thread.joinable() && !mqtt_thread_exited.load(std::memory_order_relaxed)))
+    // 如果有线程仍未退出，则强制终止
+    bool any_failed = false;
+    if (ENABLE_RTSP_THREAD && rtsp_thread.joinable() && !rtsp_thread_exited.load(std::memory_order_relaxed))
+        any_failed = true;
+    if (ENABLE_MQTT_THREAD && mqtt_thread.joinable() && !mqtt_thread_exited.load(std::memory_order_relaxed))
+        any_failed = true;
+
+    if (any_failed)
     {
         LOG_ERROR("[MAIN] Threads did not exit in time. Forcing immediate termination.");
-        _exit(1); // 强制退出（不会调用 atexit handlers）
+        _exit(1);
     }
 
     LOG_INFO("[MAIN] Program exited cleanly.");
-
     return 0;
 }
\ No newline at end of file