Learning Python Coroutine

2019/10/17

Tags: python coroutine

看了一个视频 OSB 2015 - How Do Python Coroutines Work?,从头开始讲 coroutine 是怎么抽象出来的,感觉好厉害。自己写了一点程序学习了一下。之前写的关于 coroutine 的帖子。

先准备一个 server.py ,可以接受客户端请求。要注意的是要使用 Threading ,或者 fork 的 server,要不服务端执行并不支持并发,需要处理完一个才能处理下一个,这样会发现虽然客户端那边请求是并发的,但是返回结果的时候是顺序的。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
import socketserver
from time import sleep

HOST = '127.0.0.1'
PORT = 2045

class MyTCPHandler(socketserver.BaseRequestHandler):
    def handle(self):
        data = self.request.recv(1024)
        sleep(1)
        self.request.sendall('{} ok'.format(data.decode()).encode())


def run():
    server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)
    server.serve_forever()

if __name__ == '__main__':
    run()

然后是 client.py ,可以比较下里面 sync_callasync_call 的区别。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import socket
from time import time
import asyncio

HOST = '127.0.0.1'
PORT = 2045

start = time()

def get(path):
    print('get {}'.format(path))
    s = socket.socket()
    s.connect((HOST, PORT))

    s.sendall(path.encode())

    chunk = s.recv(100)
    print('get {}'.format(chunk.decode()))

def sync_call():
    get('/bar')
    get('/foo')


async def async_get(path):
    print('get {}'.format(path))
    reader, writer = await asyncio.open_connection(HOST, PORT)

    writer.write(path.encode())
    await writer.drain()

    chunk = await reader.read(100)
    print('get {}'.format(chunk.decode()))

def async_call():
    loop = asyncio.get_event_loop()

    t1 = loop.create_task(async_get('/bar'))
    t2 = loop.create_task(async_get('/foo'))

    loop.run_until_complete(asyncio.gather(t1, t2))

if __name__ == "__main__":
    #sync_call()
    async_call()
    print('total {:.2f} seconds'.format(time() - start))

这里网络请求也需要使用非阻塞的库。创建 task 实际有挺多方法的,这里随便用了一个。

Comments