select、poll、epoll的底层原理剖析

引言

select/poll、epoll 这些词汇相信诸位都不陌生，因为在 Redis/Nginx/Netty 等一些高性能技术栈的底层原理中，大家应该都见过它们的身影，接下来重点讲解这块内容，不过在此之前，先上一张图概述 Java-NIO 的整体结构：

观察上述结构，其实 Buffer、Channel 的定义并不算复杂，仅是单纯的三层结构，因此对于源码这块不再去剖析，有兴趣的根据给出的目录结构去调试源码，自然也能摸透其原理实现。

而最关键的是 Selector 选择器，它是整个 NIO 体系中较为复杂的一块内容，同时它也作为 Java-NIO 与内核多路复用模型的“中间者”，但在上述体系中，却出现了之前未曾提及过的 SelectorProvider 系定义，那么它的作用是干嘛的呢？主要目的是用于创建选择器，在Java中创建一般是通过如下方式：

// 创建Selector选择器
Selector selector = Selector.open();

// Selector类 → open()方法
public static Selector open() throws IOException {
    return SelectorProvider.provider().openSelector();
}
复制代码

从源码中可明显得知，选择器最终是由 SelectorProvider 去进行实例化，不过值得一提的是： Selector 的实现是基于工厂模式与 SPI 机制构建的。对于不同 OS 而言，其对应的具体实现并不相同，因此在 Windows 系统下，我们只能观测到 WindowsSelectorXXX 这一系列的实现，而在 Linux 系统时，对于的则是 EPollSelectorXXX 这一系列的实现，所以要牢记的是， Java-NIO 在不同操作系统的环境中，提供了不同的实现 ，如下：

• Windows ： select
• Unix ： poll
• Mac ： kqueue
• Linux ： epoll

当然，本次则重点剖析Linux系统下的 select、poll、epoll 的具体实现，对于其他系统而言，原理大致相同。

一、JDK层面的源码入口

简单的对于 Java-NIO 体系有了全面认知后，接下来以 JDK 源码作为入口进行剖析。在Java中，会通过 Selector.select() 方法去监听事件是否被触发，如下：

// 轮询监听选择器上注册的通道是否有事件被触发
while (selector.select() > 0){}

// Selector抽象类 → select()抽象方法
public abstract int select() throws IOException;

// SelectorImpl类 → select()方法
public int select() throws IOException {
    return this.select(0L);
}
// SelectorImpl类 → select()完整方法
public int select(long var1) throws IOException {
    if (var1 < 0L) {
        throw new IllegalArgumentException("Negative timeout");
    } else {
        return this.lockAndDoSelect(var1 == 0L ? -1L : var1);
    }
}
复制代码

当调用 Selector.select() 方法后，最终会调用到 SelectorImpl 类的 select(long var1) 方法，而在该方法中，又会调用 lockAndDoSelect() 方法，如下：

// SelectorImpl类 → lockAndDoSelect()方法
private int lockAndDoSelect(long var1) throws IOException {
    // 先获取锁确保线程安全
    synchronized(this) {
        // 在判断当前选择是否处于开启状态
        if (!this.isOpen()) {
            // 如果已关闭则抛出异常
            throw new ClosedSelectorException();
        } else { // 如若处于开启状态
            // 获取所有注册在当前选择器上的事件
            Set var4 = this.publicKeys;
            int var10000;
            // 再次加锁
            synchronized(this.publicKeys) {
                // 获取所有已就绪的事件
                Set var5 = this.publicSelectedKeys;
                // 再次加锁
                synchronized(this.publicSelectedKeys) {
                    // 真正的调用select逻辑，获取已就绪的事件
                    var10000 = this.doSelect(var1);
                }
            }
            // 返回就绪事件的数量
            return var10000;
        }
    }
}
复制代码

在该方法中，对于其他逻辑不必太过在意，重点可注意：最终会调用 doSelect() 触发真正的逻辑操作，接下来再看看这个方法：

// SelectorImpl类 → doSelect()方法
protected abstract int doSelect(long var1) throws IOException;

// WindowsSelectorImpl类 → doSelect()方法
protected int doSelect(long var1) throws IOException {
    // 先判断一下选择器上是否还有注册的通道
    if (this.channelArray == null) {
        throw new ClosedSelectorException();
    } else { // 如果有的话
        // 先获取一下阻塞等待的超时时长
        this.timeout = var1;
        // 然后将一些取消的事件从选择器上移除
        this.processDeregisterQueue();
        // 再判断一下是否存在线程中断唤醒
        // 这里主要是结合之前的wakeup()方法唤醒阻塞线程的
        if (this.interruptTriggered) {
            this.resetWakeupSocket();
            return 0;
        } else { // 如果没有唤醒阻塞线程的需求出现
            // 先判断一下辅助线程的数量（守护线程），多则减，少则增
            this.adjustThreadsCount();
            // 更新一下finishLock.threadsToFinish为辅助线程数
            this.finishLock.reset();
            // 唤醒所有的辅助线程
            this.startLock.startThreads();
            try {
                // 设置主线程中断的回调函数
                this.begin();

                try {
                    // 最终执行真正的poll逻辑，开始拉取事件
                    this.subSelector.poll();
                } catch (IOException var7) {
                    this.finishLock.setException(var7);
                }
                // 唤醒并等待所有未执行完的辅助线程完成
                if (this.threads.size() > 0) {
                    this.finishLock.waitForHelperThreads();
                }
            } finally {
                this.end();
            }
            // 检测状态
            this.finishLock.checkForException();
            this.processDeregisterQueue();
            // 获取当前选择器监听的事件的触发数量
            int var3 = this.updateSelectedKeys();
            // 本轮poll结束，重置WakeupSocket，为下次执行做准备
            this.resetWakeupSocket();
            // 最终返回获取到的事件数
            return var3;
        }
    }
}
复制代码

整个过程下来其实也并不短暂，但大体就分为三步：

poll

在这里面࿰