io_uring 的异步事件轮询（IORING_OP_POLL_ADD）：构建百万并发长连接事件驱动服务器的底层解密

在即时通讯（IM）、长连接网关、物联网（IoT）传感器汇聚节点等海量并发接入场景下，单台物理机通常需要同时维持数十万乃至数百万条处于闲置状态（Idle）的 TCP 长连接。在过去的架构体系中，唯一的标准解法是基于 epoll 的事件多路复用。然而，随着并发连接数突破 100 万大关，epoll 的底层数据结构与控制流设计逐渐显露出其固有的性能天花板：基于红黑树的事件监听表在高频增删时伴随着全局互斥锁竞争，频繁的 epoll_ctl 系统调用在用户态与内核态之间撕扯，导致系统无法自拔地陷入特权级切换开销中。

io_uring 提供的异步事件轮询操作符——IORING_OP_POLL_ADD（以及现代内核支持的 IORING_POLL_ADD_MULTI），提供了一种彻底解耦且性能更强劲的替代架构。它允许开发者像提交普通异步读写请求一样，向环形无锁提交队列（SQ）投递网络套接字的事件监听意图，内核全权接管等待队列驱动，并在事件就绪时将结果统一输出至完成队列（CQ）。这不仅抹平了文件 I/O 与网络事件之间的割裂，更为百万级并发服务器赋予了前所未有的吞吐能力。

底层机理透视：IORING_OP_POLL_ADD vs epoll_ctl

要理解 IORING_OP_POLL_ADD 的威力，必须从内核等待队列（Wait Queue）的挂载路径切入：

[百万客户端 TCP 连接池]
|
[提交 IORING_OP_POLL_ADD 批量任务] (通过用户态无锁环形队列 SQ 零拷贝提交)
|
[内核驱动层] —> 直接将 io_poll_iocb 注册进 sk->sk_sleep
|
(网络数据包到达 / 唤醒中断)
v
[内核等待队列回调 io_poll_wake] —> 生成就绪事件
|
+—> 若配置了 MULTI: 保持挂载，产生 CQE (设置 IORING_CQE_F_MORE)
+—> 若未配置 MULTI: 解除挂载，产生单次 CQE
|
[用户态环形队列 CQ] <— 应用程序批量批量捞取，0 次系统调用系统陷入

C23 标准实战：百万长连接事件轮询驱动核心

下面给出符合 C23 标准的高性能网络事件驱动服务核心实现。代码运用了显式常量、内存对齐与系统调用包装器，演示如何批量注册上万个套接字的事件监听并实现持久化轮询。

#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <stdbool.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <poll.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <linux/io_uring.h>
#include <sys/syscall.h>
#include <sys/mman.h>

constexpr size_t QUEUE_DEPTH = 4096;
constexpr size_t BATCH_SUBMIT = 64;

static inline int io_uring_setup(uint32_t entries, struct io_uring_params *p) {
return (int)syscall(__NR_io_uring_setup, entries, p);
}

static inline int io_uring_enter(int fd, uint32_t to_submit, uint32_t min_complete,
uint32_t flags, sigset_t *sig) {
return (int)syscall(__NR_io_uring_enter, fd, to_submit, min_complete, flags, sig);
}

// 环形队列指针管理结构体
typedef struct {
int ring_fd;
uint32_t *sq_head;
uint32_t *sq_tail;
uint32_t *sq_ring_mask;
uint32_t *sq_array;
struct io_uring_sqe *sqes;

uint32_t *cq_head;
uint32_t *cq_tail;
uint32_t *cq_ring_mask;
struct io_uring_cqe *cqes;
} UringContext;

// 注册套接字的持久化异步事件轮询 (Poll Add Multi)
bool add_socket_poll_multishot(UringContext *ctx, int socket_fd, uint32_t poll_mask) {
uint32_t tail = *ctx->sq_tail;
uint32_t head = *ctx->sq_head;
uint32_t mask = *ctx->sq_ring_mask;

if (tail – head >= QUEUE_DEPTH) {
return false; // 队列溢出
}

uint32_t index = tail & mask;
struct io_uring_sqe *sqe = &ctx->sqes[index];
memset(sqe, 0, sizeof(*sqe));

sqe->opcode = IORING_OP_POLL_ADD;
sqe->fd = socket_fd;
sqe->poll_events = poll_mask; // 如 POLLIN | POLLRDHUP
sqe->len = IORING_POLL_ADD_MULTI; // 关键：开启持久化持续监听
sqe->user_data = (uint64_t)socket_fd; // 携带上下文句柄

ctx->sq_array[index] = index;
__atomic_store_n(ctx->sq_tail, tail + 1, __ATOMIC_RELEASE);
return true;
}

// 批量收割已就绪事件并执行零拷贝 I/O 调度
size_t reap_ready_events(UringContext *ctx, size_t max_events) {
uint32_t head = *ctx->cq_head;
uint32_t tail = __atomic_load_n(ctx->cq_tail, __ATOMIC_ACQUIRE);
uint32_t mask = *ctx->cq_ring_mask;
size_t reaped = 0;

while (head != tail && reaped < max_events) {
uint32_t index = head & mask;
struct io_uring_cqe *cqe = &ctx->cqes[index];

int socket_fd = (int)cqe->user_data;
int res = cqe->res;

if (res < 0) {
fprintf(stderr, "[-] 套接字 %d 轮询异常: %s\\n", socket_fd, strerror(-res));
} else {
uint32_t revents = (uint32_t)res;
if (revents & POLLIN) {
// 套接字可读，在此处可联动提交 IORING_OP_RECV 或执行用户态协议解析
// 此时数据已就绪，无需再调用 epoll_wait
}
if (revents & (POLLERR | POLLHUP | POLLRDHUP)) {
// 连接断开或异常，关闭套接字
close(socket_fd);
}
}

// 检查是否仍然具备多重监听效力
if (!(cqe->flags & IORING_CQE_F_MORE)) {
// 若未设置 F_MORE，说明该套接字单次监听已失效或遇到终止错误
}

head++;
reaped++;
}

__atomic_store_n(ctx->cq_head, head, __ATOMIC_RELEASE);
return reaped;
}

百万并发实战中的架构调优与内存边界

在实际压测 100 万长连接并发环境时，必须特别注意以下两项系统级调优：

通过将网络事件感知彻底融为 io_uring 统一环形管道的一部分，我们成功摒弃了传统网络服务中“epoll 监听 + 同步/非阻塞读写”的双阶段割裂架构，真正实现了面向百万连接的极简、极速底层数据驱动。