From 56adf14116ed1a8bceda33cd6a1c9a2fcfbce1ee Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Wed, 26 Jul 2023 22:04:57 +0300
Subject: [PATCH] Improve PTS gap repair.

---
 td/telegram/UpdatesManager.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/td/telegram/UpdatesManager.cpp b/td/telegram/UpdatesManager.cpp
index ee2a198d8..3f4c69420 100644
--- a/td/telegram/UpdatesManager.cpp
+++ b/td/telegram/UpdatesManager.cpp
@@ -321,7 +321,11 @@ void UpdatesManager::repair_pts_gap() {
     return;
   }
   auto pts = get_pts() + 1;
-  if (pending_pts_updates_.empty() || pending_pts_updates_.begin()->first != pts + 1) {
+  if (pending_pts_updates_.empty()) {
+    return;
+  }
+  auto it = pending_pts_updates_.begin();
+  if (it->second.pts != pts + it->second.pts_count) {
     return;
   }
   VLOG(get_difference) << "Fetch update with PTS = " << pts;