当前位置：首页 > 文章列表 > 文章 > python教程 > Python selectors.DefaultSelector 怎么收口多路 IO：注册、事件掩码与关闭顺序

Python selectors.DefaultSelector 怎么收口多路 IO：注册、事件掩码与关闭顺序

来源：17golang原创 2026-08-28 06:14:12 0浏览收藏

用 Python 写一个同时照看多条非阻塞 socket 的小服务时，最容易失控的不是监听，而是收尾：事件循环拿到的到底是哪一个连接，连接关闭后是否还留在 selector 里，selector 自己又有没有释放底层资源。selectors.DefaultSelector 的实用边界可以压缩成一句话：先用 register 声明关注的事件，再用 select 取出已就绪对象，连接关闭时先 unregister，最后关闭 selector。

把“监听对象”“就绪事件”和“连接生命周期”分开管理，才能让多路 IO 的循环既不误读，也不会把已关闭的文件描述符留在监控集合里。

要点速览

DefaultSelector 会选择当前平台可用的高效 selector 实现，通常不需要手工指定 epoll 或 kqueue。
EVENT_READ 和 EVENT_WRITE 是位掩码，select(timeout) 返回的是 SelectorKey 与就绪掩码。
key.data 可以保存回调或连接上下文，避免事件循环再维护一套容易错位的映射。
关闭连接前先 unregister(fileobj)，所有连接处理完后再 selector.close()。

DefaultSelector 解决的是哪一层问题

selectors 是 Python 对底层 select 原语的高层封装。DefaultSelector 不是一个固定的内核实现，而是当前平台可用实现中的默认选择。代码只依赖统一的注册、等待和移除接口，Linux、macOS 或其他支持的平台可以由运行环境决定具体实现。

它等待的是“文件对象是否就绪”，这里的文件对象可以是带有 fileno() 的对象，也可以是文件描述符。Windows 对可监控对象的限制与 Unix 不同，不能把“能注册”误解成“所有文件都能按同样方式等待”。

对象	职责	最容易错的边界
`SelectorKey`	保存 fileobj、fd、events 和 data	拿到 key 后仍要判断本次返回的 events
`EVENT_READ`	关注可读事件	可读也可能意味着对端已关闭
`EVENT_WRITE`	关注可写事件	持续关注会造成忙循环
`data`	附着回调或会话上下文	关闭时不能再调用已失效的上下文

注册阶段：把 socket 和处理动作绑在一起

下面的最小示例只处理读事件。监听 socket 的 data 放入 accept_connection，客户端 socket 的 data 放入 read_connection。这样事件循环只需从 key.data 取出动作，不用按文件描述符再查一次字典。

import selectors
import socket

selector = selectors.DefaultSelector()

def accept_connection(server, mask):
    connection, address = server.accept()
    connection.setblocking(False)
    selector.register(connection, selectors.EVENT_READ, read_connection)

def read_connection(connection, mask):
    data = connection.recv(4096)
    if data:
        connection.sendall(data)
    else:
        selector.unregister(connection)
        connection.close()

server = socket.socket()
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
server.bind(("127.0.0.1", 9000))
server.listen()
server.setblocking(False)
selector.register(server, selectors.EVENT_READ, accept_connection)

Python selectors DefaultSelector 从 register 注册 EVENT_READ 到 select 分发回调的调用链 — 注册监听 socket 后，事件循环才能把就绪 key 交给对应回调。

这里有两个细节值得保留。第一，注册前把 socket 设为非阻塞，否则回调里的单次 accept() 或 recv() 仍可能把整个循环卡住。第二，同一个 file object 重复 register() 会触发 KeyError，需要修改事件时使用 modify()，不要偷偷再注册一次。

select 返回什么：先看事件，再决定动作

事件循环通常写成下面这样。select() 返回一组 (key, events)，其中 key.fileobj 是已注册对象，key.data 是注册时附着的数据，events 是这一次真正就绪的位掩码。

while True:
    ready = selector.select(timeout=1)
    for key, events in ready:
        callback = key.data
        callback(key.fileobj, events)

timeout 大于零时表示最长等待秒数；小于等于零时不阻塞；传入 None 则一直等待。收到信号时也可能提前得到空列表，所以空结果不一定代表 selector 出错，循环应把它当成一次没有可处理事件的机会。

如果同时关注读写事件，回调必须继续检查位掩码，而不是看到 key 就无条件执行写操作：

def dispatch(key, events):
    if events & selectors.EVENT_READ:
        read_connection(key.fileobj, events)
    if events & selectors.EVENT_WRITE:
        flush_pending_data(key.fileobj)

Python selectors 连接生命周期从可读事件到 unregister 再到 close 的状态变化 — 对端关闭或本地结束会话时，先从 selector 移除，再关闭 socket。

关闭顺序：unregister 要早于 socket.close

读回空字节通常意味着对端已经有序关闭连接。此时要先执行 selector.unregister(connection)，再执行 connection.close()。官方文档明确要求文件对象关闭前先从 selection 中移除，否则事件集合仍可能保留一个已经失效的对象。

服务整体退出时，还要处理监听 socket 和 selector 本身。一个可复用的收尾函数如下：

def close_connection(connection):
    try:
        selector.unregister(connection)
    except KeyError:
        pass
    finally:
        connection.close()

def shutdown(server):
    close_connection(server)
    selector.close()

这里捕获 KeyError 只适合“收尾动作可能被重复调用”的场景，例如异常分支已经移除了连接。不要用它掩盖正常流程中的重复注册或对象错配。selector.close() 之后不要再次调用 select()、register() 或 get_map()。