Prechádzať zdrojové kódy

Try writer.set_write_buffer_limits()

Aleksandr Borzunov 4 rokov pred
rodič
commit
93337ac8a6
1 zmenil súbory, kde vykonal 4 pridanie a 0 odobranie
  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: