首頁 > 軟體

Netty原始碼分析NioEventLoop處理IO事件相關邏輯

2022-03-24 22:00:06

之前我們瞭解了執行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()方法執行邏輯, 會在以後的章節中詳細剖析,更多關於Netty NioEventLoop處理IO事件邏輯的資料請關注it145.com其它相關文章!


IT145.com E-mail:sddin#qq.com