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

FileNode * -> FileNodePtr

Browse Source 
GitOrigin-RevId: 1fa58d17dc71c74c78097105fc4c2acbde3b48a7
This commit is contained in:
Arseny Smirnov 2018-01-30 19:54:17 +03:00 committed by levlam
parent 74bdbedb5a
commit d3e4ad0060
2 changed files with 48 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
td/telegram/files/FileManager.cpp
View File

@ -36,7 +36,7 @@ namespace td {
static int VERBOSITY_NAME(update_file) = VERBOSITY_NAME(INFO);
/*** FileNode ***/
/*** FileNodePtr **/
void FileNode::set_local_location(const LocalFileLocation &local, int64 ready_size) {
if (local_ready_size_ != ready_size) {
local_ready_size_ = ready_size;
@ -326,7 +326,7 @@ bool FileView::get_by_hash() const {
return node_->get_by_hash_;
}
FileView::FileView(const FileNode *node) : node_(node) {
FileView::FileView(ConstFileNodePtr node) : node_(node) {
}
bool FileView::empty() const {
@ -512,7 +512,7 @@ static Status check_partial_local_location(const PartialLocalFileLocation &locat
return Status::OK();
}
Status FileManager::check_local_location(FileNode *node) {
Status FileManager::check_local_location(FileNodePtr node) {
Status status;
if (node->local_.type() == LocalFileLocation::Type::Full) {
status = check_local_location(node->local_.full(), node->size_);
@ -543,7 +543,7 @@ FileId FileManager::dup_file_id(FileId file_id) {
return result;
}
FileId FileManager::create_file_id(int32 file_node_id, FileNode *file_node) {
FileId FileManager::create_file_id(int32 file_node_id, FileNodePtr file_node) {
auto file_id = next_file_id();
get_file_id_info(file_id)->node_id_ = file_node_id;
file_node->file_ids_.push_back(file_id);
@ -803,7 +803,7 @@ static int merge_choose_encryption_key(const FileEncryptionKey &a, const FileEnc
return 2;
}
void FileManager::cancel_download(FileNode *node) {
void FileManager::cancel_download(FileNodePtr node) {
if (node->download_id_ == 0) {
return;
}
@ -813,7 +813,7 @@ void FileManager::cancel_download(FileNode *node) {
node->set_download_priority(0);
}
void FileManager::cancel_upload(FileNode *node) {
void FileManager::cancel_upload(FileNodePtr node) {
if (node->upload_id_ == 0) {
return;
}
@ -822,7 +822,7 @@ void FileManager::cancel_upload(FileNode *node) {
node->set_upload_priority(0);
}
void FileManager::cancel_generate(FileNode *node) {
void FileManager::cancel_generate(FileNodePtr node) {
if (node->generate_id_ == 0) {
return;
}
@ -839,7 +839,7 @@ Result<FileId> FileManager::merge(FileId x_file_id, FileId y_file_id, bool no_sy
return Status::Error("First file_id is invalid");
}
FileNodeId x_node_id;
FileNode *x_node = no_sync ? get_file_node(x_file_id, &x_node_id) : get_sync_file_node(x_file_id, &x_node_id);
FileNodePtr x_node = no_sync ? get_file_node(x_file_id, &x_node_id) : get_sync_file_node(x_file_id, &x_node_id);
if (!x_node) {
return Status::Error(PSLICE() << "Can't merge files. First id is invalid: " << x_file_id << " and " << y_file_id);
}
@ -848,7 +848,7 @@ Result<FileId> FileManager::merge(FileId x_file_id, FileId y_file_id, bool no_sy
return x_node->main_file_id_;
}
FileNodeId y_node_id;
FileNode *y_node = no_sync ? get_file_node(y_file_id, &y_node_id) : get_file_node(y_file_id, &y_node_id);
FileNodePtr y_node = no_sync ? get_file_node(y_file_id, &y_node_id) : get_file_node(y_file_id, &y_node_id);
if (!y_node) {
return Status::Error(PSLICE() << "Can't merge files. Second id is invalid: " << x_file_id << " and " << y_file_id);
}
@ -870,7 +870,7 @@ Result<FileId> FileManager::merge(FileId x_file_id, FileId y_file_id, bool no_sy
<< x_node->remote_.full();
}
FileNode *nodes[] = {x_node, y_node, x_node};
FileNodePtr nodes[] = {x_node, y_node, x_node};
FileNodeId node_ids[] = {x_node_id, y_node_id};
int local_i = merge_choose(x_node->local_, y_node->local_);
@ -905,8 +905,8 @@ Result<FileId> FileManager::merge(FileId x_file_id, FileId y_file_id, bool no_sy
std::make_tuple(x_node->pmc_id_, x_node->file_ids_.size(), main_file_id_i == 0);
auto other_node_i = 1 - node_i;
FileNode *node = nodes[node_i];
FileNode *other_node = nodes[other_node_i];
FileNodePtr node = nodes[node_i];
FileNodePtr other_node = nodes[other_node_i];
auto file_view = FileView(node);
LOG(INFO) << "x_node->pmc_id_ = " << x_node->pmc_id_ << ", y_node->pmc_id_ = " << y_node->pmc_id_
@ -1047,7 +1047,7 @@ Result<FileId> FileManager::merge(FileId x_file_id, FileId y_file_id, bool no_sy
return node->main_file_id_;
}
void FileManager::try_flush_node(FileNode *node, bool new_remote, bool new_local, bool new_generate,
void FileManager::try_flush_node(FileNodePtr node, bool new_remote, bool new_local, bool new_generate,
FileDbId other_pmc_id) {
if (node->need_pmc_flush()) {
if (file_db_) {
@ -1063,7 +1063,7 @@ void FileManager::try_flush_node(FileNode *node, bool new_remote, bool new_local
try_flush_node_info(node);
}
void FileManager::try_flush_node_info(FileNode *node) {
void FileManager::try_flush_node_info(FileNodePtr node) {
if (node->need_info_flush()) {
for (auto file_id : node->file_ids_) {
auto *info = get_file_id_info(file_id);
@ -1076,7 +1076,7 @@ void FileManager::try_flush_node_info(FileNode *node) {
}
}
void FileManager::clear_from_pmc(FileNode *node) {
void FileManager::clear_from_pmc(FileNodePtr node) {
if (!file_db_) {
return;
}
@ -1099,7 +1099,7 @@ void FileManager::clear_from_pmc(FileNode *node) {
file_db_->clear_file_data(node->pmc_id_, data);
}
void FileManager::flush_to_pmc(FileNode *node, bool new_remote, bool new_local, bool new_generate) {
void FileManager::flush_to_pmc(FileNodePtr node, bool new_remote, bool new_local, bool new_generate) {
if (!file_db_) {
return;
}
@ -1141,7 +1141,7 @@ void FileManager::flush_to_pmc(FileNode *node, bool new_remote, bool new_local,
(create_flag || new_generate));
}
const FileNode *FileManager::get_file_node(FileId file_id, FileNodeId *file_node_id) const {
ConstFileNodePtr FileManager::get_file_node(FileId file_id, FileNodeId *file_node_id) const {
if (file_id.get() <= 0 || file_id.get() >= static_cast<int32>(file_id_info_.size())) {
return nullptr;
}
@ -1155,10 +1155,10 @@ const FileNode *FileManager::get_file_node(FileId file_id, FileNodeId *file_node
return file_nodes_[node_id].get();
}
FileNode *FileManager::get_file_node(FileId file_id, FileNodeId *file_node_id) {
return const_cast<FileNode *>(static_cast<const FileManager *>(this)->get_file_node(file_id, file_node_id));
FileNodePtr FileManager::get_file_node(FileId file_id, FileNodeId *file_node_id) {
return const_cast<FileNodePtr>(static_cast<const FileManager *>(this)->get_file_node(file_id, file_node_id));
}
FileNode *FileManager::get_sync_file_node(FileId file_id, FileNodeId *file_node_id) {
FileNodePtr FileManager::get_sync_file_node(FileId file_id, FileNodeId *file_node_id) {
auto file_node = get_file_node(file_id, file_node_id);
if (!file_node) {
return nullptr;
@ -1166,7 +1166,7 @@ FileNode *FileManager::get_sync_file_node(FileId file_id, FileNodeId *file_node_
return load_from_pmc(file_node, true, true, true);
}
FileNode *FileManager::load_from_pmc(FileNode *node, bool new_remote, bool new_local, bool new_generate) {
FileNodePtr FileManager::load_from_pmc(FileNodePtr node, bool new_remote, bool new_local, bool new_generate) {
if (!node->need_load_from_pmc_) {
return node;
}
@ -1364,7 +1364,7 @@ void FileManager::download(FileId file_id, std::shared_ptr<DownloadCallback> cal
try_flush_node(node);
}
void FileManager::run_download(FileNode *node) {
void FileManager::run_download(FileNodePtr node) {
if (node->need_load_from_pmc_) {
return;
}
@ -1502,7 +1502,7 @@ void FileManager::external_file_generate_finish(int64 id, Status status, Promise
std::move(promise));
}
void FileManager::run_generate(FileNode *node) {
void FileManager::run_generate(FileNodePtr node) {
if (node->need_load_from_pmc_) {
return;
}
@ -1574,7 +1574,7 @@ void FileManager::run_generate(FileNode *node) {
LOG(INFO) << "File " << file_id << " generate request has sent to FileGenerateManager";
}
void FileManager::run_upload(FileNode *node, std::vector<int> bad_parts) {
void FileManager::run_upload(FileNodePtr node, std::vector<int> bad_parts) {
if (node->need_load_from_pmc_) {
return;
}
@ -2172,7 +2172,7 @@ void FileManager::on_error(QueryId query_id, Status status) {
on_error_impl(node, query.type_, was_active, std::move(status));
}
void FileManager::on_error_impl(FileNode *node, FileManager::Query::Type type, bool was_active, Status status) {
void FileManager::on_error_impl(FileNodePtr node, FileManager::Query::Type type, bool was_active, Status status) {
SCOPE_EXIT {
try_flush_node(node);
};

42
td/telegram/files/FileManager.h
View File

@ -122,10 +122,14 @@ class FileNode {
bool info_changed_flag_{false};
};
class FileManager;
using FileNodePtr = FileNode *;
using ConstFileNodePtr = const FileNode *;
class FileView {
public:
FileView() = default;
explicit FileView(const FileNode *node);
explicit FileView(ConstFileNodePtr node);
bool empty() const;
@ -182,7 +186,7 @@ class FileView {
}
private:
const FileNode *node_ = nullptr;
ConstFileNodePtr node_{};
};
class Td;
@ -344,7 +348,7 @@ class FileManager : public FileLoadManager::Callback {
FileId next_file_id();
FileNodeId next_file_node_id();
int32 next_pmc_file_id();
FileId create_file_id(int32 file_node_id, FileNode *file_node);
FileId create_file_id(int32 file_node_id, FileNodePtr file_node);
void try_forget_file_id(FileId file_id);
void load_from_pmc(FileId file_id, FullLocalFileLocation full_local);
@ -355,31 +359,31 @@ class FileManager : public FileLoadManager::Callback {
void load_from_pmc_result(FileId file_id, Result<FileData> &&result);
FileId register_pmc_file_data(FileData &&data);
Status check_local_location(FileNode *node);
Status check_local_location(FileNodePtr node);
Status check_local_location(FullLocalFileLocation &location, int64 &size);
void try_flush_node(FileNode *node, bool new_remote = false, bool new_local = false, bool new_generate = false,
void try_flush_node(FileNodePtr node, bool new_remote = false, bool new_local = false, bool new_generate = false,
FileDbId other_pmc_id = Auto());
void try_flush_node_info(FileNode *node);
void clear_from_pmc(FileNode *node);
void flush_to_pmc(FileNode *node, bool new_remote, bool new_local, bool new_generate);
FileNode *load_from_pmc(FileNode *node, bool new_remote, bool new_local, bool new_generate) TD_WARN_UNUSED_RESULT;
void try_flush_node_info(FileNodePtr node);
void clear_from_pmc(FileNodePtr node);
void flush_to_pmc(FileNodePtr node, bool new_remote, bool new_local, bool new_generate);
FileNodePtr load_from_pmc(FileNodePtr node, bool new_remote, bool new_local, bool new_generate) TD_WARN_UNUSED_RESULT;
string get_persistent_id(const FullRemoteFileLocation &location);
string fix_file_extension(Slice file_name, Slice file_type, Slice file_extension);
string get_file_name(FileType file_type, Slice path);
const FileNode *get_file_node(FileId file_id, FileNodeId *file_node_id = nullptr) const;
FileNode *get_file_node(FileId file_id, FileNodeId *file_node_id = nullptr);
FileNode *get_sync_file_node(FileId file_id, FileNodeId *file_node_id = nullptr);
ConstFileNodePtr get_file_node(FileId file_id, FileNodeId *file_node_id = nullptr) const;
FileNodePtr get_file_node(FileId file_id, FileNodeId *file_node_id = nullptr);
FileNodePtr get_sync_file_node(FileId file_id, FileNodeId *file_node_id = nullptr);
// void release_file_node(FileNodeId id);
void cancel_download(FileNode *node);
void cancel_upload(FileNode *node);
void cancel_generate(FileNode *node);
void run_upload(FileNode *node, std::vector<int> bad_parts);
void run_download(FileNode *node);
void run_generate(FileNode *node);
void cancel_download(FileNodePtr node);
void cancel_upload(FileNodePtr node);
void cancel_generate(FileNodePtr node);
void run_upload(FileNodePtr node, std::vector<int> bad_parts);
void run_download(FileNodePtr node);
void run_generate(FileNodePtr node);
void on_start_download(QueryId query_id) override;
void on_partial_download(QueryId query_id, const PartialLocalFileLocation &partial_local, int64 ready_size) override;
@ -390,7 +394,7 @@ class FileManager : public FileLoadManager::Callback {
void on_upload_full_ok(QueryId query_id, const FullRemoteFileLocation &remote) override;
void on_error(QueryId query_id, Status status) override;
void on_error_impl(FileNode *node, Query::Type type, bool was_active, Status status);
void on_error_impl(FileNodePtr node, Query::Type type, bool was_active, Status status);
void on_partial_generate(QueryId, const PartialLocalFileLocation &partial_local, int32 expected_size);
void on_generate_ok(QueryId, const FullLocalFileLocation &local);