如何使用Python中的selector()函数

Python中的selector()函数是在标准库selectors中提供的。它提供了高级的I/O多路复用机制，可以同时监控多个I/O对象的状态，并通过回调函数处理这些对象的事件。selector()函数返回一个Selector对象，可以使用Selector对象的方法进行注册和监控。

使用selector()函数需要先创建一个Selector对象，然后使用register()方法注册要监控的文件描述符和事件类型，最后使用select()方法进行事件监听。

下面是一个使用selector()函数的例子：

import selectors
import socket

# 创建Selector对象
sel = selectors.DefaultSelector()

def accept(sock, mask):
    # 接受连接并注册读事件
    conn, addr = sock.accept()
    print('accepted', conn, 'from', addr)
    conn.setblocking(False)
    sel.register(conn, selectors.EVENT_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()

# 创建服务器socket并设置为非阻塞模式
sock = socket.socket()
sock.bind(('localhost', 8888))
sock.listen()
sock.setblocking(False)

# 注册服务器socket上的连接事件
sel.register(sock, selectors.EVENT_READ, accept)

while True:
    # 监听事件并调用回调函数处理
    events = sel.select()
    for key, mask in events:
        callback = key.data
        callback(key.fileobj, mask)

这个例子是一个简单的回显服务器，使用selector()函数实现了同时监听多个客户端连接的能力。首先创建了一个Selector对象，然后通过register()方法注册了服务器socket的监听事件。在回调函数accept()中，通过accept()方法接受连接，将连接设置为非阻塞模式，并注册读事件。在回调函数read()中，通过recv()方法接收数据并发送回复，当连接关闭时取消注册。

最后，在主循环中使用select()方法进行事件监听，当有事件发生时，通过回调函数处理。

使用selector()函数可以方便地实现高性能的I/O编程，特别适合处理大量的并发连接。