Netty分布式NioSocketChannel注冊(cè)到selector方法的示例分析

這篇文章給大家分享的是有關(guān)Netty分布式NioSocketChannel注冊(cè)到selector方法的示例分析的內(nèi)容。小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，一起跟隨小編過來看看吧。

我們回到最初的NioMessageUnsafe的read()方法:

public void read() {
    //必須是NioEventLoop方法調(diào)用的, 不能通過外部線程調(diào)用
    assert eventLoop().inEventLoop();
    //服務(wù)端channel的config
    final ChannelConfig config = config();
    //服務(wù)端channel的pipeline
    final ChannelPipeline pipeline = pipeline();
    //處理服務(wù)端接入的速率
    final RecvByteBufAllocator.Handle allocHandle = unsafe().recvBufAllocHandle();
    //設(shè)置配置
    allocHandle.reset(config);
    boolean closed = false;
    Throwable exception = null;
    try {
        try {
            do {
                //創(chuàng)建jdk底層的channel
                //readBuf用于臨時(shí)承載讀到鏈接
                int localRead = doReadMessages(readBuf);
                if (localRead == 0) {
                    break;
                }
                if (localRead < 0) {
                    closed = true;
                    break;
                }
                //分配器將讀到的鏈接進(jìn)行計(jì)數(shù)
                allocHandle.incMessagesRead(localRead);
                //連接數(shù)是否超過最大值
            } while (allocHandle.continueReading());
        } catch (Throwable t) {
            exception = t;
        }
        int size = readBuf.size();
        //遍歷每一條客戶端連接
        for (int i = 0; i < size; i ++) {
            readPending = false;
            //傳遞事件, 將創(chuàng)建NioSokectChannel進(jìn)行傳遞
            //最終會(huì)調(diào)用ServerBootstrap的內(nèi)部類ServerBootstrapAcceptor的channelRead()方法
            pipeline.fireChannelRead(readBuf.get(i));
        }
        readBuf.clear();
        allocHandle.readComplete();
        pipeline.fireChannelReadComplete();
        //代碼省略
    } finally {
        //代碼省略
    }
}

在while循環(huán)結(jié)束之后, 將會(huì)通過一個(gè)for循環(huán)遍歷readBuf集合, 并將創(chuàng)建的NioSocketChannel傳入fireChannelRead()中, 傳播channel的讀取事件

有關(guān)pipeline的知識(shí), 我們下一章會(huì)詳細(xì)剖析, 并會(huì)根據(jù)剖析后的內(nèi)容回顧之前的有關(guān)pipeline的操作, 這里我們只需知道, 通過fireChannelRead()我們最終調(diào)用了ServerBootstrap的內(nèi)部類ServerBootstrapAcceptor 中的channelRead()方法

跟到channelRead()方法中:

public void channelRead(ChannelHandlerContext ctx, Object msg) {
    final Channel child = (Channel) msg;
    //代碼省略
    try {
        //work線程注冊(cè)channel
        childGroup.register(child).addListener(new ChannelFutureListener() {
            @Override
            public void operationComplete(ChannelFuture future) throws Exception {
                if (!future.isSuccess()) {
                    forceClose(child, future.cause());
                }
            }
        });
    } catch (Throwable t) {
        forceClose(child, t);
    }
}

其中參數(shù)的msg就是最初傳入fireChannelRead()方法的NioSocketChannel

所以這里可以通過 final Channel child = (Channel) msg 這種方式拿到NioSocketChannel

其中childGroup是我們最初初始化的work線程, 這里的register()方法跟boss線程一樣, 通過next()方法獲選擇一個(gè)線程進(jìn)行注冊(cè), 這里不再贅述

我們緊跟調(diào)用鏈, 跟到SingleThreadEventLoop的register()方法:

public ChannelFuture register(final ChannelPromise promise) {
    ObjectUtil.checkNotNull(promise, "promise");
    promise.channel().unsafe().register(this, promise);
    return promise;
}

這里的unsafe(), 根據(jù)我們之前的剖析, 是NioByteUnsafe, 這里的register最終會(huì)調(diào)用AbstractUnsafe的register()方法, 并NioSocketChannel

不知道同學(xué)們是否記得, 當(dāng)初NioServerSocketChannel注冊(cè)的時(shí)候也走的這個(gè)方法

我們跟到register()這個(gè)方法中:

public final void register(EventLoop eventLoop, final ChannelPromise promise) {
    //省略驗(yàn)證代碼
    //所有的復(fù)制操作, 都交給eventLoop處理
    AbstractChannel.this.eventLoop = eventLoop;

    if (eventLoop.inEventLoop()) {
        //做實(shí)際主注冊(cè)
        register0(promise);
    } else {
        try {
            eventLoop.execute(new Runnable() {
                @Override
                public void run() {
                    register0(promise);
                }
            });
        } catch (Throwable t) {
            //代碼省略
        }
    }
}

我們學(xué)習(xí)過NioEventLoop相關(guān)知識(shí)之后, 應(yīng)該對(duì)這部分代碼不太陌生, 首先判斷是不是當(dāng)前NioEventLoop線程, 如果是, 則直接進(jìn)行注冊(cè)操作, 如果不是, 則封裝成task在當(dāng)前NioEventLoop中執(zhí)行

走到這里不難明白, 這里并不是當(dāng)前NioEventLoop線程, 這是boss線程執(zhí)行的, 所以這里會(huì)走到else, 如果是第一次的連接操作, work線程的NioEventLoop并沒有啟動(dòng), 所以這里也會(huì)啟動(dòng)NioEventLoop, 并開始輪詢操作

跟到register0(promise)中看其是如何做實(shí)際操作的:

private void register0(ChannelPromise promise) {
    try {
        //省略代碼
        //做實(shí)際的注冊(cè)
        doRegister();
        neverRegistered = false;
        registered = true;
        //觸發(fā)事件
        pipeline.invokeHandlerAddedIfNeeded();
        safeSetSuccess(promise);
        //觸發(fā)注冊(cè)成功事件
        pipeline.fireChannelRegistered();
        if (isActive()) {
            if (firstRegistration) {
                //傳播active事件(4)
                pipeline.fireChannelActive();
            } else if (config().isAutoRead()) {
                beginRead();
            }
        }
    } catch (Throwable t) {
        //省略代碼
    }
}

這段代碼我們同樣并不陌生, 因?yàn)镹ioServerSokectChannel中也走這一部分, 我們繼續(xù)關(guān)注doRegister()方法:

protected void doRegister() throws Exception {
    boolean selected = false;
    for (;;) {
        try {
            //jdk底層的注冊(cè)方法
            //第一個(gè)參數(shù)為selector, 第二個(gè)參數(shù)表示不關(guān)心任何事件
            selectionKey = javaChannel().register(eventLoop().selector, 0, this);
            return;
        } catch (CancelledKeyException e) {
            //省略代碼
        }
    }
}

這部分也是我們之前剖析過的jdk底層的注冊(cè), 只是不同的是, 這里的javaChannel()是SocketChanel而不是ServerSocketChannel

同樣, 這里也是表示不關(guān)心任何事件, 只是在當(dāng)前NioEventLoop綁定的selector上注冊(cè)

感謝各位的閱讀！關(guān)于“Netty分布式NioSocketChannel注冊(cè)到selector方法的示例分析”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，讓大家可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到吧！