selectors模块 - 实现多路复用简单介绍

常见的多路复用有select、epoll、poll方式。但并不是每种操作系统都能使用，比如：windows操作系统不支持使用epoll来实现多路复用。而selectors模块能根据操作系统来判断选择最好的多路复用方式，相当于一个通用的接口。重点看代码注释

import selectors
import socket

# windows不能用epoll,selectors能根据操作系统来选择最好的多路复用方式select或epoll
sel = selectors.DefaultSelector()

# 如果是sock时，就执行accpet操作
def accept(sock, mask):
    conn, addr = sock.accept()
    conn.setblocking(False)  # 设置非阻塞
    sel.register(conn, selectors.EVENT_READ, read)  # 注册绑定conn

# 如果是conn时，就执行read函数
def read(conn, mask):
    try:
        data = conn.recv(1024)
        if not data:
            raise Exception
        conn.send(data)  # 接收有数据就发送数据
    except Exception as e:
        # 没有数据就解除绑定，关闭conn
        sel.unregister(conn)  # 解除绑定
        conn.close()  # 关闭conn


sock = socket.socket()
sock.bind(('127.0.0.1', 8090))
sock.listen(5)
sock.setblocking(False)

# 只是注册，将sock与accept绑定,意思是sock发生变化时就执行accept。人如果这行代码后面没有代码就什么都不执行
sel.register(sock, selectors.EVENT_READ, accept)

print('server...')
while True:
    events = sel.select()  # 真正的监听，[sock,[conn,.....]]有变化往下走
    for key, mask in events:
        callback = key.data  
        # 可以打印callback,发现callback拿到的是函数名accept或read
        callback(key.fileobj, mask)  # key.fileobj是socket对象