zhangsheng/WLG
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refine codes.

Browse Source
This commit is contained in:
pandx 2025-01-09 11:57:32 +08:00
parent 74f6d55773
commit 087c260942
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scheduler/upgrade_cfg.cpp
View File

@ -12,6 +12,14 @@ int UpgradeCfg::ReadCfg(std::map<int, UpgradeInfo> &upgrade) {
zlog_info(zbt, "[UpgradeCfg] no such file"); zlog_info(zbt, "[UpgradeCfg] no such file");
return 0; return 0;
} }
upgrade_file.seekg(0, std::ios::end);
std::streampos file_size = upgrade_file.tellg();
if (file_size > 256000) {
zlog_error(zct, "upgrade file exception, will remove it");
ClearCfg();
return 1;
}
upgrade_file.seekg(0, std::ios::beg);
Json::Reader reader; Json::Reader reader;
Json::Value root; Json::Value root;
@ -51,6 +59,11 @@ int UpgradeCfg::ReadCfg(std::map<int, UpgradeInfo> &upgrade) {
} }
int UpgradeCfg::WriteCfg(std::map<int, UpgradeInfo> &upgrade) { int UpgradeCfg::WriteCfg(std::map<int, UpgradeInfo> &upgrade) {
if (upgrade.size() == 0) {
ClearCfg();
return 0;
}
Json::Value root; Json::Value root;
for (auto item : upgrade) { for (auto item : upgrade) {
Json::Value upgrade_item; Json::Value upgrade_item;