Skip to content

Commit 2af8299

Browse files
committed
Fix request_id resetting
1 parent 9fd9176 commit 2af8299

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

pymongo/network_layer.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -598,6 +598,7 @@ def buffer_updated(self, nbytes: int) -> None:
598598
if self._expecting_header:
599599
try:
600600
self._body_size, self._op_code = self.process_header()
601+
self._request_id = None
601602
except ProtocolError as exc:
602603
self.connection_lost(exc)
603604
return
@@ -626,8 +627,12 @@ def buffer_updated(self, nbytes: int) -> None:
626627
self._overflow_index,
627628
)
628629
)
630+
# If the current message has an overflow buffer, then the entire default buffer is full
631+
if self._overflow:
632+
self._start_index = self._end_index
629633
# Update the buffer's first written offset to reflect this message's size
630-
self._start_index += self._body_size
634+
else:
635+
self._start_index += self._body_size
631636
self._done_messages.append(result)
632637
# If at least one header's worth of data remains after the current message, reprocess all leftover data
633638
if self._end_index - self._start_index >= 16:

0 commit comments

Comments
 (0)