tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Clear download counters 60 seconds after all downloads are completed or when user enters Downloads section.

Browse Source
This commit is contained in:
levlam 2022-03-02 16:16:24 +03:00
parent 0ea961ebea
commit 4acfae167f
1 changed files with 34 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
td/telegram/DownloadManager.cpp
View File

@ -172,6 +172,7 @@ class DownloadManagerImpl final : public DownloadManager {
void search(string query, bool only_active, bool only_completed, string offset, int32 limit, void search(string query, bool only_active, bool only_completed, string offset, int32 limit,
Promise<td_api::object_ptr<td_api::foundFileDownloads>> promise) final { Promise<td_api::object_ptr<td_api::foundFileDownloads>> promise) final {
clear_counters();
return do_search(std::move(query), only_active, only_completed, std::move(offset), limit, std::move(promise), return do_search(std::move(query), only_active, only_completed, std::move(offset), limit, std::move(promise),
Unit{}); Unit{});
} }
@ -364,7 +365,12 @@ class DownloadManagerImpl final : public DownloadManager {
auto serialized_counter = G()->td_db()->get_binlog_pmc()->get("dlds_counter"); auto serialized_counter = G()->td_db()->get_binlog_pmc()->get("dlds_counter");
if (!serialized_counter.empty()) { if (!serialized_counter.empty()) {
log_event_parse(sent_counters_, serialized_counter).ensure(); log_event_parse(sent_counters_, serialized_counter).ensure();
callback_->update_counters(sent_counters_); if (sent_counters_.downloaded_size == sent_counters_.total_size || sent_counters_.total_size == 0) {
G()->td_db()->get_binlog_pmc()->erase("dlds_counter");
sent_counters_ = Counters();
} else {
callback_->update_counters(sent_counters_);
}
} }
auto downloads_in_kv = G()->td_db()->get_binlog_pmc()->prefix_get("dlds#"); auto downloads_in_kv = G()->td_db()->get_binlog_pmc()->prefix_get("dlds#");
@ -468,6 +474,27 @@ class DownloadManagerImpl final : public DownloadManager {
try_start(); try_start();
} }
void timeout_expired() final {
clear_counters();
}
void clear_counters() {
CHECK(counters_ == sent_counters_);
if (counters_.downloaded_size != counters_.total_size || counters_.total_size == 0) {
return;
}
for (auto &it : files_) {
with_file_info(*it.second, [&](auto &file_info) {
if (!file_info.is_paused) {
file_info.is_counted = false;
}
});
}
counters_ = Counters();
update_counters();
}
void tear_down() final { void tear_down() final {
callback_.reset(); callback_.reset();
} }
@ -497,6 +524,12 @@ class DownloadManagerImpl final : public DownloadManager {
if (counters_ == sent_counters_) { if (counters_ == sent_counters_) {
return; return;
} }
CHECK(counters_.total_size >= 0);
CHECK(counters_.total_count >= 0);
CHECK(counters_.downloaded_size >= 0);
if (counters_.downloaded_size == counters_.total_size && counters_.total_size != 0) {
set_timeout_in(60.0);
}
sent_counters_ = counters_; sent_counters_ = counters_;
callback_->update_counters(counters_); callback_->update_counters(counters_);
} }