###

#2# 我们来写一个简单的 Coroutine 框架。

#3# 虽然Python有Coroutine库。

#3# 但是，编程嘛，最主要的是开心！

###

import socket # on top of TCP

import time

from selectors import DefaultSelector, EVENT_WRITE, EVENT_READ

# select: System Call -----> watch the readiness of a unix-file(socket) i/o

# only socket is possible in Windows

# non-blocking socket

selector = DefaultSelector()

class Future: # ~=Promise, return the caller scope a promise

# about something in the future

def __init__(self):

self.callbacks = []

def resolve(self): # on future event callback

for func in self.callbacks:

func()

class Task: # responsible for calling next() on generators

# in charge of the async functions

def __init__(self, gen, eventLoop):

self.gen = gen

self.step()

def step(self): # go to next step/next yield

try:

f = next(self.gen)

f.callbacks.append(self.step)

except StopIteration as e:

# task is finished

eventLoop.n_task -= 1

print('--------------------------------------', 'Byte Received:', e, '\n\n')

class EventLoop:

def __init__(self):

self.n_task = 0

def add_task(self, generator):

self.n_task += 1

Task(generator, self)

def start(self):

while self.n_task:

events = selector.select()

for event, mask in events:

f = event.data

f.resolve()

def async_get(path):

s = socket.socket()

s.setblocking(False)

try:

s.connect(('localhost', 3000))

except BlockingIOError as e:

print(e)

request = 'GET %s HTTP/1.0\r\n\r\n' % path

f = Future()

selector.register(s.fileno(), EVENT_WRITE, data=f)

yield f

# the socket is writable

selector.unregister(s.fileno())

s.send(request.encode())

totalReceived = []

while True:

f = Future()

selector.register(s.fileno(), EVENT_READ, data=f)

yield f

# socket is readable

selector.unregister(s.fileno())

received = s.recv(1000)

if received:

totalReceived.append(received)

else:

body = (b''.join(totalReceived)).decode()

print('--------------------------------------')

print(body)

return len(body)

if __name__ == '__main__':

start = time.time()

eventLoop = EventLoop()

for i in range(20):

eventLoop.add_task(async_get('/super-slow'))

eventLoop.start()

print('%.1f sec' % (time.time() - start))