Browse Source

smaller chunks yet

justheuristic 3 năm trước cách đây
mục cha
commit
7613407b80
1 tập tin đã thay đổi với 5 bổ sung7 xóa
  1. 5 7
      hivemind/p2p/p2p_daemon.py

+ 5 - 7
hivemind/p2p/p2p_daemon.py

@@ -283,16 +283,18 @@ class P2P:
             writer.write(P2P.ERROR_MARKER)
         else:
             writer.write(P2P.MESSAGE_MARKER)
+        t = time.time()
         data = memoryview(protobuf.SerializeToString())
+        print("SER:", time.time() - t)
         writer.write(len(data).to_bytes(P2P.HEADER_LEN, P2P.BYTEORDER))
         for offset in range(0, len(data), chunk_size):
-            await P2P.send_raw_data(data[offset : offset + chunk_size], writer, drain=True)
+            await P2P.send_raw_data(data[offset: offset + chunk_size], writer, drain=True)
+        print("FULL:", time.time() - t)
 
     @staticmethod
     async def receive_protobuf(
         input_protobuf_type: Type[Message], reader: asyncio.StreamReader
     ) -> Tuple[Optional[TInputProtobuf], Optional[RPCError]]:
-        t = time.time()
         msg_type = await reader.readexactly(1)
         if msg_type == P2P.MESSAGE_MARKER:
             protobuf = input_protobuf_type()
@@ -302,13 +304,9 @@ class P2P:
             offset = 0
             while offset < len(data):
                 chunk = await P2P.receive_raw_data(reader)
-                data[offset : offset + len(chunk)] = chunk
+                data[offset: offset + len(chunk)] = chunk
                 offset += len(chunk)
-            print("RECV:", time.time() - t)
-            t = time.time()
             protobuf.ParseFromString(data)
-            print("DES:", time.time() - t)
-
             return protobuf, None
         elif msg_type == P2P.ERROR_MARKER:
             #TODO