简介

Selector 是 java.nio.channels 包下的重要组件，阅读本文可以带你了解常用的 API。本文中把 Channel 翻译成信道，按照个人习惯也可以称作是通道、管道。

Selector 的核心组件有 SelectableChannel、Selector、SelectionKey。

多路复用选择器更多的是用在 Java 网络编程，网络编程的常用到就是 UDP / TCP 。SelectableChannel 的子类如下：

抽象类 SelectableChannel，它可以通过 Selector 多路复用。要使用多路复用功能，首先要注册。SelectableChannel 通过 register(Selector, int) 方法注册到 Selector 中，并返回 SelectionKey 作为注册代表。
抽象类 AbstractSelectableChannel ，它实现了 configureBlocking() 和 register() 方法，分别用于设置阻塞模式和注册事件
抽象类 DatagramChannel 提供 UDP 信道服务。
抽象类 ServerSocketChannel 提供 TCP 信道服务。主要用于服务器端
抽象类 SocketChannel 代表连接服务端与客户端的信道。通过 ServerSocketChannel.accept() 获得。当服务端接收到一个来自客户端的连接时就会产生一个信道。

各信道支持的事件

通过 AbstractSelectableChannel#validOps() 方法，可以返回一个操作集合（类型是 int）。这个集合表示该信道支持的操作。int 的每一个 bit 位代表信道是否支持某类操作。

同一个 SelectableChannel 的完全实现类的 validOps 方法总是返回固定的值。

目前 Selector 支持的事件类型是 4 种，分别是 SelectionKey.OP_ACCEPT、SelectionKey.OP_CONNECT、SelectionKey.OP_WRITE、SelectionKey.OP_READ，分别表示接受连接，主动连接，写，读

	OP_ACCEPT	OP_CONNECT	OP_WRITE	OP_READ
DatagramChannel	不支持	不支持	支持	支持
ServerSocketChannel	支持	不支持	不支持	不支持
SocketChannel	不支持	支持	支持	支持

注册事件及异常

AbstractSelectableChannel#register 方法可以注册指定事件到 Selector 中，并且还会保存注册成功后返回的 SelectionKey。

public final SelectionKey register(Selector sel, int ops, Object att)

{

      synchronized (regLock) {

            if (!isOpen())

                  throw new ClosedChannelException();

            if ((ops & ~validOps()) != 0)

                  throw new IllegalArgumentException();

            if (blocking)

                  throw new IllegalBlockingModeException();

            // 从数组 SelectionKey[] 寻找选择器 sel 对应的 SelectionKey

            SelectionKey k = findKey(sel);

            if (k != null) {

                  k.interestOps(ops);

                  k.attach(att);

            }

            if (k == null) {

                  // 新注册一个 SelectionKey

                  synchronized (keyLock) {

                        if (!isOpen())

                              throw new ClosedChannelException();

			// Selector 注册当前信道，并返回 SelectionKey

			k = ((AbstractSelector)sel).register(this, ops, att);

			// 将 SelectionKey 保存到数组中

			addKey(k);

                  }

            }

            return k;

      }

}

从这段源码中，我们还会留意到我们写代码常常会出现的几个异常：

如果我们向 SelectableChannel 实例对象注册它不支持的事件，抛出 IllegalArgumentException
我们注册事件前，没有调用 configureBlocking(false) 设置非阻塞，抛出 IllegalBlockingModeException

以上这两个是首次运用 Selector 时最常碰到的异常，另外还有一些其他的异常，比如
关闭 SelectorChannel 后，注册事件，抛出 ClosedChannelException

@Test(expected = ClosedChannelException.class)

public void ClosedChannelException() throws IOException {

    DatagramChannel channel = DatagramChannel.open();

    channel.configureBlocking(false);

    channel.close();

    channel.register(selector, SelectionKey.OP_WRITE);

}

关闭 Selector 后，注册事件，抛出 ClosedSelectorException

@Test(expected = ClosedSelectorException.class)

public void ClosedSelectorException() throws IOException {

    DatagramChannel channel = DatagramChannel.open();

    channel.configureBlocking(false);

    Selector selector = Selector.open();

    selector.close();

    channel.register(selector, SelectionKey.OP_READ);

}

取消 SelectionKey 后，注册事件，抛出 CancelledKeyException

@Test(expected = CancelledKeyException.class)

public void CancelledKeyException() throws IOException {

    DatagramChannel channel = DatagramChannel.open();

    Selector selector = Selector.open();

    channel.configureBlocking(false);

    SelectionKey key = channel.register(selector, SelectionKey.OP_READ);

    key.cancel();

    // selector.selectNow(); // 如果刷新一下就不会抛出异常

    channel.register(selector, SelectionKey.OP_WRITE);

}

通过 SelectableChannel#register 的注册方法，可以形成以下关系：

先看标识“五角星”的 AbstractSelectableChannel，包含一个可扩容的 SelectionKey 数组，通过数组元素对应多个不同的 Selector
再看标识“三角形”的 SelectionKey，通过这个对象，可以找到注册时对应的 selector 和 channel
然后看标识“五角星”的 Selector，包含一个 HashSet<SelectionKey>，通过集合保存与多个 SelectableChannel 的对应关系
总而言之，SelectableChannel 和 Selector 通过 SelectionKey 形成了多对多的关系

Selector 操作状态说明

下图是常见的 Selector 操作及其键集的变化示意图：

keys 键集，保存所有注册的 SelectionKey 的集合。
selectedKeys 就绪键集，保存所有准备就绪的键，通过 select 方法增加
cancelledKeys 取消键集，保存所有已取消的键，通过 select 方法，可以从 keys 中清除所有取消键，同时清空取消键集。

总结

Selector、SelectableChannel、SelectionKey 是 Java NIO 多路复用中的核心组件。

多路复用技术主要用于网络编程，SelectableChannel 其主要实现包含 DatagramChannel、ServerSocketChannel 和 SocketChannel。

Selecter 是 SelectableChannel 在设置非阻塞情况下使用的多路复用选择器。
SelectableChannel 支持注册事件的信道。
SelectionKey 是注册事件后，建立 Selector 与 SelectableChannel 之间关系的桥梁。

巴特西

NIO入门之多路复用选择器Selector

简介

各信道支持的事件

注册事件及异常

Selector 操作状态说明

总结

最新文章

热门文章