andreacavalli/tdlib-fork
Archived
1
0
Fork 0
Code Pull Requests Releases Activity

Add basic protectiong from cyclic successful file reference repair.

Browse Source 
GitOrigin-RevId: f0d3ef24dc95ed1f5ed06801abc435ee40c80f01
This commit is contained in:
levlam 2019-10-18 04:30:49 +03:00
parent 9fa88eb0ec
commit 6b001fedd9
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
td/telegram/FileReferenceManager.cpp
View File

@ -201,6 +201,14 @@ void FileReferenceManager::run_node(NodeId node_id) {
node.query = {}; node.query = {};
return; return;
} }
if (node.last_successful_repair_time >= Time::now() - 60) {
VLOG(file_references) << "Recently repaired file reference for file " << node_id << ", do not try again";
for (auto &p : node.query->promises) {
p.set_error(Status::Error(429, "Too Many Requests: retry after 60"));
}
node.query = {};
return;
}
auto file_source_id = node.file_source_ids.next(); auto file_source_id = node.file_source_ids.next();
send_query({node_id, node.query->generation}, file_source_id); send_query({node_id, node.query->generation}, file_source_id);
} }
@ -313,6 +321,7 @@ FileReferenceManager::Destination FileReferenceManager::on_query_result(Destinat
} }
if (status.is_ok()) { if (status.is_ok()) {
node.last_successful_repair_time = Time::now();
for (auto &p : query->promises) { for (auto &p : query->promises) {
p.set_value(Unit()); p.set_value(Unit());
} }

1
td/telegram/FileReferenceManager.h
View File

@ -87,6 +87,7 @@ class FileReferenceManager : public Actor {
struct Node { struct Node {
SetWithPosition<FileSourceId> file_source_ids; SetWithPosition<FileSourceId> file_source_ids;
unique_ptr<Query> query; unique_ptr<Query> query;
double last_successful_repair_time = -1e10;
}; };
struct FileSourceMessage { struct FileSourceMessage {