python异步回调转为同步并实现超时

Question

场景：一个服务端A，一个客户端B，存在一个socket连接。现在写的是客户端B部分，服务端不可控。原来是 B先发送一个包，等待A返回指定内容，B再发送下一个包 {代码...} 现在想实现一个timeout，并且实现阻塞。B发...

黄舟 · Answer

用 Tornado 的话，写不了几行代码吧。

先作个简单的 Server ，以方便演示：

# -*- coding: utf-8 -*-

from tornado.ioloop import IOLoop
from tornado.tcpserver import TCPServer
from tornado import gen

class Server(TCPServer):
    @gen.coroutine
    def handle_stream(self, stream, address):
        while 1:
            data = yield stream.read_until('
')

            if data.strip() == 'exit':
                stream.close()
                break

            if data.strip() == '5':
                IOLoop.current().call_at(IOLoop.current().time() + 5, lambda: stream.write('ok 5
'))
            else:
                stream.write('ok
')


if __name__ == '__main__':
    Server().listen(8000)
    IOLoop.current().start()

然后，来实现 Client ，基本逻辑是，超时就关闭连接，然后再重新建立连接：

# -*- coding: utf-8 -*-

import functools
from tornado.ioloop import IOLoop
from tornado.tcpclient import TCPClient
from tornado import gen


def when_error(stream):
    print 'ERROR'
    stream.close()
    main()

@gen.coroutine
def main():
    client = TCPClient()
    stream = yield client.connect('localhost', 8000)

    count = 0
    IL = IOLoop.current()
    while 1:
        count += 1
        stream.write(str(count) + '
')
        print count, '...'

        timer = IL.call_at(IL.time() + 4, functools.partial(when_error, stream))

        try:
            data = yield stream.read_until('
')
        except:
            break

        IL.remove_timeout(timer)

        print data
        yield gen.Task(IL.add_timeout, IOLoop.current().time() + 1)



if __name__ == '__main__':
    main()
    IOLoop.current().start()