From 89d5d680d6da02b12451bab6d3eafe551b26df40 Mon Sep 17 00:00:00 2001
From: Arseny Smirnov <arseny30@gmail.com>
Date: Fri, 28 Dec 2018 09:42:36 +0300
Subject: [PATCH] FileLoader: restart cancelled queries

GitOrigin-RevId: e71c71b25098631ca68de852eece9cb9a37ec775
---
 td/telegram/files/FileLoader.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/td/telegram/files/FileLoader.cpp b/td/telegram/files/FileLoader.cpp
index 137453e96..5a5b384aa 100644
--- a/td/telegram/files/FileLoader.cpp
+++ b/td/telegram/files/FileLoader.cpp
@@ -238,6 +238,9 @@ void FileLoader::on_result(NetQueryPtr query) {
   bool next = false;
   auto status = [&] {
     TRY_RESULT(should_restart, should_restart_part(part, query));
+    if (query->is_error() && query->error().code() == NetQuery::Error::Cancelled) {
+      should_restart = true;
+    }
     if (should_restart) {
       VLOG(files) << "Restart part " << tag("id", part.id) << tag("size", part.size);
       resource_state_.stop_use(static_cast<int64>(part.size));