selectors
一、Selectors模块
它具有根据平台选出最佳的IO多路机制,比如在win的系统上他默认的是select模式而在linux上它默认的epoll,建议使用selectors。
常用共分为三种:select、poll、epoll
二、用法
selectors server
# -*- coding: UTF-8 -*- import selectors import socket # 生成一个selectors对象 # 根据平台选择最佳的IO多路机制,比如linux就会选择epoll sel = selectors.DefaultSelector() def accept(sock, mask): # 参数scok就是while循环中的key.fileobj conn, addr = sock.accept() # 相当于select 的readable 列表 print('accepted', conn, 'from', addr, '****', mask) conn.setblocking(False) sel.register(conn, selectors.EVENT_READ, read) # 新连接注册read回调函数 def read(conn, mask): data = conn.recv(1024) if data: print('echoing', repr(data), 'to', conn) conn.send(data) else: # 没数据代表断了 print('closing', conn) sel.unregister(conn) conn.close() server = socket.socket() server.bind(('localhost', 9999)) server.listen() server.setblocking(False) # 注册事件,将server注册到sel中,让sel去监听 # accept 只要来一个新连接,就调用这个函数 sel.register(server, selectors.EVENT_READ, accept) while True: events = sel.select() # 默认是阻塞的,有活动连接时返回活动的连接列表,event就是这活动列表 # select(), 可能是select/epoll 看系统支持 print(events) for key, mask in events: callback = key.data # 相当于调用 accept, callback获取了函数地址 callback(key.fileobj, mask) # key.fileobj 是文件句柄
下面开始强行解释了。。。。。
2.1 生成一个selectors对象
sel = selectors.DefaultSelector()
根据平台选择最佳的IO多路机制,比如linux就会选择epoll,windows不支持epoll,所以会用select。
2.2 监听server
server = socket.socket() server.bind(('localhost', 9999)) server.listen() server.setblocking(False)
2.3 注册事件
sel.register(server, selectors.EVENT_READ, accept)
将server注册到sel中,让sel去监听,只要有一个新连接进来,就调用accept函数
2.4 循环事件
while True: events = sel.select() print(events) for key, mask in events: callback = key.data callback(key.fileobj, mask)
循环接收连接和数据,默认是阻塞的。只要有活动连接进来,就返回一个列表events。同时sel.select()可根据系统自动选定epoll/select模式。
events:
[(SelectorKey(fileobj=<socket.socket fd=268, family=AddressFamily.AF_INET, type=SocketKind.SOCK_STREAM, proto=0, laddr=('127.0.0.1', 9999)>, fd=268, events=1, data=<function accept at 0x000002575F87E488>), 1)]
callback = key.data,其实就是accept函数的内存地址
callback() 就是调用accept函数。
key.fileobj ,mask分别是accept函数的两个参数。fileobj很好理解,就是监听的socket的文件句柄。而mask指的是事件类型,EVENT_READ 是 1 ,EVENT_WRITE 是 2。
2.5 accept函数
accept函数的功能类似于 select中的 readable列表,当有一个连接进来时,不立刻接收或发生数据,而是放到一个队列中,再从队列中获取/返回数据。accept函数将监听的连接注册到一个事件中,调用read函数,数据的收发就在read函数中进行。
有数据进来时,events:
[(SelectorKey(fileobj=<socket.socket fd=352, family=AddressFamily.AF_INET, type=SocketKind.SOCK_STREAM, proto=0, laddr=('127.0.0.1', 9999), raddr=('127.0.0.1', 50853)>, fd=352, events=1, data=<function read at 0x000002575F87EA60>), 1)] echoing b'sdfsdf' to <socket.socket fd=352, family=AddressFamily.AF_INET, type=SocketKind.SOCK_STREAM, proto=0, laddr=('127.0.0.1', 9999), raddr=('127.0.0.1', 50853)>
accept函数执行完毕后,对一个连接的监听建立。如果这个连接再发生数据到server端,events中的fileobj就会变成 这个scoket连接的信息,key.data就是read函数的内存地址:
此时,callback(key.fileobj, mask) = read(key.fileobj, mask)
2.6 read函数
对监听的连接收发数据,断开连接等处理。