67 lines
1.9 KiB
C++
67 lines
1.9 KiB
C++
// 看门狗 Monitor 入口
|
||
//
|
||
// 用法：vdi-kiosk-monitor -- <主进程命令行>
|
||
// 例： vdi-kiosk-monitor -- /home/armbian/vdi-terminal/bin/kiosk-shell
|
||
//
|
||
// 对应会议纪要要求：
|
||
// "必须部署独立的 Monitor（看门狗）程序监控主进程状态，
|
||
// 主进程假死或崩溃时由 Monitor 执行 kill 操作并重启"
|
||
|
||
#include <cstdio>
|
||
#include <string>
|
||
|
||
#include "monitor.h"
|
||
|
||
int main(int argc, char** argv) {
|
||
if (argc < 2) {
|
||
std::fprintf(stderr, "用法: vdi-kiosk-monitor [--verbose] -- <主进程命令行>\n");
|
||
return 2;
|
||
}
|
||
|
||
int idx = 1;
|
||
vdi::MonitorConfig config;
|
||
config.verbose = true;
|
||
|
||
while (idx < argc && std::string(argv[idx]).rfind("--", 0) == 0) {
|
||
const std::string opt = argv[idx];
|
||
if (opt == "--") {
|
||
++idx;
|
||
break;
|
||
}
|
||
if (opt == "--quiet") {
|
||
config.verbose = false;
|
||
} else if (opt == "--verbose") {
|
||
config.verbose = true;
|
||
} else if (opt == "--heartbeat-timeout" && idx + 1 < argc) {
|
||
config.heartbeatTimeoutMs = std::atoi(argv[++idx]);
|
||
} else if (opt == "--check-interval" && idx + 1 < argc) {
|
||
config.checkIntervalMs = std::atoi(argv[++idx]);
|
||
} else if (opt == "--max-restart" && idx + 1 < argc) {
|
||
config.maxRestartAttempts = std::atoi(argv[++idx]);
|
||
} else if (opt == "--heartbeat-file" && idx + 1 < argc) {
|
||
config.heartbeatFile = argv[++idx];
|
||
} else {
|
||
// 未知选项视为主进程命令行的起点
|
||
break;
|
||
}
|
||
++idx;
|
||
}
|
||
|
||
if (idx >= argc) {
|
||
std::fprintf(stderr, "缺少主进程命令行\n");
|
||
return 2;
|
||
}
|
||
|
||
// 拼出主进程命令行
|
||
std::string command;
|
||
for (int i = idx; i < argc; ++i) {
|
||
if (i > idx) {
|
||
command += " ";
|
||
}
|
||
command += argv[i];
|
||
}
|
||
|
||
vdi::Monitor monitor(config);
|
||
return monitor.run(command);
|
||
}
|