tdlight/td/telegram/DownloadManager.cpp

840 lines
29 KiB
C++
Raw Normal View History

2022-02-23 19:34:09 +01:00
//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
2022-02-25 14:18:23 +01:00
#include "td/telegram/DownloadManager.h"
#include "td/telegram/files/FileId.hpp"
2022-02-26 19:55:12 +01:00
#include "td/telegram/files/FileSourceId.hpp"
2022-02-25 23:18:07 +01:00
#include "td/telegram/Global.h"
2022-02-26 19:55:12 +01:00
#include "td/telegram/logevent/LogEvent.h"
2022-02-25 23:18:07 +01:00
#include "td/telegram/TdDb.h"
2022-03-11 22:50:04 +01:00
#include "td/telegram/TdParameters.h"
#include "td/actor/MultiPromise.h"
#include "td/actor/PromiseFuture.h"
2022-02-26 19:55:12 +01:00
#include "td/utils/algorithm.h"
#include "td/utils/FlatHashMap.h"
2022-03-09 21:01:10 +01:00
#include "td/utils/FlatHashSet.h"
#include "td/utils/Hints.h"
2022-02-28 16:25:07 +01:00
#include "td/utils/logging.h"
#include "td/utils/misc.h"
#include "td/utils/Slice.h"
#include "td/utils/SliceBuilder.h"
#include "td/utils/tl_helpers.h"
2022-02-25 23:18:07 +01:00
2022-02-28 16:25:07 +01:00
#include <algorithm>
#include <functional>
#include <limits>
#include <set>
2022-02-28 16:25:07 +01:00
2022-02-23 19:34:09 +01:00
namespace td {
2022-02-25 14:18:23 +01:00
struct FileDownloadInDatabase {
2022-02-26 19:55:12 +01:00
int64 download_id{};
FileId file_id;
2022-02-26 19:55:12 +01:00
FileSourceId file_source_id;
2022-02-28 16:25:07 +01:00
int32 priority{};
int32 created_at{};
int32 completed_at{};
2022-02-26 21:54:01 +01:00
bool is_paused{};
2022-02-26 19:55:12 +01:00
template <class StorerT>
void store(StorerT &storer) const {
BEGIN_STORE_FLAGS();
STORE_FLAG(is_paused);
END_STORE_FLAGS();
td::store(download_id, storer);
td::store(file_id, storer);
2022-02-26 19:55:12 +01:00
td::store(file_source_id, storer);
td::store(priority, storer);
2022-02-26 21:54:01 +01:00
td::store(created_at, storer);
td::store(completed_at, storer);
2022-02-26 19:55:12 +01:00
}
template <class ParserT>
void parse(ParserT &parser) {
BEGIN_PARSE_FLAGS();
PARSE_FLAG(is_paused);
END_PARSE_FLAGS();
td::parse(download_id, parser);
td::parse(file_id, parser);
2022-02-26 19:55:12 +01:00
td::parse(file_source_id, parser);
td::parse(priority, parser);
2022-02-26 21:54:01 +01:00
td::parse(created_at, parser);
td::parse(completed_at, parser);
2022-02-26 19:55:12 +01:00
}
};
2022-02-23 19:34:09 +01:00
class DownloadManagerImpl final : public DownloadManager {
public:
explicit DownloadManagerImpl(unique_ptr<Callback> callback) : callback_(std::move(callback)) {
}
void start_up() final {
2022-03-03 15:59:24 +01:00
init();
2022-02-23 19:34:09 +01:00
}
2022-03-03 16:26:25 +01:00
void after_get_difference() final {
2022-08-04 17:49:05 +02:00
load_database_files("after_get_difference");
2022-03-03 16:26:25 +01:00
}
void toggle_is_paused(FileId file_id, bool is_paused, Promise<Unit> promise) final {
2022-08-04 17:49:05 +02:00
TRY_STATUS_PROMISE(promise, check_is_active("toggle_is_paused"));
TRY_RESULT_PROMISE(promise, file_info_ptr, get_file_info(file_id));
2022-02-28 16:25:07 +01:00
toggle_is_paused(*file_info_ptr, is_paused);
promise.set_value(Unit());
2022-02-23 19:34:09 +01:00
}
void toggle_all_is_paused(bool is_paused, Promise<Unit> promise) final {
2022-08-04 17:49:05 +02:00
TRY_STATUS_PROMISE(promise, check_is_active("toggle_all_is_paused"));
2022-02-26 19:55:12 +01:00
2022-07-23 13:52:42 +02:00
vector<FileId> to_toggle;
2022-02-26 19:55:12 +01:00
for (auto &it : files_) {
2022-07-23 13:52:42 +02:00
FileInfo &file_info = *it.second;
if (!is_completed(file_info) && is_paused != file_info.is_paused) {
to_toggle.push_back(file_info.file_id);
}
}
for (auto file_id : to_toggle) {
auto r_file_info_ptr = get_file_info(file_id);
if (r_file_info_ptr.is_ok()) {
toggle_is_paused(*r_file_info_ptr.ok(), is_paused);
}
2022-02-23 19:34:09 +01:00
}
promise.set_value(Unit());
}
void remove_file(FileId file_id, FileSourceId file_source_id, bool delete_from_cache, Promise<Unit> promise) final {
2022-09-05 16:54:03 +02:00
promise.set_result(remove_file_impl(file_id, file_source_id, delete_from_cache, "remove_file"));
2022-02-23 19:34:09 +01:00
}
void remove_file_if_finished(FileId file_id) final {
remove_file_if_finished_impl(file_id).ignore();
}
void remove_all_files(bool only_active, bool only_completed, bool delete_from_cache, Promise<Unit> promise) final {
2022-08-04 17:49:05 +02:00
TRY_STATUS_PROMISE(promise, check_is_active("remove_all_files"));
2022-02-28 16:25:07 +01:00
vector<FileId> to_remove;
2022-02-26 19:55:12 +01:00
for (auto &it : files_) {
FileInfo &file_info = *it.second;
if (only_active && is_completed(file_info)) {
2022-02-26 19:55:12 +01:00
continue;
}
if (only_completed && !is_completed(file_info)) {
2022-02-26 19:55:12 +01:00
continue;
2022-02-25 13:11:10 +01:00
}
2022-02-26 19:55:12 +01:00
to_remove.push_back(file_info.file_id);
}
for (auto file_id : to_remove) {
2022-09-05 16:54:03 +02:00
remove_file_impl(file_id, {}, delete_from_cache, "remove_all_files");
2022-02-23 19:34:09 +01:00
}
promise.set_value(Unit());
2022-02-23 19:34:09 +01:00
}
void add_file(FileId file_id, FileSourceId file_source_id, string search_text, int8 priority,
Promise<td_api::object_ptr<td_api::file>> promise) final {
2022-08-04 17:49:05 +02:00
TRY_STATUS_PROMISE(promise, check_is_active("add_file"));
2022-02-26 19:55:12 +01:00
2022-09-05 16:54:03 +02:00
remove_file_impl(file_id, {}, false, "add_file");
2022-02-26 19:55:12 +01:00
2022-02-28 16:25:07 +01:00
auto download_id = next_download_id();
auto file_info = make_unique<FileInfo>();
file_info->download_id = download_id;
file_info->file_id = file_id;
file_info->file_source_id = file_source_id;
file_info->is_paused = false;
file_info->priority = priority;
file_info->created_at = G()->unix_time();
file_info->need_save_to_database = true;
2022-02-28 16:25:07 +01:00
2022-02-28 17:09:41 +01:00
add_file_info(std::move(file_info), search_text);
promise.set_value(callback_->get_file_object(file_id));
2022-02-23 19:34:09 +01:00
}
void change_search_text(FileId file_id, FileSourceId file_source_id, string search_text) final {
if (!is_search_inited_) {
return;
}
2022-08-04 17:49:05 +02:00
if (check_is_active("change_search_text").is_error()) {
return;
}
auto r_file_info_ptr = get_file_info(file_id, file_source_id);
if (r_file_info_ptr.is_error()) {
return;
}
auto &file_info = *r_file_info_ptr.ok();
hints_.add(file_info.download_id, search_text.empty() ? string(" ") : search_text);
}
void hints_synchronized(Result<Unit>) {
if (G()->close_flag()) {
return;
}
LOG(INFO) << "DownloadManager: hints are synchronized";
is_search_inited_ = true;
}
2022-02-25 14:18:23 +01:00
void search(string query, bool only_active, bool only_completed, string offset, int32 limit,
Promise<td_api::object_ptr<td_api::foundFileDownloads>> promise) final {
return do_search(std::move(query), only_active, only_completed, std::move(offset), limit, std::move(promise),
Unit{});
}
void do_search(string query, bool only_active, bool only_completed, string offset, int32 limit,
Promise<td_api::object_ptr<td_api::foundFileDownloads>> promise, Result<Unit>) {
TRY_STATUS_PROMISE(promise, G()->close_status());
2022-08-04 17:49:05 +02:00
TRY_STATUS_PROMISE(promise, check_is_active("do_search"));
if (!is_search_inited_) {
Promise<Unit> lock;
if (load_search_text_multipromise_.promise_count() == 0) {
load_search_text_multipromise_.add_promise(
promise_send_closure(actor_id(this), &DownloadManagerImpl::hints_synchronized));
load_search_text_multipromise_.set_ignore_errors(true);
lock = load_search_text_multipromise_.get_promise();
prepare_hints();
}
load_search_text_multipromise_.add_promise(promise_send_closure(actor_id(this), &DownloadManagerImpl::do_search,
std::move(query), only_active, only_completed,
std::move(offset), limit, std::move(promise)));
lock.set_value(Unit());
return;
}
2022-02-28 16:25:07 +01:00
if (limit <= 0) {
return promise.set_error(Status::Error(400, "Limit must be positive"));
}
2022-02-26 21:54:01 +01:00
int64 offset_int64 = std::numeric_limits<int64>::max();
if (!offset.empty()) {
auto r_offset = to_integer_safe<int64>(offset);
if (r_offset.is_error()) {
return promise.set_error(Status::Error(400, "Invalid offset"));
}
offset_int64 = r_offset.move_as_ok();
2022-02-23 19:34:09 +01:00
}
auto download_ids = hints_.search(query, 10000, true).second;
FileCounters counters;
td::remove_if(download_ids, [&](int64 download_id) {
2022-02-26 21:54:01 +01:00
auto r = get_file_info(download_id);
2022-02-28 16:25:07 +01:00
CHECK(r.is_ok());
2022-02-26 21:54:01 +01:00
auto &file_info = *r.ok();
2022-03-03 14:30:16 +01:00
if (is_completed(file_info)) {
counters.completed_count++;
2022-03-03 14:30:16 +01:00
if (only_active) {
return true;
}
} else {
counters.active_count++;
2022-03-03 14:30:16 +01:00
if (file_info.is_paused) {
counters.paused_count++;
2022-03-03 14:30:16 +01:00
}
if (only_completed) {
return true;
}
2022-02-26 19:55:12 +01:00
}
2022-02-28 16:25:07 +01:00
if (download_id >= offset_int64) {
return true;
}
2022-02-26 19:55:12 +01:00
return false;
});
std::sort(download_ids.begin(), download_ids.end(), std::greater<>());
2022-03-03 14:30:16 +01:00
if (static_cast<int32>(download_ids.size()) > limit) {
download_ids.resize(limit);
2022-02-26 19:55:12 +01:00
}
auto file_downloads = transform(download_ids, [&](int64 download_id) {
on_file_viewed(download_id);
auto it = files_.find(download_id);
CHECK(it != files_.end());
const FileInfo &file_info = *it->second;
return callback_->get_file_download_object(file_info.file_id, file_info.file_source_id, file_info.created_at,
file_info.completed_at, file_info.is_paused);
2022-02-26 19:55:12 +01:00
});
td::remove_if(file_downloads, [](const auto &file_download) { return file_download->message_ == nullptr; });
string next_offset;
if (!download_ids.empty()) {
next_offset = to_string(download_ids.back());
}
promise.set_value(td_api::make_object<td_api::foundFileDownloads>(counters.get_downloaded_file_counts_object(),
std::move(file_downloads), next_offset));
2022-02-23 19:34:09 +01:00
}
2022-03-02 16:29:49 +01:00
void update_file_download_state(FileId internal_file_id, int64 downloaded_size, int64 size, int64 expected_size,
bool is_paused) final {
2022-03-03 16:26:25 +01:00
if (!callback_ || !is_database_loaded_) {
2022-02-23 19:34:09 +01:00
return;
}
2022-03-03 16:26:25 +01:00
LOG(INFO) << "Update file download state for file " << internal_file_id << " of size " << size << '/'
<< expected_size << " to downloaded_size = " << downloaded_size << " and is_paused = " << is_paused;
2022-02-26 21:54:01 +01:00
auto r_file_info_ptr = get_file_info_by_internal(internal_file_id);
if (r_file_info_ptr.is_error()) {
2022-02-23 19:34:09 +01:00
return;
}
2022-02-26 21:54:01 +01:00
auto &file_info = *r_file_info_ptr.ok();
if (file_info.link_token != get_link_token()) {
LOG(INFO) << "Ignore update_file_download_state because of outdated link_token";
2022-03-01 15:39:05 +01:00
return;
}
2022-02-26 21:54:01 +01:00
bool need_update = false;
2022-02-26 21:54:01 +01:00
with_file_info(file_info, [&](FileInfo &file_info) {
file_info.size = size;
file_info.expected_size = expected_size;
2022-03-02 16:29:49 +01:00
file_info.downloaded_size = downloaded_size;
if (is_paused && file_info.is_paused != is_paused) {
2022-02-26 21:54:01 +01:00
file_info.is_paused = is_paused;
file_info.need_save_to_database = true;
need_update = true;
2022-02-26 21:54:01 +01:00
}
});
if (is_search_inited_ && need_update) {
callback_->update_file_changed(file_info.file_id, file_info.completed_at, file_info.is_paused, file_counters_);
}
2022-02-23 19:34:09 +01:00
}
void update_file_viewed(FileId file_id, FileSourceId file_source_id) final {
2022-03-03 16:26:25 +01:00
if (unviewed_completed_download_ids_.empty() || !callback_ || !is_database_loaded_) {
return;
}
2022-04-04 18:08:47 +02:00
LOG(INFO) << "File " << file_id << " was viewed from " << file_source_id;
auto r_file_info_ptr = get_file_info(file_id, file_source_id);
if (r_file_info_ptr.is_error()) {
return;
}
auto &file_info = *r_file_info_ptr.ok();
on_file_viewed(file_info.download_id);
}
2022-02-23 19:34:09 +01:00
private:
unique_ptr<Callback> callback_;
struct FileInfo {
2022-02-26 19:55:12 +01:00
int64 download_id{};
2022-02-28 16:25:07 +01:00
FileId file_id;
FileId internal_file_id;
FileSourceId file_source_id;
int8 priority{};
2022-02-23 19:34:09 +01:00
bool is_paused{};
bool is_counted{};
mutable bool is_registered{};
mutable bool need_save_to_database{};
2022-02-23 19:34:09 +01:00
int64 size{};
int64 expected_size{};
2022-02-23 19:34:09 +01:00
int64 downloaded_size{};
2022-02-26 21:54:01 +01:00
int32 created_at{};
int32 completed_at{};
uint64 link_token{};
2022-02-23 19:34:09 +01:00
};
2022-02-26 19:55:12 +01:00
FlatHashMap<FileId, int64, FileIdHash> by_file_id_;
FlatHashMap<FileId, int64, FileIdHash> by_internal_file_id_;
FlatHashMap<int64, unique_ptr<FileInfo>> files_;
std::set<int64> completed_download_ids_;
FlatHashSet<int64> unviewed_completed_download_ids_;
2022-02-26 19:55:12 +01:00
Hints hints_;
2022-02-23 19:34:09 +01:00
Counters counters_;
2022-02-26 19:55:12 +01:00
Counters sent_counters_;
FileCounters file_counters_;
2022-08-04 17:49:05 +02:00
const char *database_loading_source_ = nullptr;
2022-03-03 15:59:24 +01:00
bool is_inited_{false};
2022-03-03 16:26:25 +01:00
bool is_database_loaded_{false};
bool is_search_inited_{false};
2022-02-26 19:55:12 +01:00
int64 max_download_id_{0};
uint64 last_link_token_{0};
MultiPromiseActor load_search_text_multipromise_{"LoadFileSearchTextMultiPromiseActor"};
2022-02-26 19:55:12 +01:00
int64 next_download_id() {
2022-02-28 16:25:07 +01:00
return ++max_download_id_;
2022-02-26 19:55:12 +01:00
}
static bool is_completed(const FileInfo &file_info) {
return file_info.completed_at != 0;
2022-02-26 19:55:12 +01:00
}
2022-02-28 16:25:07 +01:00
2022-03-02 16:29:49 +01:00
static int64 get_file_size(const FileInfo &file_info) {
return file_info.size == 0 ? max(file_info.downloaded_size + 1, file_info.expected_size) : file_info.size;
}
static bool is_database_enabled() {
return G()->parameters().use_message_db;
}
2022-02-28 16:25:07 +01:00
static string pmc_key(const FileInfo &file_info) {
return PSTRING() << "dlds#" << file_info.download_id;
2022-02-26 19:55:12 +01:00
}
2022-03-03 16:26:25 +01:00
void sync_with_database(const FileInfo &file_info) {
if (!file_info.need_save_to_database) {
2022-02-26 19:55:12 +01:00
return;
}
file_info.need_save_to_database = false;
2022-02-28 16:25:07 +01:00
if (!is_database_enabled()) {
return;
}
2022-03-01 14:29:08 +01:00
LOG(INFO) << "Saving to download database file " << file_info.file_id << '/' << file_info.internal_file_id
<< " with is_paused = " << file_info.is_paused;
FileDownloadInDatabase to_save;
2022-02-26 19:55:12 +01:00
to_save.download_id = file_info.download_id;
to_save.file_source_id = file_info.file_source_id;
to_save.is_paused = file_info.is_paused;
to_save.priority = file_info.priority;
2022-02-28 16:25:07 +01:00
to_save.created_at = file_info.created_at;
to_save.completed_at = file_info.completed_at;
to_save.file_id = file_info.file_id;
2022-02-26 19:55:12 +01:00
G()->td_db()->get_binlog_pmc()->set(pmc_key(file_info), log_event_store(to_save).as_slice().str());
}
static void remove_from_database(const FileInfo &file_info) {
if (!is_database_enabled()) {
return;
}
2022-02-26 19:55:12 +01:00
G()->td_db()->get_binlog_pmc()->erase(pmc_key(file_info));
}
2022-03-03 15:59:24 +01:00
void init() {
if (is_inited_) {
2022-02-26 19:55:12 +01:00
return;
}
if (is_database_enabled()) {
auto serialized_counter = G()->td_db()->get_binlog_pmc()->get("dlds_counter");
if (!serialized_counter.empty()) {
log_event_parse(sent_counters_, serialized_counter).ensure();
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 {
G()->td_db()->get_binlog_pmc()->erase("dlds_counter");
G()->td_db()->get_binlog_pmc()->erase_by_prefix("dlds#");
2022-02-26 19:55:12 +01:00
}
callback_->update_counters(sent_counters_);
2022-03-03 15:59:24 +01:00
is_inited_ = true;
2022-02-26 19:55:12 +01:00
}
void add_file_from_database(FileDownloadInDatabase in_db) {
if (!in_db.file_id.is_valid() || !in_db.file_source_id.is_valid()) {
LOG(INFO) << "Skip adding file " << in_db.file_id << " from " << in_db.file_source_id;
return;
}
if (by_file_id_.count(in_db.file_id) != 0) {
2022-02-28 16:25:07 +01:00
// file has already been added
return;
2022-02-28 16:25:07 +01:00
}
auto file_info = make_unique<FileInfo>();
2022-02-28 17:09:41 +01:00
file_info->download_id = in_db.download_id;
file_info->file_id = in_db.file_id;
file_info->file_source_id = in_db.file_source_id;
file_info->is_paused = in_db.is_paused;
file_info->priority = narrow_cast<int8>(in_db.priority);
file_info->completed_at = in_db.completed_at;
file_info->created_at = in_db.created_at;
2022-02-28 17:09:41 +01:00
add_file_info(std::move(file_info), "");
}
2022-08-04 17:49:05 +02:00
void load_database_files(const char *source) {
2022-03-03 16:26:25 +01:00
if (is_database_loaded_) {
return;
}
if (!is_database_enabled()) {
is_database_loaded_ = true;
return;
}
CHECK(is_inited_);
2022-08-04 17:49:05 +02:00
LOG_CHECK(database_loading_source_ == nullptr) << database_loading_source_ << ' ' << source;
database_loading_source_ = source;
2022-03-03 16:26:25 +01:00
LOG(INFO) << "Start Download Manager database loading";
auto downloads_in_kv = G()->td_db()->get_binlog_pmc()->prefix_get("dlds#");
for (auto &it : downloads_in_kv) {
Slice key = it.first;
Slice value = it.second;
FileDownloadInDatabase in_db;
2022-03-03 16:26:25 +01:00
log_event_parse(in_db, value).ensure();
CHECK(in_db.download_id == to_integer_safe<int64>(key).ok());
max_download_id_ = max(in_db.download_id, max_download_id_);
add_file_from_database(in_db);
2022-03-03 16:26:25 +01:00
}
is_database_loaded_ = true;
2022-08-04 17:49:05 +02:00
database_loading_source_ = nullptr;
2022-03-03 16:26:25 +01:00
update_counters();
check_completed_downloads_size();
LOG(INFO) << "Finish Download Manager database loading";
}
void prepare_hints() {
for (auto &it : files_) {
const auto &file_info = *it.second;
auto promise =
PromiseCreator::lambda([actor_id = actor_id(this), promise = load_search_text_multipromise_.get_promise(),
download_id = it.first](Result<string> r_search_text) mutable {
send_closure(actor_id, &DownloadManagerImpl::add_download_to_hints, download_id, std::move(r_search_text),
std::move(promise));
});
callback_->get_file_search_text(file_info.file_id, file_info.file_source_id, std::move(promise));
}
}
void add_download_to_hints(int64 download_id, Result<string> r_search_text, Promise<Unit> promise) {
auto it = files_.find(download_id);
if (it == files_.end()) {
return promise.set_value(Unit());
}
if (r_search_text.is_error()) {
if (!G()->close_flag()) {
2022-09-05 16:54:03 +02:00
remove_file_impl(it->second->file_id, {}, false, "add_download_to_hints");
}
} else {
auto search_text = r_search_text.move_as_ok();
2022-03-09 01:27:03 +01:00
// TODO: This is a race. Synchronous call would be better.
hints_.add(download_id, search_text.empty() ? string(" ") : search_text);
}
promise.set_value(Unit());
2022-02-28 17:09:41 +01:00
}
void add_file_info(unique_ptr<FileInfo> &&file_info, const string &search_text) {
CHECK(file_info != nullptr);
auto download_id = file_info->download_id;
file_info->internal_file_id = callback_->dup_file_id(file_info->file_id);
auto file_view = callback_->get_sync_file_view(file_info->file_id);
CHECK(!file_view.empty());
file_info->size = file_view.size();
file_info->expected_size = file_view.expected_size();
file_info->downloaded_size = file_view.local_total_size();
file_info->is_counted = !is_completed(*file_info);
if (file_info->completed_at > 0 && (file_info->size == 0 || file_info->downloaded_size != file_info->size)) {
LOG(INFO) << "Skip adding file " << file_info->file_id << " to recently downloaded files, because local size is "
<< file_info->downloaded_size << " instead of expected " << file_info->size;
remove_from_database(*file_info);
return;
}
by_internal_file_id_[file_info->internal_file_id] = download_id;
by_file_id_[file_info->file_id] = download_id;
2022-02-28 17:09:41 +01:00
hints_.add(download_id, search_text.empty() ? string(" ") : search_text);
file_info->link_token = ++last_link_token_;
2022-02-28 16:25:07 +01:00
2022-03-02 16:29:49 +01:00
LOG(INFO) << "Adding to downloads file " << file_info->file_id << '/' << file_info->internal_file_id << " of size "
<< file_info->size << '/' << file_info->expected_size
<< " with downloaded_size = " << file_info->downloaded_size
<< " and is_paused = " << file_info->is_paused;
auto it = files_.emplace(download_id, std::move(file_info)).first;
2022-03-03 12:30:39 +01:00
bool was_completed = is_completed(*it->second);
register_file_info(*it->second); // must be called before start_file, which can call update_file_download_state
if (is_completed(*it->second)) {
bool is_inserted = completed_download_ids_.insert(it->second->download_id).second;
2022-03-03 12:30:39 +01:00
CHECK(is_inserted == was_completed);
} else {
if (!it->second->is_paused) {
callback_->start_file(it->second->internal_file_id, it->second->priority,
actor_shared(this, it->second->link_token));
}
2022-02-28 16:25:07 +01:00
}
2022-03-05 01:14:31 +01:00
if (is_search_inited_) {
callback_->update_file_added(it->second->file_id, it->second->file_source_id, it->second->created_at,
it->second->completed_at, it->second->is_paused, file_counters_);
2022-03-05 01:14:31 +01:00
}
2022-02-26 19:55:12 +01:00
}
2022-02-23 19:34:09 +01:00
2022-09-05 16:54:03 +02:00
Status remove_file_impl(FileId file_id, FileSourceId file_source_id, bool delete_from_cache, const char *source) {
LOG(INFO) << "Remove from downloads file " << file_id << " from " << file_source_id;
2022-09-05 16:54:03 +02:00
TRY_STATUS(check_is_active(source));
TRY_RESULT(file_info_ptr, get_file_info(file_id, file_source_id));
auto &file_info = *file_info_ptr;
auto download_id = file_info.download_id;
if (!is_completed(file_info) && !file_info.is_paused) {
callback_->pause_file(file_info.internal_file_id);
}
unregister_file_info(file_info);
if (delete_from_cache) {
callback_->delete_file(file_info.internal_file_id);
}
by_internal_file_id_.erase(file_info.internal_file_id);
by_file_id_.erase(file_info.file_id);
hints_.remove(download_id);
completed_download_ids_.erase(download_id);
remove_from_database(file_info);
files_.erase(download_id);
if (is_search_inited_) {
callback_->update_file_removed(file_id, file_counters_);
}
update_counters();
on_file_viewed(download_id);
return Status::OK();
}
Status remove_file_if_finished_impl(FileId file_id) {
2022-08-04 17:49:05 +02:00
TRY_STATUS(check_is_active("remove_file_if_finished_impl"));
TRY_RESULT(file_info_ptr, get_file_info(file_id, {}));
if (!is_completed(*file_info_ptr)) {
return Status::Error("File is active");
}
2022-09-05 16:54:03 +02:00
return remove_file_impl(file_id, {}, false, "remove_file_if_finished_impl");
}
void timeout_expired() final {
clear_counters();
}
void clear_counters() {
2022-03-03 16:26:25 +01:00
if (!is_database_loaded_) {
return;
}
CHECK(counters_ == sent_counters_);
if (counters_.downloaded_size != counters_.total_size || counters_.total_size == 0) {
return;
}
for (auto &it : files_) {
2022-03-03 14:30:16 +01:00
if (is_completed(*it.second) || !it.second->is_paused) {
2022-03-02 16:51:56 +01:00
it.second->is_counted = false;
}
}
counters_ = Counters();
update_counters();
}
2022-02-25 14:18:23 +01:00
void tear_down() final {
2022-02-23 19:34:09 +01:00
callback_.reset();
}
2022-02-26 19:55:12 +01:00
2022-02-28 16:25:07 +01:00
void toggle_is_paused(const FileInfo &file_info, bool is_paused) {
if (is_completed(file_info) || is_paused == file_info.is_paused) {
2022-02-28 16:25:07 +01:00
return;
}
2022-03-01 14:29:08 +01:00
LOG(INFO) << "Change is_paused state of file " << file_info.file_id << " to " << is_paused;
2022-02-28 16:25:07 +01:00
with_file_info(file_info, [&](FileInfo &file_info) {
2022-02-28 16:25:07 +01:00
file_info.is_paused = is_paused;
file_info.need_save_to_database = true;
file_info.link_token = ++last_link_token_;
2022-02-28 16:25:07 +01:00
});
if (is_paused) {
callback_->pause_file(file_info.internal_file_id);
} else {
callback_->start_file(file_info.internal_file_id, file_info.priority, actor_shared(this, file_info.link_token));
2022-02-28 16:25:07 +01:00
}
2022-03-05 01:14:31 +01:00
if (is_search_inited_) {
callback_->update_file_changed(file_info.file_id, file_info.completed_at, file_info.is_paused, file_counters_);
2022-03-05 01:14:31 +01:00
}
2022-02-28 16:25:07 +01:00
}
2022-02-26 19:55:12 +01:00
void update_counters() {
2022-03-03 16:26:25 +01:00
if (!is_database_loaded_) {
2022-02-26 19:55:12 +01:00
return;
}
if (counters_ == sent_counters_) {
return;
}
CHECK(counters_.total_size >= 0);
CHECK(counters_.total_count >= 0);
CHECK(counters_.downloaded_size >= 0);
2022-03-02 16:51:56 +01:00
if ((counters_.downloaded_size == counters_.total_size && counters_.total_size != 0) || counters_ == Counters()) {
if (counters_.total_size != 0) {
constexpr double EMPTY_UPDATE_DELAY = 60.0;
set_timeout_in(EMPTY_UPDATE_DELAY);
} else {
cancel_timeout();
2022-03-02 16:51:56 +01:00
}
G()->td_db()->get_binlog_pmc()->erase("dlds_counter");
} else {
cancel_timeout();
G()->td_db()->get_binlog_pmc()->set("dlds_counter", log_event_store(counters_).as_slice().str());
}
2022-03-02 16:51:56 +01:00
sent_counters_ = counters_;
2022-02-26 19:55:12 +01:00
callback_->update_counters(counters_);
}
2022-02-26 21:54:01 +01:00
Result<const FileInfo *> get_file_info(FileId file_id, FileSourceId file_source_id = {}) {
2022-02-26 21:54:01 +01:00
auto it = by_file_id_.find(file_id);
if (it == by_file_id_.end()) {
return Status::Error(400, "Can't find file");
}
return get_file_info(it->second, file_source_id);
}
Result<const FileInfo *> get_file_info_by_internal(FileId file_id) {
2022-02-26 21:54:01 +01:00
auto it = by_internal_file_id_.find(file_id);
if (it == by_internal_file_id_.end()) {
return Status::Error(400, "Can't find file");
}
return get_file_info(it->second);
}
Result<const FileInfo *> get_file_info(int64 download_id, FileSourceId file_source_id = {}) {
2022-02-26 21:54:01 +01:00
auto it = files_.find(download_id);
if (it == files_.end()) {
return Status::Error(400, "Can't find file");
}
if (file_source_id.is_valid() && file_source_id != it->second->file_source_id) {
2022-02-26 21:54:01 +01:00
return Status::Error(400, "Can't find file with such source");
}
return it->second.get();
2022-02-26 21:54:01 +01:00
}
void unregister_file_info(const FileInfo &file_info) {
CHECK(file_info.is_registered);
file_info.is_registered = false;
2022-03-03 14:30:16 +01:00
if (file_info.is_counted && (is_completed(file_info) || !file_info.is_paused)) {
2022-03-02 16:51:56 +01:00
LOG(INFO) << "Unregister file " << file_info.file_id;
counters_.downloaded_size -= file_info.downloaded_size;
counters_.total_size -= get_file_size(file_info);
counters_.total_count--;
}
if (is_completed(file_info)) {
file_counters_.completed_count--;
CHECK(file_counters_.completed_count >= 0);
} else {
if (file_info.is_paused) {
file_counters_.paused_count--;
CHECK(file_counters_.paused_count >= 0);
}
file_counters_.active_count--;
CHECK(file_counters_.active_count >= file_counters_.paused_count);
}
2022-02-26 21:54:01 +01:00
}
2022-02-26 21:54:01 +01:00
void register_file_info(FileInfo &file_info) {
CHECK(!file_info.is_registered);
file_info.is_registered = true;
bool need_update = false;
if (!is_completed(file_info) && file_info.size != 0 && file_info.downloaded_size == file_info.size) {
2022-03-02 16:51:56 +01:00
LOG(INFO) << "Register file " << file_info.file_id;
file_info.is_paused = false;
2022-02-28 16:25:07 +01:00
file_info.completed_at = G()->unix_time();
file_info.need_save_to_database = true;
bool is_inserted = completed_download_ids_.insert(file_info.download_id).second;
CHECK(is_inserted);
if (file_info.is_counted) {
unviewed_completed_download_ids_.insert(file_info.download_id);
}
2022-03-05 01:14:31 +01:00
need_update = true;
2022-02-26 21:54:01 +01:00
}
2022-03-03 14:30:16 +01:00
if (file_info.is_counted && (is_completed(file_info) || !file_info.is_paused)) {
2022-03-02 16:51:56 +01:00
counters_.downloaded_size += file_info.downloaded_size;
counters_.total_size += get_file_size(file_info);
counters_.total_count++;
}
if (is_completed(file_info)) {
file_counters_.completed_count++;
} else {
if (file_info.is_paused) {
file_counters_.paused_count++;
}
file_counters_.active_count++;
}
if (is_search_inited_ && need_update) {
callback_->update_file_changed(file_info.file_id, file_info.completed_at, file_info.is_paused, file_counters_);
}
sync_with_database(file_info);
2022-02-26 21:54:01 +01:00
update_counters();
CHECK(file_info.is_registered);
check_completed_downloads_size();
}
void check_completed_downloads_size() {
if (!is_database_loaded_) {
return;
}
constexpr size_t MAX_COMPLETED_DOWNLOADS = 200;
while (completed_download_ids_.size() > MAX_COMPLETED_DOWNLOADS) {
auto download_id = *completed_download_ids_.begin();
auto file_info = get_file_info(download_id).move_as_ok();
2022-09-05 16:54:03 +02:00
remove_file_impl(file_info->file_id, FileSourceId(), false, "check_completed_downloads_size");
}
2022-02-26 21:54:01 +01:00
}
void on_file_viewed(int64 download_id) {
if (unviewed_completed_download_ids_.empty()) {
return;
}
2022-04-04 18:08:47 +02:00
LOG(INFO) << "Mark download " << download_id << " as viewed";
unviewed_completed_download_ids_.erase(download_id);
if (unviewed_completed_download_ids_.empty()) {
clear_counters();
}
}
2022-02-26 21:54:01 +01:00
template <class F>
void with_file_info(const FileInfo &const_file_info, F &&f) {
unregister_file_info(const_file_info);
auto &file_info = const_cast<FileInfo &>(const_file_info);
f(file_info);
register_file_info(file_info);
}
2022-08-04 17:49:05 +02:00
Status check_is_active(const char *source) {
2022-02-26 21:54:01 +01:00
if (!callback_) {
2022-08-04 17:49:05 +02:00
LOG(ERROR) << "DownloadManager is closed in " << source;
2022-03-03 15:59:24 +01:00
return Status::Error(500, "DownloadManager is closed");
2022-02-26 21:54:01 +01:00
}
2022-03-03 15:59:24 +01:00
CHECK(is_inited_);
2022-08-04 17:49:05 +02:00
load_database_files(source);
2022-02-26 21:54:01 +01:00
return Status::OK();
}
2022-02-23 19:34:09 +01:00
};
unique_ptr<DownloadManager> DownloadManager::create(unique_ptr<Callback> callback) {
return make_unique<DownloadManagerImpl>(std::move(callback));
2022-02-23 19:34:09 +01:00
}
2022-02-25 14:18:23 +01:00
td_api::object_ptr<td_api::updateFileDownloads> DownloadManager::Counters::get_update_file_downloads_object() const {
return td_api::make_object<td_api::updateFileDownloads>(total_size, total_count, downloaded_size);
2022-02-25 15:57:08 +01:00
}
2022-02-26 21:54:01 +01:00
td_api::object_ptr<td_api::downloadedFileCounts> DownloadManager::FileCounters::get_downloaded_file_counts_object()
const {
return td_api::make_object<td_api::downloadedFileCounts>(active_count, paused_count, completed_count);
}
2022-02-26 19:55:12 +01:00
template <class StorerT>
void DownloadManager::Counters::store(StorerT &storer) const {
BEGIN_STORE_FLAGS();
END_STORE_FLAGS();
td::store(total_size, storer);
td::store(total_count, storer);
td::store(downloaded_size, storer);
}
2022-02-26 21:54:01 +01:00
2022-02-26 19:55:12 +01:00
template <class ParserT>
void DownloadManager::Counters::parse(ParserT &parser) {
BEGIN_PARSE_FLAGS();
END_PARSE_FLAGS();
2022-02-26 21:54:01 +01:00
td::parse(total_size, parser);
2022-02-26 19:55:12 +01:00
td::parse(total_count, parser);
td::parse(downloaded_size, parser);
}
2022-02-23 19:34:09 +01:00
} // namespace td