Skip to content

Commit 3941670

Browse files
committed
bench/server: Use TCP_NODELAY; raise the number of bytes to recv
1 parent 7763e8e commit 3941670

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

examples/bench/server.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import asyncio
33
import gc
44
import uvloop
5+
import socket as socket_module
56

67
from socket import *
78

@@ -41,14 +42,15 @@ async def echo_client(loop, client):
4142

4243
async def echo_client_streams(reader, writer):
4344
sock = writer.get_extra_info('socket')
45+
if hasattr(socket_module, 'TCP_NODELAY'):
46+
sock.setsockopt(IPPROTO_TCP, TCP_NODELAY, 1)
4447
if PRINT:
4548
print('Connection from', sock.getpeername())
4649
while True:
47-
data = await reader.read(10000)
50+
data = await reader.read(100000)
4851
if not data:
4952
break
5053
writer.write(data)
51-
await writer.drain()
5254
if PRINT:
5355
print('Connection closed')
5456
writer.close()

0 commit comments

Comments
 (0)