Reuse protocol buffer

Author: NoahStapp

pymongo/asynchronous/network.py

@@ -330,8 +330,8 @@ async def receive_message_stream(
     #         deadline = None
     deadline = None
     # Ignore the response's request id.
-    mv = memoryview(bytearray(max_message_size))
-    conn.conn[1].reset(mv)
+    # data = bytearray(max_message_size)
+    conn.conn[1].reset()
     length, op_code = await asyncio.wait_for(conn.conn[1].read(), timeout=None)
 
     # length, _, response_to, op_code = _UNPACK_HEADER(await async_receive_data_stream(conn, 16, deadline))
@@ -362,5 +362,5 @@ async def receive_message_stream(
         raise ProtocolError(
             f"Got opcode {op_code!r} but expected {_UNPACK_REPLY.keys()!r}"
         ) from None
-    return unpack_reply(mv[16:length])
+    return unpack_reply(conn.conn[1].data()[16:length])
 

pymongo/network_layer.py

@@ -77,7 +77,7 @@
 class PyMongoProtocol(asyncio.BufferedProtocol):
     def __init__(self):
         self.transport = None
-        self._buffer = memoryview(bytearray(65536))  # 64KB default buffer for SSL handshakes
+        self._buffer = memoryview(bytearray(MAX_MESSAGE_SIZE))
         self.expected_length = 0
         self.expecting_header = False
         self.bytes_read = 0
@@ -146,8 +146,8 @@ async def _drain_helper(self):
         self._drain_waiter = self._loop.create_future()
         await self._drain_waiter
 
-    def reset(self, buffer: memoryview):
-        self._buffer = buffer
+    def reset(self):
+        # self._buffer = buffer
         self.bytes_read = 0
         self.expecting_header = True
         self.op_code = None