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操做

下一節: 執行任務隊列