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

fix index

Browse Source
This commit is contained in:
andrew (from workstation) 2021-01-02 19:15:12 +01:00
parent eb41d4bb20
commit 0eedaeea3a
2 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
td/telegram/FileReferenceManager.cpp
View File

@ -64,9 +64,10 @@ FileSourceId FileReferenceManager::get_current_file_source_id() const {
template <class T> template <class T>
FileSourceId FileReferenceManager::add_file_source_id(T source, Slice source_str) { FileSourceId FileReferenceManager::add_file_source_id(T source, Slice source_str) {
file_sources_[++unique_file_source_id] = (std::move(source)); auto source_id = ++unique_file_source_id;
VLOG(file_references) << "Create file source " << file_sources_.size() << " for " << source_str; file_sources_[source_id] = (std::move(source));
return get_current_file_source_id(); VLOG(file_references) << "Create file source " << source_id << " for " << source_str;
return FileSourceId(narrow_cast<int32>((int32) source_id));
} }
FileSourceId FileReferenceManager::create_message_file_source(FullMessageId full_message_id) { FileSourceId FileReferenceManager::create_message_file_source(FullMessageId full_message_id) {
@ -144,8 +145,7 @@ vector<FullMessageId> FileReferenceManager::get_some_message_file_sources(NodeId
vector<FullMessageId> result; vector<FullMessageId> result;
for (auto file_source_id : file_source_ids) { for (auto file_source_id : file_source_ids) {
auto index = static_cast<size_t>(file_source_id.get()) - 1; const auto &file_source = file_sources_[file_source_id.get()];
const auto &file_source = file_sources_[index];
if (file_source.get_offset() == 0) { if (file_source.get_offset() == 0) {
result.push_back(file_source.get<FileSourceMessage>().full_message_id); result.push_back(file_source.get<FileSourceMessage>().full_message_id);
} }
@ -248,8 +248,7 @@ void FileReferenceManager::send_query(Destination dest, FileSourceId file_source
send_closure(file_manager, &FileManager::on_file_reference_repaired, dest.node_id, file_source_id, send_closure(file_manager, &FileManager::on_file_reference_repaired, dest.node_id, file_source_id,
std::move(result), std::move(new_promise)); std::move(result), std::move(new_promise));
}); });
auto index = static_cast<size_t>(file_source_id.get()) - 1; file_sources_[file_source_id.get()].visit(overloaded(
file_sources_[index].visit(overloaded(
[&](const FileSourceMessage &source) { [&](const FileSourceMessage &source) {
send_closure_later(G()->messages_manager(), &MessagesManager::get_message_from_server, source.full_message_id, send_closure_later(G()->messages_manager(), &MessagesManager::get_message_from_server, source.full_message_id,
std::move(promise), nullptr); std::move(promise), nullptr);

3
td/telegram/FileReferenceManager.hpp
View File

@ -28,8 +28,7 @@ namespace td {
template <class StorerT> template <class StorerT>
void FileReferenceManager::store_file_source(FileSourceId file_source_id, StorerT &storer) const { void FileReferenceManager::store_file_source(FileSourceId file_source_id, StorerT &storer) const {
auto index = static_cast<size_t>(file_source_id.get()) - 1; auto source_tuple = file_sources_.find(file_source_id.get());
auto source_tuple = file_sources_.find(index);
auto source = source_tuple->second; auto source = source_tuple->second;
td::store(source.get_offset(), storer); td::store(source.get_offset(), storer);