瀏覽代碼

Try writer.set_write_buffer_limits()

Aleksandr Borzunov 4 年之前
父節點
當前提交
93337ac8a6
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      hivemind/p2p/p2p_daemon.py

+ 4 - 0
hivemind/p2p/p2p_daemon.py

@@ -336,6 +336,8 @@ class P2P:
                         await P2P.send_protobuf(RPCError(message=str(e)), writer)
                         await P2P.send_protobuf(RPCError(message=str(e)), writer)
 
 
             with closing(writer):
             with closing(writer):
+                writer.transport.set_write_buffer_limits(high=2 ** 20)
+
                 processing_task = asyncio.create_task(_process_stream())
                 processing_task = asyncio.create_task(_process_stream())
                 try:
                 try:
                     while True:
                     while True:
@@ -370,6 +372,8 @@ class P2P:
             await P2P.send_protobuf(P2P.END_OF_STREAM, writer)
             await P2P.send_protobuf(P2P.END_OF_STREAM, writer)
 
 
         with closing(writer):
         with closing(writer):
+            writer.transport.set_write_buffer_limits(high=2 ** 20)
+
             writing_task = asyncio.create_task(_write_to_stream())
             writing_task = asyncio.create_task(_write_to_stream())
             try:
             try:
                 while True:
                 while True: