問(wèn)題描述

場(chǎng)景：一個(gè)服務(wù)端A，一個(gè)客戶(hù)端B，存在一個(gè)socket連接。現(xiàn)在寫(xiě)的是客戶(hù)端B部分，服務(wù)端不可控。原來(lái)是 B先發(fā)送一個(gè)包，等待A返回指定內(nèi)容，B再發(fā)送下一個(gè)包

def do(): s.send(...) yield 1 s.send(...) yield 2# 接收到數(shù)據(jù)后的回調(diào)def callback(): global f next(f) f=do()next(f)

現(xiàn)在想實(shí)現(xiàn)一個(gè)timeout，并且實(shí)現(xiàn)阻塞。B發(fā)送數(shù)據(jù)后阻塞，直到A返回?cái)?shù)據(jù)(或5秒內(nèi)未接受到來(lái)自A的返回raise一個(gè)錯(cuò)誤)，請(qǐng)教如何實(shí)現(xiàn)？

問(wèn)題解答

用 Tornado 的話(huà)，寫(xiě)不了幾行代碼吧。

先作個(gè)簡(jiǎn)單的 Server ，以方便演示：

# -*- coding: utf-8 -*-from tornado.ioloop import IOLoopfrom tornado.tcpserver import TCPServerfrom tornado import genclass Server(TCPServer): @gen.coroutine def handle_stream(self, stream, address):while 1: data = yield stream.read_until(’n’) if data.strip() == ’exit’:stream.close()break if data.strip() == ’5’:IOLoop.current().call_at(IOLoop.current().time() + 5, lambda: stream.write(’ok 5n’)) else:stream.write(’okn’)if __name__ == ’__main__’: Server().listen(8000) IOLoop.current().start()

然后，來(lái)實(shí)現(xiàn) Client ，基本邏輯是，超時(shí)就關(guān)閉連接，然后再重新建立連接：

# -*- coding: utf-8 -*-import functoolsfrom tornado.ioloop import IOLoopfrom tornado.tcpclient import TCPClientfrom tornado import gendef when_error(stream): print ’ERROR’ stream.close() main()@gen.coroutinedef main(): client = TCPClient() stream = yield client.connect(’localhost’, 8000) count = 0 IL = IOLoop.current() while 1:count += 1stream.write(str(count) + ’n’)print count, ’...’timer = IL.call_at(IL.time() + 4, functools.partial(when_error, stream))try: data = yield stream.read_until(’n’)except: breakIL.remove_timeout(timer)print datayield gen.Task(IL.add_timeout, IOLoop.current().time() + 1)if __name__ == ’__main__’: main() IOLoop.current().start()