selectors

Posted 2020-10-18 Bigberg

一、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函数

 　　对监听的连接收发数据，断开连接等处理。