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

更新時間：2022年03月25日 15:03:25 作者：向南是個萬人迷

這篇文章主要介紹了Netty源碼分析NioEventLoop處理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;
                    //執(zhí)行task(3)
                    runAllTasks(ioTime * (100 - ioRatio) / ioRatio);
                }
            }
        } catch (Throwable t) {
            handleLoopException(t);
        }
        //代碼省略
    }
}

我們首先看 if (ioRatio == 100) 這個判斷, ioRatio主要是用來控制processSelectedKeys()方法執(zhí)行時間和任務隊列執(zhí)行時間的比例, 其中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優(yōu)化之后, 會初始化 selectedKeys這個屬性, 所以這個屬性不為空就會走到 processSelectedKeysOptimized(selectedKeys.flip()) 方法, 這個方法就是對應優(yōu)化過的selector進行操作的

如果是非優(yōu)化的selector, 則會進入 processSelectedKeysPlain(selector.selectedKeys()) 方法

selectedKeys.flip()為selectedKey中綁定的數組, 我們之前小節(jié)講過selectedKeys其實是通過數組存儲的, 所以經過select()操作如果監(jiān)聽到事件selectedKeys的數組就會有值

processSelectedKeysOptimized(selectedKeys.flip())方法

private void processSelectedKeysOptimized(SelectionKey[] selectedKeys) {
    //通過for循環(huán)遍歷數組
    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循環(huán)遍歷數組中的每一個key, 獲得key之后首先將數組中對應的下標清空, 因為selector不會自動清空, 這與我們使用原生selector時候, 通過遍歷selector.selectedKeys()的set的時候, 拿到key之后要執(zhí)行remove()是一個意思

之后獲取注冊在key上的channel, 判斷channel是不是AbstractNioChannel, 通常情況都是AbstractNioChannel, 所以這里會執(zhí)行 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注冊的事件

我們關注