From 773bb07d85fae83856189ddb10687c9d681c0546 Mon Sep 17 00:00:00 2001 From: cxh Date: Fri, 14 Nov 2025 14:13:14 +0800 Subject: [PATCH] 1 --- include/record_manager.hpp | 14 ++++-- src/main.cpp | 38 +-------------- src/record_manager.cpp | 97 ++++++++++++++++++++++++++++++-------- 3 files changed, 88 insertions(+), 61 deletions(-) diff --git a/include/record_manager.hpp b/include/record_manager.hpp index b918fc3..2dbc05e 100644 --- a/include/record_manager.hpp +++ b/include/record_manager.hpp @@ -28,7 +28,7 @@ struct RecordSegment class RecordManager { public: - explicit RecordManager(const std::string& base_dir); + explicit RecordManager(const std::string& srs_record_cfg_path); // 执行一次全扫描 void scanAll(); @@ -42,11 +42,15 @@ class RecordManager static int64_t toMsTimestamp(const std::string& s); static std::string toReadable(int64_t ms); - // 调试打印全部索引 - void dumpIndex() const; - private: - std::string base_dir_; // /sata/record/ + // === SRS 配置相关 === + std::string srs_record_cfg_path_; + std::string record_dir_; // /sata/record/ + int dvr_duration_sec_ = 60; // 从配置文件读取，默认 60 + + bool loadSrsConfig(); // 解析配置文件，填充 record_dir_ 和 dvr_duration_sec_ + + // === 文件列表索引 === std::unordered_map> index_; // 解析文件名得到 start_ms 和 end_ms diff --git a/src/main.cpp b/src/main.cpp index 34739cc..8c7206d 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -36,43 +36,7 @@ int main() Logger::init(get_executable_dir_file_path("logs"), 7); LOG_INFO("[MAIN] ===== Vehicle Video Service Starting ====="); - std::string base = "/sata/record/"; - RecordManager rm(base); - - std::cout << "[RecordManager] scanning " << base << " ..." << std::endl; - - rm.scanAll(); - - std::cout << "\n=== 全部扫描结果 ===\n"; - - // 遍历所有 stream - rm.dumpIndex(); - - // 测试一次查询：你可以换成真实时间戳 - std::cout << "\n=== 测试 querySegments() ===\n"; - - // 随便选一个时间区间，比如 2025-11-13 10:23:00 ~ 11:00:00 - // 你可以换成真实值 - int64_t start = RecordManager::toMsTimestamp("2025-11-13 10:00:00"); - int64_t end = RecordManager::toMsTimestamp("2025-11-13 14:30:00"); - - auto segments = rm.querySegments("AHD1_main", start, end); - - std::cout << "找到录像段数量 = " << segments.size() << "\n"; - - for (auto& seg : segments) - { - std::cout << "\n--- Segment " << seg.index << " ---\n"; - std::cout << "segmentId = " << seg.segment_id << "\n"; - std::cout << "start_ms = " << seg.start_ms << "\n"; - std::cout << "end_ms = " << seg.end_ms << "\n"; - std::cout << "files = " << seg.files.size() << "\n"; - - for (auto& f : seg.files) - { - std::cout << " file: " << f.path << "\n [" << f.start_ms << " ~ " << f.end_ms << "]\n"; - } - } + RecordManager rm("/home/aiec/srs/conf/kun_record.conf"); // try // { diff --git a/src/record_manager.cpp b/src/record_manager.cpp index 86df248..33bde84 100644 --- a/src/record_manager.cpp +++ b/src/record_manager.cpp @@ -6,7 +6,84 @@ namespace fs = std::filesystem; -RecordManager::RecordManager(const std::string& base_dir) : base_dir_(base_dir) {} +// +// 构造函数：自动加载 SRS 配置 +// +RecordManager::RecordManager(const std::string& srs_record_cfg_path) : srs_record_cfg_path_(srs_record_cfg_path) +{ + std::cout << "[RecordManager] Loading SRS config: " << srs_record_cfg_path_ << "\n"; + + bool ok = loadSrsConfig(); + if (!ok) + { + std::cout << "[RecordManager] ERROR loading SRS config\n"; + } + else + { + std::cout << "[RecordManager] SRS config loaded successfully.\n"; + std::cout << " record_dir_ = " << record_dir_ << "\n"; + std::cout << " dvr_duration_sec = " << dvr_duration_sec_ << "\n"; + } +} + +// +// 解析 SRS DVR 配置，提取 record_dir_ 与 dvr_duration_sec_ +// +bool RecordManager::loadSrsConfig() +{ + std::ifstream ifs(srs_record_cfg_path_); + if (!ifs.is_open()) + { + std::cerr << "[RecordManager] Failed to open SRS config: " << srs_record_cfg_path_ << "\n"; + return false; + } + + std::string line; + std::string dvr_path; + while (std::getline(ifs, line)) + { + // 去掉前后空白 + line.erase(0, line.find_first_not_of(" \t")); + if (line.empty() || line[0] == '#') continue; + + if (line.find("dvr_path") != std::string::npos) + { + auto pos = line.find('/'); + auto semicolon = line.find(';'); + if (pos != std::string::npos && semicolon != std::string::npos) + { + dvr_path = line.substr(pos, semicolon - pos); + } + } + else if (line.find("dvr_duration") != std::string::npos) + { + int sec = 0; + if (sscanf(line.c_str(), "dvr_duration %d", &sec) == 1) + { + dvr_duration_sec_ = sec; + } + } + } + + if (!dvr_path.empty()) + { + // 例子：/sata/record/[stream]/[2006]-[01]-[02]/... + // → record_dir_ = /sata/record/ + auto pos = dvr_path.find("[stream]"); + if (pos != std::string::npos) + { + record_dir_ = dvr_path.substr(0, pos); + } + } + + if (record_dir_.empty()) + { + std::cerr << "[RecordManager] ERROR: cannot parse dvr_path.\n"; + return false; + } + + return true; +} // // 扫描所有文件 @@ -223,21 +300,3 @@ std::string RecordManager::toReadable(int64_t ms) strftime(buf, sizeof(buf), "%Y-%m-%d %H:%M:%S", tm); return buf; } - -void RecordManager::dumpIndex() const -{ - std::cout << "\n=== dumpIndex() ===\n"; - - for (const auto& kv : index_) - { - const auto& stream = kv.first; - const auto& files = kv.second; - - std::cout << "\n>>> Stream = " << stream << ", file_count = " << files.size() << "\n"; - - for (const auto& f : files) - { - std::cout << " - " << f.path << "\n start_ms=" << f.start_ms << ", end_ms=" << f.end_ms << "\n"; - } - } -}