Restore TOTALS

Author: NoahStapp

pymongo/asynchronous/network.py

@@ -63,10 +63,10 @@
 
 _IS_SYNC = False
 
-TOTAL = []
-TOTAL_WRITE = []
-TOTAL_READ = []
-print(f"TOTALS: {TOTAL, TOTAL_WRITE, TOTAL_READ}")
+# TOTAL = []
+# TOTAL_WRITE = []
+# TOTAL_READ = []
+# print(f"TOTALS: {TOTAL, TOTAL_WRITE, TOTAL_READ}")
 
 
 async def command_stream(
@@ -211,13 +211,13 @@ async def command_stream(
             read_start = time.monotonic()
             reply = await receive_message_stream(conn, request_id)
             read_elapsed = time.monotonic() - read_start
-            if name == "insert":
-                TOTAL.append(write_elapsed + read_elapsed)
-                TOTAL_READ.append(read_elapsed)
-                TOTAL_WRITE.append(write_elapsed)
-            if name == "endSessions":
-                print(
-                    f"AVERAGE READ: {statistics.mean(TOTAL_READ)}, AVERAGE WRITE: {statistics.mean(TOTAL_WRITE)}, AVERAGE ELAPSED: {statistics.mean(TOTAL)}")
+            # if name == "insert":
+            #     TOTAL.append(write_elapsed + read_elapsed)
+            #     TOTAL_READ.append(read_elapsed)
+            #     TOTAL_WRITE.append(write_elapsed)
+            # if name == "endSessions":
+            #     print(
+            #         f"AVERAGE READ: {statistics.mean(TOTAL_READ)}, AVERAGE WRITE: {statistics.mean(TOTAL_WRITE)}, AVERAGE ELAPSED: {statistics.mean(TOTAL)}")
             conn.more_to_come = reply.more_to_come
             unpacked_docs = reply.unpack_response(
                 codec_options=codec_options, user_fields=user_fields

pymongo/network_layer.py

@@ -34,6 +34,7 @@
 from pymongo._asyncio_task import create_task
 from pymongo.common import MAX_MESSAGE_SIZE
 from pymongo.errors import _OperationCancelled
+from pymongo.message import _OpReply, _UNPACK_REPLY
 from pymongo.socket_checker import _errno_from_exception
 
 try:
@@ -75,7 +76,7 @@
 class PyMongoProtocol(asyncio.BufferedProtocol):
     def __init__(self):
         self.transport = None
-        self._buffer = memoryview(bytearray(MAX_MESSAGE_SIZE))
+        self._buffer = memoryview(bytearray(65536))  # 64KB default buffer for SSL handshakes
         self.expected_length = 0
         self.expecting_header = False
         self.bytes_read = 0