Java NIO学习总结三（Selector）

上一篇讲了NIO的缓冲区，在给出的例子中当有多个SocketChannel连接到服务器时，我们是通过遍历各个Channel来管理它们的输入输出，其实也可以交给NIO的选择器来管理，下面给出使用Selector的同等效果的实现：

public class SelectorTest {
public static void main(String[] args) {
try {
Selector selector = Selector.open(); // 开启一个选择器

ServerSocketChannel ssc = ServerSocketChannel.open(); // 开启一个服务器套接字渠道
ssc.bind(new InetSocketAddress(8189)); // 将服务器套接字渠道绑定到本地的8189端口
ssc.configureBlocking(false); // 设定为非阻塞
ssc.register(selector, SelectionKey.OP_ACCEPT); // 注册到选择器，并对接收就绪事件感兴趣

Map sbMap = new HashMap<>(); // 每个选择器key对应一个字节缓冲区

while (true) { // 轮询监听选择器
if (selector.select() > 0) { // 大于0表示有感兴趣的事件发生
Set keys = selector.selectedKeys(); // 获得所有发生感兴趣事件的keys

for (SelectionKey skTemp : keys) { // 遍历这些keys
if (skTemp.isAcceptable()) { // 如果是接收就绪（肯定是ServerSocketChannel）
SocketChannel sc = ((ServerSocketChannel) skTemp.channel()).accept(); // 获得就绪的套接字渠道
sc.configureBlocking(false); // 设定为非阻塞
SelectionKey sk = sc.register(selector, SelectionKey.OP_READ); // 将新的套接字渠道也注册到选择器，
// 并对读就绪事件感兴趣

sbMap.put(sk, ByteBuffer.allocate(10)); // 为新的套接字渠道分配一个新的缓冲区
System.out.println("new socket");

keys.remove(skTemp); // 刚兴趣的keys处理完后就要移除
} else if (skTemp.isReadable()) { // 如果是读就绪（肯定是SocketChannel）
SocketChannel sc = (SocketChannel) skTemp.channel();
ByteBuffer bb = sbMap.get(skTemp); // 获得套接字渠道对应的缓冲区

if (sc.read(bb) == -1) { // 连接断开时移除其缓冲区，以后代码解释看上一篇
sbMap.remove(skTemp);
continue;
}

if (bb.position() == bb.capacity()) {
bb.flip();

while (bb.hasRemaining()) {
System.out.print((char) bb.get());
}

bb.clear();
}

keys.remove(skTemp);
}
}
}
}
} catch (IOException e) {
e.printStackTrace();
}
}
}

上例中可以看出使用Selector的一般过程，Channel通过register方法注册到Selector，并注明它感兴趣的事件，之后通过Selector的select方法判断有没有感兴趣时间发生，如果有，通过selectedKeys方法获得兴趣事件的集合。

SelectionKey可以理解为Channel和Selector之间交互信息的桥梁，其中有几种默认的事件，包括OP_CONNECT（连接就绪）、OP_ACCEPT（接受就绪）、OP_READ（读就绪）和OP_WRITE（写就绪），可以通过上例中的代码注释和前两篇的内容来加深理解。