LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-24 18:33:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Further improve MessageBox trace logging

Browse Source
This commit is contained in:
Lonami Exo 2023-08-29 13:54:43 +02:00
parent 574e8876ec
commit d40aae75f3
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
telethon/_updates/messagebox.py
View File

@ -495,6 +495,9 @@ class MessageBox:
self.apply_deadlines_reset() self.apply_deadlines_reset()
if self.possible_gaps: if self.possible_gaps:
if __debug__:
self._trace('Trying to re-apply %r possible gaps', len(self.possible_gaps))
# For each update in possible gaps, see if the gap has been resolved already. # For each update in possible gaps, see if the gap has been resolved already.
for key in list(self.possible_gaps.keys()): for key in list(self.possible_gaps.keys()):
self.possible_gaps[key].updates.sort(key=_sort_gaps) self.possible_gaps[key].updates.sort(key=_sort_gaps)
@ -536,6 +539,8 @@ class MessageBox:
pts = PtsInfo.from_update(update) pts = PtsInfo.from_update(update)
if not pts: if not pts:
# No pts means that the update can be applied in any order. # No pts means that the update can be applied in any order.
if __debug__:
self._trace('No pts in update, so it can be applied in any order: %s', update)
return update return update
# As soon as we receive an update of any form related to messages (has `PtsInfo`), # As soon as we receive an update of any form related to messages (has `PtsInfo`),
@ -550,6 +555,8 @@ class MessageBox:
if pts.entry in self.getting_diff_for: if pts.entry in self.getting_diff_for:
# Note: early returning here also prevents gap from being inserted (which they should # Note: early returning here also prevents gap from being inserted (which they should
# not be while getting difference). # not be while getting difference).
if __debug__:
self._trace('Skipping update with %r as its difference is being fetched', pts)
return None return None
if pts.entry in self.map: if pts.entry in self.map:
@ -557,15 +564,13 @@ class MessageBox:
if local_pts + pts.pts_count > pts.pts: if local_pts + pts.pts_count > pts.pts:
# Ignore # Ignore
if __debug__: if __debug__:
self._trace('Skipping update since local pts %r + %r > %r: %s', self._trace('Skipping update since local pts %r > %r: %s', local_pts, pts, update)
local_pts, pts.pts_count, pts.pts, update)
return None return None
elif local_pts + pts.pts_count < pts.pts: elif local_pts + pts.pts_count < pts.pts:
# Possible gap # Possible gap
# TODO store chats too? # TODO store chats too?
if __debug__: if __debug__:
self._trace('Adding update as possible gap since local pts %r + %r < %r: %s', self._trace('Possible gap since local pts %r < %r: %s', local_pts, pts, update)
local_pts, pts.pts_count, pts.pts, update)
if pts.entry not in self.possible_gaps: if pts.entry not in self.possible_gaps:
self.possible_gaps[pts.entry] = PossibleGap( self.possible_gaps[pts.entry] = PossibleGap(
deadline=get_running_loop().time() + POSSIBLE_GAP_TIMEOUT, deadline=get_running_loop().time() + POSSIBLE_GAP_TIMEOUT,
@ -576,7 +581,8 @@ class MessageBox:
return None return None
else: else:
# Apply # Apply
pass if __debug__:
self._trace('Applying update pts since local pts %r = %r: %s', local_pts, pts, update)
# In a channel, we may immediately receive: # In a channel, we may immediately receive:
# * ReadChannelInbox (pts = X, pts_count = 0) # * ReadChannelInbox (pts = X, pts_count = 0)