latency-tornado.py

from tornado.tcpserver import TCPServer
from tornado.tcpclient import TCPClient
from tornado.ioloop import IOLoop
from tornado.iostream import StreamClosedError
from tornado import gen
from distributed.utils import format_time
from time import time

class EchoServer(TCPServer):
    @gen.coroutine
    def handle_stream(self, stream, address):
        while True:
            try:
                data = yield stream.read_bytes(1)
                yield stream.write(data)
            except StreamClosedError:
                break


@gen.coroutine
def f():
    port = 1234
    n = 1000
    server = EchoServer()
    server.bind(port)
    server.start()

    client = TCPClient()
    stream = yield client.connect('localhost', port)

    start = time()
    for i in range(n):
        yield stream.write(b'0')
        yield stream.read_bytes(1)
    stop = time()
    print('duration', format_time((stop - start) / n))  # I get around 200us on my machine


if __name__ == '__main__':
    IOLoop().run_sync(f)