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（写就绪），可以通过上例中的代码注释和前两篇的内容来加深理解。