Set TCP_NODELAY in all echo server benchmarks (go sets it by default)

Author: 1st1

servers/asyncioecho.py

@@ -30,6 +30,11 @@ async def echo_server(loop, address, unix):
 
 
 async def echo_client(loop, client):
+    try:
+        sock.setsockopt(IPPROTO_TCP, TCP_NODELAY, 1)
+    except (OSError, NameError):
+        pass
+
     with client:
         while True:
             data = await loop.sock_recv(client, 102400)
@@ -41,8 +46,12 @@ async def echo_client(loop, client):
 
 
 async def echo_client_streams(reader, writer):
+    sock = writer.get_extra_info('socket')
+    try:
+        sock.setsockopt(IPPROTO_TCP, TCP_NODELAY, 1)
+    except (OSError, NameError):
+        pass
     if PRINT:
-        sock = writer.get_extra_info('socket')
         print('Connection from', sock.getpeername())
     while True:
          data = await reader.read(102400)
@@ -57,6 +66,11 @@ async def echo_client_streams(reader, writer):
 class EchoProtocol(asyncio.Protocol):
     def connection_made(self, transport):
         self.transport = transport
+        sock = transport.get_extra_info('socket')
+        try:
+            sock.setsockopt(IPPROTO_TCP, TCP_NODELAY, 1)
+        except (OSError, NameError):
+            pass
 
     def connection_lost(self, exc):
         self.transport = None

servers/curioecho.py

@@ -3,10 +3,16 @@
 # A simple echo server
 
 from curio import Kernel, new_task, run_server
+from socket import *
 
 
 async def echo_handler(client, addr):
     print('Connection from', addr)
+    try:
+        client.setsockopt(IPPROTO_TCP, TCP_NODELAY, 1)
+    except (OSError, NameError):
+        pass
+
     while True:
         data = await client.recv(102400)
         if not data:

servers/curioecho_streams.py

@@ -1,8 +1,12 @@
 from curio import Kernel, new_task, run_server
-
+from socket import *
 
 async def echo_handler(client, addr):
     print('Connection from', addr)
+    try:
+        client.setsockopt(IPPROTO_TCP, TCP_NODELAY, 1)
+    except (OSError, NameError):
+        pass
     reader, writer = client.make_streams()
     async with reader, writer:
         while True:

servers/gevecho.py

@@ -1,10 +1,15 @@
 # Taken from curio: https://github.com/dabeaz/curio
 
 from gevent.server import StreamServer
+from socket import *
 
 # this handler will be run for each incoming connection in a dedicated greenlet
 def echo(socket, address):
     print('New connection from %s:%s' % address)
+    try:
+        socket.setsockopt(IPPROTO_TCP, TCP_NODELAY, 1)
+    except (OSError, NameError):
+        pass
     while True:
         data = socket.recv(100000)
         if not data:

servers/nodeecho.js

@@ -5,6 +5,7 @@
 
 var net = require('net');
 net.createServer(function(socket){
+    socket.setNoDelay();
     socket.on('data', function(data){
         socket.write(data)
     });

servers/torecho.py

@@ -5,6 +5,7 @@
 class StreamHandler:
     def __init__(self, stream):
         self._stream = stream
+        stream.set_nodelay(True)
         self._stream.read_until_close(None, self._handle_read)
 
     def _handle_read(self, data):

servers/twistedecho.py

@@ -8,6 +8,9 @@
 class Echo(protocol.Protocol):
     """This is just about the simplest possible protocol"""
 
+    def connectionMade(self):
+         self.transport.setTcpNoDelay(True)
+
     def dataReceived(self, data):
         "As soon as any data is received, write it back."
         self.transport.write(data)