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

Add FileReferenceManager::get_current_file_source_id.

Browse Source 
GitOrigin-RevId: 29f3bef8926e205c0c262ffe740c4cbd444e46b7
This commit is contained in:
levlam 2019-01-18 22:10:38 +03:00
parent 1f5a86ea58
commit 94f94750fe
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
td/telegram/FileReferenceManager.cpp
View File

@ -16,6 +16,7 @@
#include "td/actor/MultiPromise.h" #include "td/actor/MultiPromise.h"
#include "td/utils/format.h" #include "td/utils/format.h"
#include "td/utils/misc.h"
#include "td/utils/overloaded.h" #include "td/utils/overloaded.h"
#include "td/utils/Variant.h" #include "td/utils/Variant.h"
@ -33,12 +34,15 @@ fileSourceWallpapers = FileSource; // repair
fileSourceSavedAnimations = FileSource; // repaired with messages.getSavedGifs fileSourceSavedAnimations = FileSource; // repaired with messages.getSavedGifs
*/ */
FileSourceId FileReferenceManager::get_current_file_source_id() const {
return FileSourceId(narrow_cast<int32>(file_sources_.size()));
}
FileSourceId FileReferenceManager::create_message_file_source(FullMessageId full_message_id) { FileSourceId FileReferenceManager::create_message_file_source(FullMessageId full_message_id) {
VLOG(file_references) << "Create file source for " << full_message_id; VLOG(file_references) << "Create file source for " << full_message_id;
auto source_id = FileSourceId{++last_file_source_id_};
FileSourceMessage source{full_message_id}; FileSourceMessage source{full_message_id};
file_sources_.emplace_back(source); file_sources_.emplace_back(source);
return source_id; return get_current_file_source_id();
} }
void FileReferenceManager::add_file_source(NodeId node_id, FileSourceId file_source_id) { void FileReferenceManager::add_file_source(NodeId node_id, FileSourceId file_source_id) {

3
td/telegram/FileReferenceManager.h
View File

@ -83,7 +83,6 @@ class FileReferenceManager : public Actor {
FileSourceWallpapers, FileSourceWebPage, FileSourceSavedAnimations>; FileSourceWallpapers, FileSourceWebPage, FileSourceSavedAnimations>;
vector<FileSource> file_sources_; vector<FileSource> file_sources_;
int32 last_file_source_id_{0};
int64 query_generation{0}; int64 query_generation{0};
std::unordered_map<NodeId, Node, FileIdHash> nodes_; std::unordered_map<NodeId, Node, FileIdHash> nodes_;
@ -91,6 +90,8 @@ class FileReferenceManager : public Actor {
void run_node(NodeId node); void run_node(NodeId node);
void send_query(Destination dest, FileSourceId file_source_id); void send_query(Destination dest, FileSourceId file_source_id);
Destination on_query_result(Destination dest, FileSourceId file_source_id, Status status, int32 sub = 0); Destination on_query_result(Destination dest, FileSourceId file_source_id, Status status, int32 sub = 0);
FileSourceId get_current_file_source_id() const;
}; };
} // namespace td } // namespace td