Netty源码分析第2章(NioEventLoop)---->第7节: 处理IO事件

 

Netty源码分析第二章: NioEventLoop

 

第七节:处理IO事件

 

上一小节咱们了解了执行select()操做的相关逻辑, 这一小节咱们继续学习select()以后, 轮询到io事件的相关逻辑:

回到NioEventLoop的run()方法:

 

protected void run() { for (;;) { try { switch (selectStrategy.calculateStrategy(selectNowSupplier, hasTasks())) { case SelectStrategy.CONTINUE: continue; case SelectStrategy.SELECT: //轮询io事件(1)
                    select(wakenUp.getAndSet(false)); if (wakenUp.get()) { selector.wakeup(); } default: } cancelledKeys = 0; needsToSelectAgain = false; //默认是50
            final int ioRatio = this.ioRatio; if (ioRatio == 100) { try { processSelectedKeys(); } finally { runAllTasks(); } } else { //记录下开始时间
                final long ioStartTime = System.nanoTime(); try { //处理轮询到的key(2)
 processSelectedKeys(); } finally { //计算耗时
                    final long ioTime = System.nanoTime() - ioStartTime; //执行task(3)
                    runAllTasks(ioTime * (100 - ioRatio) / ioRatio); } } } catch (Throwable t) { handleLoopException(t); } //代码省略
 } }

咱们首先看 if (ioRatio == 100) 这个判断, ioRatio主要是用来控制processSelectedKeys()方法执行时间和任务队列执行时间的比例, 其中ioRatio默认是50, 因此会走到下一步else

首先经过 final long ioStartTime = System.nanoTime() 记录下开始时间, 再经过processSelectedKeys()方法处理轮询到的key

咱们跟到processSelectedKeys()方法中:

private void processSelectedKeys() { if (selectedKeys != null) { //flip()方法会直接返回key的数组
 processSelectedKeysOptimized(selectedKeys.flip()); } else { processSelectedKeysPlain(selector.selectedKeys()); } }

咱们知道selector经过netty优化以后, 会初始化 selectedKeys这个属性, 因此这个属性不为空就会走到 processSelectedKeysOptimized(selectedKeys.flip()) 方法, 这个方法就是对应优化过的selector进行操做的

若是是非优化的selector, 则会进入 processSelectedKeysPlain(selector.selectedKeys()) 方法

 

selectedKeys.flip()为selectedKey中绑定的数组, 咱们以前小节讲过selectedKeys实际上是经过数组存储的, 因此通过select()操做若是监听到事件selectedKeys的数组就会有值

跟进到processSelectedKeysOptimized(selectedKeys.flip())方法中:

private void processSelectedKeysOptimized(SelectionKey[] selectedKeys) { //经过for循环遍历数组
    for (int i = 0;; i ++) { //拿到当前的selectionKey
        final SelectionKey k = selectedKeys[i]; if (k == null) { break; } //将当前引用设置为null
        selectedKeys[i] = null; //获取channel(NioSeverSocketChannel)
        final Object a = k.attachment(); //若是是AbstractNioChannel, 则调用processSelectedKey()方法处理io事件
        if (a instanceof AbstractNioChannel) { processSelectedKey(k, (AbstractNioChannel) a); } else { @SuppressWarnings("unchecked") NioTask<SelectableChannel> task = (NioTask<SelectableChannel>) a; processSelectedKey(k, task); } //代码省略
 } }

首先经过for循环遍历数组中的每个key, 得到key以后首先将数组中对应的下标清空, 由于selector不会自动清空, 这与咱们使用原生selector时候, 经过遍历selector.selectedKeys()的set的时候, 拿到key以后要执行remove()是一个意思

以后获取注册在key上的channel, 判断channel是否是AbstractNioChannel, 一般状况都是AbstractNioChannel, 因此这里会执行 processSelectedKey(k, (AbstractNioChannel) a) 

跟到processSelectedKey(k, (AbstractNioChannel) a)方法中:

private void processSelectedKey(SelectionKey k, AbstractNioChannel ch) { //获取到channel中的unsafe
    final AbstractNioChannel.NioUnsafe unsafe = ch.unsafe(); //若是这个key不是合法的, 说明这个channel可能有问题
    if (!k.isValid()) { //代码省略
 } try { //若是是合法的, 拿到key的io事件
        int readyOps = k.readyOps(); //连接事件
        if ((readyOps & SelectionKey.OP_CONNECT) != 0) { int ops = k.interestOps(); ops &= ~SelectionKey.OP_CONNECT; k.interestOps(ops); unsafe.finishConnect(); } //写事件
        if ((readyOps & SelectionKey.OP_WRITE) != 0) { ch.unsafe().forceFlush(); } //读事件和接受连接事件 //若是当前NioEventLoop是work线程的话, 这里就是op_read事件 //若是是当前NioEventLoop是boss线程的话, 这里就是op_accept事件
        if ((readyOps & (SelectionKey.OP_READ | SelectionKey.OP_ACCEPT)) != 0 || readyOps == 0) { unsafe.read(); if (!ch.isOpen()) { return; } } } catch (CancelledKeyException ignored) { unsafe.close(unsafe.voidPromise()); } }

咱们首先获取和channel绑定的unsafe, 以后拿到channel注册的事件

 

咱们关注 if ((readyOps & (SelectionKey.OP_READ | SelectionKey.OP_ACCEPT)) != 0 || readyOps == 0) 这个判断, 这个判断相信注释上写的很明白, 若是当前NioEventLoop是work线程的话, 这里就是op_read事件, 若是是当前NioEventLoop是boss线程的话, 这里就是op_accept事件

 

而后会经过channel绑定的unsafe对象执行read()方法用于处理连接或者读写事件

以上就是NioEventLoop对io事件的处理过程, 有关read()方法执行逻辑, 会在之后的章节中详细剖析

 

上一节: 执行select操做

下一节: 执行任务队列