Netty原理详解系列(二)---NIO中的Selector

1.概述

在上一篇博客《Netty原理详解系列(一)—NIO中的Buffer&Chanel》介绍了NIO中的两个重要组件Buffer和Chanel，这篇博客将介绍第三个重要的组件Selector多路复用选择器。

如果没有多路复用选择器，那么每个管道都需要一个线程来维护，一直阻塞着等待着消息。而有了多路复用选择器，则需要将多个管道都注册到这个选择器中。通过选择器来帮忙查看管道的状态。当有消息的时候到达后就会得到通知，分配线程去执行。避免了前一种方式多个线程的阻塞。

2.选择器核心组件

选择器核心组件有三个:管道(SelectableChannel)、选择器(Selector)、选择键(SelectorKey)。

并不是所有的管道都可以注册到选择器中的。只有SelectableChannel的子类才可以。当管道注册到Selector中会返回一个key，通过它就可以获得到关联的管道。

2.1 SelectableChannel

可注册管道，默认设置的是阻塞的模式(true)。如果想要将它注册到选择器中，需要先将阻塞模式设置为false。调用configureBlocking(false)，设置成非阻塞模式。

注册到选择器中，调用register方法，第一个参数是选择要注册到哪个选择器中，第二个参数是监听的事件。

可选择的监听的事件有4个：

• CONNECT 建立连接
• ACCEPT 接受连接
• READ 可读
• WRITE 可写

但并不是所有的SelectableChannel 子类管道都支持这4个事件。可以通过validOps()，来查看当前管道可支持哪些事件。

2.2 Selector

管道注册到Selector中，会生成一个键(SelectorKey) 该键维护在Selector的keys中。并调用select方法进行刷新键。如果返回的数字大于0，说明有指定数量的键状态发生了变化。

• select(): 有键更新，立马返回。否则会一直阻塞直到有键更新为止。

• select(long )：有键更新，立马返回。否则会阻塞参数指定毫秒时。

• selectNow()：无论有没有键更新都会立马返回

如果有键更新，接下来就可以调用selectedKeys 获取更新的键集了，也就是得到就绪的键集。

选择器中一共维护了3个键集，底层都是Set实现所以不会重复。

• 全部键集(keys)：所有向该选择器注册的键都放置于此

• 选择键集(selectedKeys)：存放准备就续的键，调用select会刷新键集，会将已就绪的键添加到这个集合中，同时清空取消键集，并移除已取消的键。

• 取消键集(cancelledKeys) ：存放已取消的键，在关闭管道或者关闭选择器都会将键添加到取消键集。但并不会立马清除，而是等到下一次刷新键集的时候才会清除

2.3 SelectionKey

键用于关联管道与选择器，并监听维护管道1至多个事件，监听事件可在注册时指定，也可后续通过调用SelectionKey.interestOps 来改变。

3.代码演示

public void test3() throws IOException {
  
    Selector selector = Selector.open();// 打开选择器
    DatagramChannel channel = DatagramChannel.open();// 打开管道
    channel.bind(new InetSocketAddress(8080));// 绑定端口号
    channel.configureBlocking(false);// 设置为非阻塞
    channel.register(selector, SelectionKey.OP_READ);//将管道注册到选择器中


    DatagramChannel channel2 = DatagramChannel.open();
    channel2.bind(new InetSocketAddress(8081));
    channel2.configureBlocking(false);
    channel2.register(selector, SelectionKey.OP_READ);

    for (; ; ) {
        int count = selector.select();//遍历管道状态 刷新键集
        if (count > 0) {
            Set<SelectionKey> selectionKeys = selector.selectedKeys();//获取就绪键集
            Iterator<SelectionKey> iterator = selectionKeys.iterator();
            while (iterator.hasNext()) {//遍历就续集
                SelectionKey key = iterator.next();
                handle(key);//处理该就续键
                iterator.remove();//从就续集中移除
            }
        }
    }
}

public void handle(SelectionKey key) throws IOException {
    ByteBuffer buffer = ByteBuffer.allocate(8192);
    DatagramChannel channel = (DatagramChannel) key.channel();
    buffer.clear(); //  清空还原
    channel.receive(buffer); // 阻塞
    buffer.flip();
    byte[] bytes = new byte[buffer.remaining()];
    buffer.get(bytes);
    System.out.println(new String(bytes));
}

上面代码就体现了多路复用选择器的用处。步骤如下：

1. 构建选择器
2. 构建管道，并绑定端口号
3. 修改管道为非阻塞模式
4. 将管道注册到选择器中，同时设置监听事件。上面的代码注册了两个管道，分别是8080端口的和8081端口的两个管道。
5. select() 刷新键集，返回就绪键的个数。
6. 若大于0，说明有管道的状态变更了。获取就绪键集
7. 遍历就绪键集，处理就绪键。在hanlde方法里面，根据就绪键可以得到对应的管道。然后就可以做一些业务处理了。此时的业务处理可以开线程异步的去执行。

代码运行，打开两个终端窗口，输入nc -uv ip port 命令

连接成功

发送udp消息 8080 发送消息“我是8080”,8081 发送消息“我是8081”

idea控制台输出的结果如下：

4. 后续

学习完了NIO中的三个重要的组件，下一篇博客来实战演练一下，写几个小demo。