From fa8feefed70d64271945e9d5fd010b957d93c8cd Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Mon, 3 Jan 2022 08:57:11 +0300
Subject: [PATCH] Reduce total timeout limit to 2 for messages.setTyping.

---
 td/telegram/MessagesManager.cpp       | 7 ++++---
 td/telegram/net/SessionMultiProxy.cpp | 3 +--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/td/telegram/MessagesManager.cpp b/td/telegram/MessagesManager.cpp
index 83ce5d8b1..596d99c73 100644
--- a/td/telegram/MessagesManager.cpp
+++ b/td/telegram/MessagesManager.cpp
@@ -3900,11 +3900,12 @@ class SetTypingQuery final : public Td::ResultHandler {
     if (message_id.is_valid()) {
       flags |= telegram_api::messages_setTyping::TOP_MSG_ID_MASK;
     }
-    auto net_query = G()->net_query_creator().create(telegram_api::messages_setTyping(
+    auto query = G()->net_query_creator().create(telegram_api::messages_setTyping(
         flags, std::move(input_peer), message_id.get_server_message_id().get(), std::move(action)));
-    auto result = net_query.get_weak();
+    query->total_timeout_limit_ = 2;
+    auto result = query.get_weak();
     generation_ = result.generation();
-    send_query(std::move(net_query));
+    send_query(std::move(query));
     return result;
   }
 
diff --git a/td/telegram/net/SessionMultiProxy.cpp b/td/telegram/net/SessionMultiProxy.cpp
index c4d9f3690..462a0765b 100644
--- a/td/telegram/net/SessionMultiProxy.cpp
+++ b/td/telegram/net/SessionMultiProxy.cpp
@@ -39,8 +39,7 @@ SessionMultiProxy::SessionMultiProxy(int32 session_count, std::shared_ptr<AuthDa
 
 void SessionMultiProxy::send(NetQueryPtr query) {
   size_t pos = 0;
-  // TODO temporary hack with total_timeout_limit
-  if (query->auth_flag() == NetQuery::AuthFlag::On && query->total_timeout_limit_ > 7) {
+  if (query->auth_flag() == NetQuery::AuthFlag::On) {
     if (query->session_rand()) {
       pos = query->session_rand() % sessions_.size();
     } else {