Ver Fonte

Try writer.set_write_buffer_limits()

Aleksandr Borzunov há 4 anos atrás
pai
commit
93337ac8a6
1 ficheiros alterados com 4 adições e 0 exclusões
  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)
 
             with closing(writer):
+                writer.transport.set_write_buffer_limits(high=2 ** 20)
+
                 processing_task = asyncio.create_task(_process_stream())
                 try:
                     while True:
@@ -370,6 +372,8 @@ class P2P:
             await P2P.send_protobuf(P2P.END_OF_STREAM, writer)
 
         with closing(writer):
+            writer.transport.set_write_buffer_limits(high=2 ** 20)
+
             writing_task = asyncio.create_task(_write_to_stream())
             try:
                 while True: