选择器是用来处理多个通道并监听其通道事件的组件。
-
Create
只需要调用 open() 即可创建一个Selector对象:Selector selector = Selector.open();
-
Register
通过 register() 方法注册通道:ServerSocketChannel channel = ServerSocketChannel.open(); channel.configureBlocking(false); SelectionKey selectionKey = channel.register(selector, SelectionKey.OP_ACCEPT);
在注册通道之前,把通道设置成非阻塞模式,观察源码会发现 register() 会校验当前通道是否为非阻塞模式,当是阻塞模式时,会抛出IllegalBlockingModeException 异常。在前面一个章节也提过,为什么FileChannel没有继承SelectableChannel,因为它不需要多路复用,所以在使用通道的时候,只有FileChannel不能向选择器注册通道,凡是继承SelectableChannel都能够向选择器注册通道。
注册通道方法的第二个参数是SelectionKey中定义的操作类型,你可以填入任何你感兴趣的操作类型,只要这个通道支持,同样,在执行 register() 方法时也会校验该通道是否能够支持该操作。
注册方法同样也会返回一个SelectionKey对象。 -
Attach Object
注册通道的 register() 方法有一个重载方法,可以向选择器注册通道的时候,选择想要带上的附加对象:public abstract SelectionKey register(Selector sel, int ops, Object att) throws ClosedChannelException;
例如,使用时附加上一个字符串:
String ch_name = "123"; SelectionKey selectionKey = channel.register(selector, SelectionKey.OP_ACCEPT,ch_name);
获取这个字符串可通过 attachment() 来获取:
// 接收数据 String ch_name_accept = (String) selectionKey.attachment();
当然,注册时返回的SelectionKey对象也可以在使用时候附加你想要的附加对象:
selectionKey.attach(ch_name);
-
Block
因为是一个线程通过选择器来操作通道,那么选择器在操作通道时,必定在处理一个通道的时候,另一个事件已就绪的通道处于等待状态。在确定一个通道事件就绪之后,才能去操作这个通道。上文中讲到使用注册方法register使用的代码示例,将ServerSocketChannel对象向选择器注册,同时关注了这个通道的OP_ACCEPT操作类型事件,那么我们什么时候能确定该通道的accept事件就绪,可以操作这个通道了。选择器为我们提供了三个重载的 select() 方法,这三个方法的主要功能就是选择是否阻塞直到该选择器中的通道所关注的事件就绪,来让程序继续往下运行。
首先看 select() 方法,该方法会一直阻塞下去,直到选择器中的通道关注的事件就绪:selector.select();
参数5000是5秒,参数以毫秒为单位。这个方法会一直阻塞5秒,5秒之内如果没有通道事件就绪的话程序会往下运行:
selector.select(5000);
selectNow()其实就是非阻塞,无论有无通道事件就绪,程序都会向下执行:
selector.selectNow();
这三个方法的本质区别无非是选择器阻塞或者等待一个或多个通道的事件就绪有多长时间。
-
Keys & SelectionKeys
我们每为一个通道执行 register() 注册方法,就会返回一个SelectionKey,那么这个选择器所有已就绪的SelectionKey就是通过selectedKeys()来获取:Set<SelectionKey> selectionKeys = selector.selectedKeys();
一般这个方法是在select() 之后执行,因为到这一步就意味着要通过这个轮询每个就绪的通道。
Iterator<SelectionKey> iterator = selectionKeys.iterator(); while (iterator.hasNext()) { SelectionKey key = iterator.next(); if (key.isAcceptable()) { // 执行通道的操作 } //执行完成移除 iterator.remove(); }
到这里说的是已就绪的通道,那么所有的 SelectionKey 集可以通过 keys() 方法获取:
Set<SelectionKey> keys = selector.keys();
- wake up
在使用Selector对象的 select() 或者 select(long) 方法时候,当前线程很可能一直阻塞下去,那么用另一个线程去执行 Selector.wakeUp() 方法会唤醒当前被阻塞的线程,使其 select() 立即返回。
当然,如果当前线程没有阻塞,那么执行了wakeUp() 方法之后,下一个线程的 select() 方法会被立即返回,不再被阻塞下去。 -
close
显然,close() 方法能够关闭当前的选择器。
当一个线程当前呈阻塞状态,那么中止这种状态需要执行选择器的 wakeUp() 方法,close()方法的实现正是这么做的,先唤醒被阻塞的线程,然后继续接下来的操作。接下来就会会置空所有的通道、所有就绪的SelectionKey,让这个选择器上的轮询组件也闲置下来。
SelectionKey
SelectionKey的功能类似于通道的一个注册令牌。
这个类定义了4个操作类型,每种操作类型都对应了相应的事件,通过监听这几种不同的事件,在触发该事件时表示所对应的操作已准备就绪:
操作类型 | 值 | 描述 | |
---|---|---|---|
OP_READ | 1 << 0 | 读操作 | |
OP_WRITE | 1 << 2 | 写操作 | |
OP_CONNECT | 1 << 3 | 连接socket操作 | |
OP_ACCEPT | 1 << 4 | 接受socket操作 |
这里得提一句,所有继承SelectableChannel的通道都会定义自己能够支持的操作类型,可以通过具体通道的 validOps() 方法查看,例如SocketChannel支持read、write、connect这几种操作:
// SocketChannel类
public final int validOps() { return (SelectionKey.OP_READ | SelectionKey.OP_WRITE | SelectionKey.OP_CONNECT); }
-
interestOps & readyOps
这是SelectionKey的实现类中定义的变量:private volatile int interestOps; private int readyOps;
interestOps用来存储感兴趣的操作集,readyOps用来存储已经就绪的操作集。
其中 interestOps() 方法和 nioInterestOps() 都会返回interestOps,不同的是interestOps()会校验是否已执行 cancel() ,如果已经取消则会抛出 CancelledKeyException 异常。readyOps同样也有 readyOps() 和 nioReadyOps() 方法,逻辑与interestOps几乎一致。
观察这段SelectionKey抽象类已经实现的代码:// SelectionKey 类 public final boolean isAcceptable() { return (readyOps() & OP_ACCEPT) != 0; }
当判断是否访问就绪的时候,只要 readyOps() 与相应的操作类型相与,非零就返回true,代表接受请求操作已就绪。这个是SelectionKey已提供的方法,但是SelectionKey并未提供同样返回boolean判断某个操作在interestOps集是否存在,我们可以自己实现这些方法:
private boolean isInterestRead(SelectionKey selectionKey){ return (selectionKey.interestOps() & SelectionKey.OP_ACCEPT) == SelectionKey.OP_READ; } private boolean isInterestWrite(SelectionKey selectionKey){ return (selectionKey.interestOps() & SelectionKey.OP_WRITE) == SelectionKey.OP_WRITE; } private boolean isInterestConnect(SelectionKey selectionKey){ return (selectionKey.interestOps() & SelectionKey.OP_CONNECT) == SelectionKey.OP_CONNECT; } private boolean isInterestAccept(SelectionKey selectionKey){ return (selectionKey.interestOps() & SelectionKey.OP_ACCEPT) == SelectionKey.OP_ACCEPT; }
-
Channel、Selector
在SelectionKey中获取通道或者选择器只需要调用其中的两个方法即可:SelectableChannel selectableChannel = selectionKey.channel(); Selector sel = selectionKey.selector();