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

Move DownloadManagerCallback to separate file.

This commit is contained in:
levlam 2022-03-02 14:03:41 +03:00
parent 6da151a402
commit 8ebba6633b
3 changed files with 101 additions and 73 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CMakeLists.txt
View File

@ -507,6 +507,7 @@ set(TDLIB_SOURCE
td/telegram/Document.h
td/telegram/DocumentsManager.h
td/telegram/DownloadManager.h
td/telegram/DownloadManagerCallback.h
td/telegram/DraftMessage.h
td/telegram/EncryptedFile.h
td/telegram/FileReferenceManager.h

99
td/telegram/DownloadManagerCallback.h Normal file
View File

@ -0,0 +1,99 @@
//
// 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)
//
#pragma once
#include "td/telegram/FileReferenceManager.h"
#include "td/telegram/files/FileId.h"
#include "td/telegram/files/FileManager.h"
#include "td/telegram/files/FileSourceId.h"
#include "td/telegram/td_api.h"
#include "td/actor/actor.h"
#include "td/actor/PromiseFuture.h"
#include "td/utils/common.h"
#include "td/utils/Status.h"
namespace td {
class DownloadManagerCallback final : public DownloadManager::Callback {
public:
explicit DownloadManagerCallback(ActorShared<> parent) : parent_(std::move(parent)) {
}
void update_counters(DownloadManager::Counters counters) final {
send_closure(G()->td(), &Td::send_update, counters.get_update_file_downloads_object());
}
void update_file_removed(FileId file_id) final {
send_closure(G()->td(), &Td::send_update,
td_api::make_object<td_api::updateFileRemovedFromDownloads>(file_id.get()));
}
void start_file(FileId file_id, int8 priority, ActorShared<DownloadManager> download_manager) final {
send_closure(G()->file_manager(), &FileManager::download, file_id,
make_download_file_callback(std::move(download_manager)), priority,
FileManager::KEEP_DOWNLOAD_OFFSET, FileManager::IGNORE_DOWNLOAD_LIMIT);
}
void pause_file(FileId file_id) final {
send_closure(G()->file_manager(), &FileManager::download, file_id, nullptr, 0, FileManager::KEEP_DOWNLOAD_OFFSET,
FileManager::KEEP_DOWNLOAD_LIMIT);
}
void delete_file(FileId file_id) final {
send_closure(G()->file_manager(), &FileManager::delete_file, file_id, Promise<Unit>(),
"download manager callback");
}
FileId dup_file_id(FileId file_id) final {
auto td = G()->td().get_actor_unsafe();
return td->file_manager_->dup_file_id(file_id);
}
FileView get_file_view(FileId file_id) final {
auto td = G()->td().get_actor_unsafe();
return td->file_manager_->get_file_view(file_id);
}
td_api::object_ptr<td_api::fileDownload> get_file_download_object(FileId file_id, FileSourceId file_source_id,
int32 add_date, int32 complete_date,
bool is_paused) {
auto td = G()->td().get_actor_unsafe();
return td_api::make_object<td_api::fileDownload>(td->file_manager_->get_file_view(file_id).file_id().get(),
td->file_reference_manager_->get_message_object(file_source_id),
add_date, complete_date, is_paused);
}
private:
ActorShared<> parent_;
static std::shared_ptr<FileManager::DownloadCallback> make_download_file_callback(
ActorShared<DownloadManager> download_manager) {
class Impl final : public FileManager::DownloadCallback {
public:
explicit Impl(ActorShared<DownloadManager> download_manager) : download_manager_(std::move(download_manager)) {
}
void on_progress(FileId file_id) final {
send_update(file_id, false);
}
void on_download_ok(FileId file_id) final {
send_update(file_id, true);
}
void on_download_error(FileId file_id, Status error) final {
send_update(file_id, true);
}
private:
ActorShared<DownloadManager> download_manager_;
void send_update(FileId file_id, bool is_paused) const {
auto td = G()->td().get_actor_unsafe();
auto file_view = td->file_manager_->get_file_view(file_id);
send_closure(download_manager_, &DownloadManager::update_file_download_state, file_id,
file_view.local_total_size(), file_view.size(), is_paused);
// TODO: handle deleted state?
}
};
return std::make_shared<Impl>(std::move(download_manager));
}
};
} // namespace td

74
td/telegram/Td.cpp
View File

@ -36,6 +36,7 @@
#include "td/telegram/DialogSource.h"
#include "td/telegram/DocumentsManager.h"
#include "td/telegram/DownloadManager.h"
#include "td/telegram/DownloadManagerCallback.h"
#include "td/telegram/FileReferenceManager.h"
#include "td/telegram/files/FileGcParameters.h"
#include "td/telegram/files/FileId.h"
@ -3970,79 +3971,6 @@ void Td::init_managers() {
G()->set_contacts_manager(contacts_manager_actor_.get());
country_info_manager_ = make_unique<CountryInfoManager>(this, create_reference());
country_info_manager_actor_ = register_actor("CountryInfoManager", country_info_manager_.get());
class DownloadManagerCallback final : public DownloadManager::Callback {
public:
explicit DownloadManagerCallback(ActorShared<> parent) : parent_(std::move(parent)) {
}
void update_counters(DownloadManager::Counters counters) final {
send_closure(G()->td(), &Td::send_update, counters.get_update_file_downloads_object());
}
void update_file_removed(FileId file_id) final {
send_closure(G()->td(), &Td::send_update,
td_api::make_object<td_api::updateFileRemovedFromDownloads>(file_id.get()));
}
void start_file(FileId file_id, int8 priority, ActorShared<DownloadManager> download_manager) final {
send_closure(G()->file_manager(), &FileManager::download, file_id,
make_download_file_callback(std::move(download_manager)), priority,
FileManager::KEEP_DOWNLOAD_OFFSET, FileManager::IGNORE_DOWNLOAD_LIMIT);
}
void pause_file(FileId file_id) final {
send_closure(G()->file_manager(), &FileManager::download, file_id, nullptr, 0, FileManager::KEEP_DOWNLOAD_OFFSET,
FileManager::KEEP_DOWNLOAD_LIMIT);
}
void delete_file(FileId file_id) final {
send_closure(G()->file_manager(), &FileManager::delete_file, file_id, Promise<Unit>(),
"download manager callback");
}
FileId dup_file_id(FileId file_id) final {
auto td = G()->td().get_actor_unsafe();
return td->file_manager_->dup_file_id(file_id);
}
FileView get_file_view(FileId file_id) final {
auto td = G()->td().get_actor_unsafe();
return td->file_manager_->get_file_view(file_id);
}
td_api::object_ptr<td_api::fileDownload> get_file_download_object(FileId file_id, FileSourceId file_source_id,
int32 add_date, int32 complete_date,
bool is_paused) {
auto td = G()->td().get_actor_unsafe();
return td_api::make_object<td_api::fileDownload>(td->file_manager_->get_file_view(file_id).file_id().get(),
td->file_reference_manager_->get_message_object(file_source_id),
add_date, complete_date, is_paused);
}
private:
ActorShared<> parent_;
static std::shared_ptr<FileManager::DownloadCallback> make_download_file_callback(
ActorShared<DownloadManager> download_manager) {
class Impl final : public FileManager::DownloadCallback {
public:
explicit Impl(ActorShared<DownloadManager> download_manager) : download_manager_(std::move(download_manager)) {
}
void on_progress(FileId file_id) final {
send_update(file_id, false);
}
void on_download_ok(FileId file_id) final {
send_update(file_id, true);
}
void on_download_error(FileId file_id, Status error) final {
send_update(file_id, true);
}
private:
ActorShared<DownloadManager> download_manager_;
void send_update(FileId file_id, bool is_paused) const {
auto td = G()->td().get_actor_unsafe();
auto file_view = td->file_manager_->get_file_view(file_id);
send_closure(download_manager_, &DownloadManager::update_file_download_state, file_id,
file_view.local_total_size(), file_view.size(), is_paused);
// TODO: handle deleted state?
}
};
return std::make_shared<Impl>(std::move(download_manager));
}
};
download_manager_ = DownloadManager::create(td::make_unique<DownloadManagerCallback>(create_reference()));
download_manager_actor_ = register_actor("DownloadManager", download_manager_.get());
game_manager_ = make_unique<GameManager>(this, create_reference());