cxh/kunlang_video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
cxh 2025-09-10 13:03:16 +08:00
parent ac50a5c01c
commit b0ef86224a
1 changed files with 71 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

108
src/main.cpp
View File

@ -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;
}