73 lines
2.7 KiB
C++
73 lines
2.7 KiB
C++
|
//
|
||
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2019
|
||
|
|
//
|
||
|
|
// 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)
|
||
|
|
//
|
||
|
|
#include "td/telegram/FileReferenceManager.h"
|
||
|
|
|
||
|
|
#include "td/telegram/Global.h"
|
||
|
|
#include "td/telegram/MessagesManager.h"
|
||
|
|
|
||
|
|
#include "td/actor/MultiPromise.h"
|
||
|
|
|
||
|
|
#include "td/utils/format.h"
|
||
|
|
|
||
|
|
namespace td {
|
||
|
|
|
||
|
|
FileSourceId FileReferenceManager::create_file_source(FullMessageId full_message_id) {
|
||
|
|
auto it = from_full_message_id_.find(full_message_id);
|
||
|
|
if (it != from_full_message_id_.end()) {
|
||
|
|
return it->second;
|
||
|
|
}
|
||
|
|
++last_file_source_id_;
|
||
|
|
to_full_message_id_[last_file_source_id_] = full_message_id;
|
||
|
|
from_full_message_id_[full_message_id] = FileSourceId{last_file_source_id_};
|
||
|
|
return FileSourceId{last_file_source_id_};
|
||
|
|
}
|
||
|
|
|
||
|
|
void FileReferenceManager::update_file_reference(FileId file_id, std::vector<FileSourceId> sources, Promise<> promise) {
|
||
|
|
LOG(ERROR) << "update file reference: " << file_id << " " << format::as_array(sources);
|
||
|
|
//if (td::Random::fast(0, 3) == 0) {
|
||
|
|
//promise.set_error(td::Status::Error("Error"));
|
||
|
|
//return;
|
||
|
|
//}
|
||
|
|
|
||
|
|
//if (td::Random::fast(0, 3) == 0) {
|
||
|
|
//promise.set_value(Unit());
|
||
|
|
//return;
|
||
|
|
//}
|
||
|
|
|
||
|
|
MultiPromiseActorSafe mpas{"UpdateFileReferenceMultiPromiseActor"};
|
||
|
|
mpas.add_promise(std::move(promise));
|
||
|
|
auto lock = mpas.get_promise();
|
||
|
|
SCOPE_EXIT {
|
||
|
|
lock.set_value(Unit());
|
||
|
|
};
|
||
|
|
for (auto &source_id : sources) {
|
||
|
|
auto it = to_full_message_id_.find(source_id);
|
||
|
|
if (it != to_full_message_id_.end()) {
|
||
|
|
std::vector<FullMessageId> message_ids = {it->second};
|
||
|
|
auto new_promise = PromiseCreator::lambda([promise = mpas.get_promise(), file_id, source_id,
|
||
|
|
file_manager = G()->file_manager()](Result<Unit> res) mutable {
|
||
|
|
if (res.is_error()) {
|
||
|
|
send_closure(file_manager, &FileManager::remove_file_source, file_id, source_id);
|
||
|
|
send_lambda(file_manager, [promise = std::move(promise)]() mutable { promise.set_value({}); });
|
||
|
|
} else {
|
||
|
|
promise.set_value(Unit());
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
LOG(ERROR) << "Ask for " << it->second;
|
||
|
|
send_closure_later(G()->messages_manager(), &MessagesManager::get_messages_from_server, std::move(message_ids),
|
||
|
|
std::move(new_promise), nullptr);
|
||
|
|
} else {
|
||
|
|
LOG(ERROR) << "Invalid source id " << source_id << " " << file_id;
|
||
|
|
send_closure(G()->file_manager(), &FileManager::remove_file_source, file_id, source_id);
|
||
|
|
send_lambda(G()->file_manager(), [promise = mpas.get_promise()]() mutable { promise.set_value(Unit()); });
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace td
|