Add FileReferenceManager::get_current_file_source_id.

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

View File

@ -16,6 +16,7 @@
#include "td/actor/MultiPromise.h"
#include "td/utils/format.h"
#include "td/utils/misc.h"
#include "td/utils/overloaded.h"
#include "td/utils/Variant.h"
@ -33,12 +34,15 @@ fileSourceWallpapers = FileSource; // repair
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) {
VLOG(file_references) << "Create file source for " << full_message_id;
auto source_id = FileSourceId{++last_file_source_id_};
FileSourceMessage source{full_message_id};
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) {

View File

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