关于python使用epoll进行监听处理事件的demo

关于python使用epoll进行监听处理事件的demo

import select
import socket

# 创建 epoll 对象
epoll = select.epoll()

# 创建套接字
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_address = ('127.0.0.1', 8080)
server_socket.bind(server_address)
server_socket.listen(10)

# 将服务器套接字添加到 epoll 中，监听读事件
epoll.register(server_socket.fileno(), select.EPOLLIN)

connections = {}

try:
    while True:
        # 等待事件发生
        events = epoll.poll(1)

        for fileno, event_type in events:
            if fileno == server_socket.fileno():
                # 有新连接
                client_socket, client_address = server_socket.accept()
                print(f"New connection from {client_address}")
                # 将新连接的套接字添加到 epoll 中
                epoll.register(client_socket.fileno(), select.EPOLLIN)
                connections[client_socket.fileno()] = client_socket
            elif event_type & select.EPOLLIN:
                # 可读事件
                client_socket = connections[fileno]
                data = client_socket.recv(1024)
                if data:
                    print(f"Received data: {data.decode('utf-8')}")
                else:
                    # 客户端关闭连接
                    epoll.unregister(fileno)
                    client_socket.close()
                    del connections[fileno]
finally:
    epoll.close()
    server_socket.close()

epoll = select.epoll()：这就是创建了 epoll 对象，相当于 epoll_create 的操作。

相当于在底层完成了与 epoll_create 类似的初始化操作。它为后续的事件监控和管理创建了一个基础的数据结构。

epoll.register(server_socket.fileno(), select.EPOLLIN)：

这里register 方法实际上就类似于执行了 epoll_ctl 来添加要监控的文件描述符和事件类型。

当执行 epoll.register(server_socket.fileno(), select.EPOLLIN) 时，就如同执行了 epoll_ctl 的功能。这里将服务器套接字的文件描述符以及要监控的事件类型（这里是可读事件 select.EPOLLIN）传递给 epoll 对象，让其将该套接字加入到监控列表中，并且指定了具体的事件类型。这就类似于告诉 epoll 模型要关注这个套接字的可读状态变化。

events = epoll.poll(1)：这就是在执行 epoll_wait ，等待事件的发生并获取相关信息。

events = epoll.poll(1) 这一行就是关键的等待事件发生的操作，相当于 epoll_wait。它会阻塞（如果没有事件发生）直到有事件就绪或者超时（这里设置超时时间为 1 秒）。当有事件发生时，它会返回一个包含就绪事件的列表，其中每个元素包含文件描述符和对应的事件类型。然后在后续代码中，就可以根据这些信息进行相应的处理，比如处理新连接或读取数据等操作。