1 files changed, 9 insertions, 5 deletions
diff --git a/protocols/Telegram/tdlib/td/td/telegram/files/FileStatsWorker.h b/protocols/Telegram/tdlib/td/td/telegram/files/FileStatsWorker.h
index f70fcac1e0..71c4a7003f 100644
--- a/protocols/Telegram/tdlib/td/td/telegram/files/FileStatsWorker.h
+++ b/protocols/Telegram/tdlib/td/td/telegram/files/FileStatsWorker.h
@@ -1,26 +1,30 @@
 //
-// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
+// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
 //
 // 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)
 //
 #pragma once
 
+#include "td/telegram/files/FileStats.h"
+
 #include "td/actor/actor.h"
-#include "td/actor/PromiseFuture.h"
 
-#include "td/telegram/files/FileStats.h"
+#include "td/utils/CancellationToken.h"
+#include "td/utils/Promise.h"
 
 namespace td {
 
-class FileStatsWorker : public Actor {
+class FileStatsWorker final : public Actor {
  public:
-  explicit FileStatsWorker(ActorShared<> parent) : parent_(std::move(parent)) {
+  FileStatsWorker(ActorShared<> parent, CancellationToken token)
+      : parent_(std::move(parent)), token_(std::move(token)) {
   }
   void get_stats(bool need_all_files, bool split_by_owner_dialog_id, Promise<FileStats> promise);
 
  private:
   ActorShared<> parent_;
+  CancellationToken token_;
 };
 
 }  // namespace td