前置掌握点
- 事件发生后,要么处理,要么取消(cancel),不能什么都不做,否则下次该事件仍会触发,这是因为 nio 底层使用的是水平触发
- select 在事件发生后,就会将相关的 key 放入 selectedKeys 集合,但不会在处理完后从 selectedKeys 集合中移除,需要我们自己编码删除。
- cancel 会取消注册在 selector 上的 channel,并从 keys 集合中删除 key 后续不会再监听事件
代码例子(含详细步骤解析注释)
public static void main(String[] args) {
// 开启最大的通信通道
try (ServerSocketChannel channel = ServerSocketChannel.open()) {
// 最大通信通道所监听的端口
channel.bind(new InetSocketAddress(8080));
// 给我来个管道加强器selector
Selector selector = Selector.open();
// 确定管道是顺畅的
channel.configureBlocking(false);
// 指定该管道加强器selector所监听的事件为通信连接(对于最大的管道,没有通信哪来的读写)
channel.register(selector, SelectionKey.OP_ACCEPT);
while (true) {
// 阻塞监听事件,有事件的话才会放行到下面进行事件的处理
selector.select();
// 获取所有事件,因为现在我们所接收的是一堆人,所以说接收的事件很多
Set<SelectionKey> keys = selector.selectedKeys();
// 遍历所有事件,逐一处理(之所以用迭代器是为了能够在事件处理的过程中删除已经执行过的事件对象)
Iterator<SelectionKey> iter = keys.iterator();
while (iter.hasNext()) {
SelectionKey key = iter.next();
// 判断事件类型
if (key.isAcceptable()) {
// 如果是连接事务,那么我们要开启的就是连接的大通道ServerSocketChannel
ServerSocketChannel c = (ServerSocketChannel) key.channel();
// 必须处理,连接后再建立小通道SocketChannel来进行读写操作
SocketChannel sc = c.accept();
sc.configureBlocking(false);
sc.register(selector, SelectionKey.OP_READ);
} else if (key.isReadable()) {
// 如果是读/写事务,那么我们要开启的就是连接的小通道SocketChannel来进行读写操作
SocketChannel sc = (SocketChannel) key.channel();
ByteBuffer buffer = ByteBuffer.allocate(128);
// 必须处理,正常读取后才算处理完该都事件(读到的结果为-1不算正常读取)
int read = sc.read(buffer);
// 如果是正常断开,客户端一样会执行一次读操作,read 的方法的返回值是 -1,由于这种正常断开并不会抛出异常,所以无法被catch住,所以要我们手动去判断消费掉该读事件,不然该没有被消费的读事件就会使得select()失去阻塞效果,不停进行读操作
if(read == -1) {
key.cancel();
sc.close();
} else {
buffer.flip();
debug(buffer);
}
}
// 处理key 时,要从 selectedKeys 集合中删除,否则会报空指针异常,具体原因在于selector 会在发生事件后,向绿色集合中加入key,执行完后事件标识就消失了,但事件的id不会删除,这就导致了当我们再次通过SelectionKey key = iter.next();能获取到该已经被处理过的事件对象,而这类事件对象在SocketChannel sc = c.accept();中是不会奏效的,但是由于该通道为非阻塞的通道,所以会返回一个null来避免在此位置阻塞,这就导致了空指针异常
iter.remove();
}
}
} catch (IOException e) {
e.printStackTrace();
key.cancel();// 如果因为客户端断开了导致的事件未被处理,就会一直死循环报错,因为只要selector的key集合中有未被处理的key,select()就无法实现阻塞监听的状态,有未被(从 selector 的 keys 集合中真正删除 key)
}
}
}