Python3.8/Linux下用asyncio为不同FD设置不同Poll事件掩码可行吗？

问题描述 #

需要为两个不同的文件描述符（FD）设置不同的监听事件掩码：

• 监听fd_1的POLLPRI | POLLERR事件
• 监听fd_2的POLLIN事件

当前实现仅能通过自定义PollSelector固定_EVENT_READ掩码来监听单个FD的特定事件，但使用asyncio的add_reader时，所有FD都会复用这个固定掩码，无法为第二个FD单独设置POLLIN。直接调用selector.register(fd_2, select.POLLIN)又无法关联事件循环的回调函数，同时疑惑为何asyncio未暴露poll/epoll的完整功能。

当前代码示例：

import asyncio
import selectors
import select

class CustomPollSelector(selectors.PollSelector):
    _selector_cls = select.poll
    _EVENT_READ = select.POLLPRI | select.POLLERR
    _EVENT_WRITE = select.POLLOUT

selector = CustomPollSelector()
event_loop = asyncio.SelectorEventLoop(selector)
...
# 监听fd_1的select.POLLPRI | select.POLLERR事件
event_loop.add_reader(fd_1, read_data_cb)
...
# TODO: 如何仅监听另一个fd的POLLIN事件？
...
event_loop.run_forever()

尝试过直接注册但无法关联回调：

selector.register(fd_2, select.POLLIN)

解决方案 #

核心思路 #

asyncio的高层API（add_reader/add_writer）为了跨平台兼容性，屏蔽了底层多路复用器的事件掩码细节，统一用"读"/"写"抽象事件。要实现不同FD的自定义掩码，需要直接操作底层selector并手动关联回调到事件循环。

实现代码 #

import asyncio
import selectors
import select

class CustomPollSelector(selectors.PollSelector):
    _selector_cls = select.poll

# 初始化selector和事件循环
selector = CustomPollSelector()
loop = asyncio.SelectorEventLoop(selector)

# 定义两个FD的回调函数
def handle_fd1():
    # 处理fd_1的POLLPRI/POLLERR事件逻辑
    read_data_cb()
    # 重新注册事件（poll事件触发后会自动取消注册，需重新添加）
    selector.modify(fd_1, select.POLLPRI | select.POLLERR)

def handle_fd2():
    # 处理fd_2的POLLIN事件逻辑
    read_data_fd2_cb()
    selector.modify(fd_2, select.POLLIN)

# 注册FD和对应的事件掩码，将回调函数存入data字段
selector.register(fd_1, select.POLLPRI | select.POLLERR, data=handle_fd1)
selector.register(fd_2, select.POLLIN, data=handle_fd2)

# 自定义事件轮询任务，处理selector返回的事件
async def poll_events():
    while True:
        # 阻塞等待事件触发，timeout设为1秒避免完全阻塞
        events = selector.select(timeout=1)
        for key, mask in events:
            # 将回调放入事件循环的任务队列执行
            loop.call_soon(key.data)

# 启动事件轮询任务
loop.create_task(poll_events())
loop.run_forever()

关键说明 #

1. 直接操作selector：跳过asyncio的add_reader，直接用selector.register为每个FD设置自定义事件掩码，并将回调函数存在data字段中。
2. 事件重注册：poll/epoll的事件触发后会自动解除注册，所以回调执行完需要用selector.modify重新注册事件，保证持续监听。
3. 回调入队：通过loop.call_soon将回调函数放入事件循环的任务队列执行，确保回调在asyncio的上下文运行，避免线程安全问题。

关于asyncio未暴露完整功能的原因 #

asyncio的设计目标是提供跨平台的异步编程抽象，高层API需要兼容Windows、macOS等不同系统的多路复用器（如select、kqueue），这些系统的事件模型差异较大。如果直接暴露底层的POLL*掩码，会破坏跨平台兼容性，因此选择用统一的"读"/"写"事件来封装底层细节。对于需要底层特性的场景，开发者可以直接操作selector对象来实现自定义需求。

内容的提问来源于stack exchange，提问作者sergs