比喻：餐厅

image.png
单线程NIO和单线程BIO一样，它只有一个员工，这个员工有分流(Select)的能力

  1. 他负责接待新来的客人，安排客人坐下，给客人菜单（安顿客人 事件）
  2. 客人要点单了，他过去帮客人下单（帮客人下单 事件）

BIO处理的单元是用户；而NIO处理的单元是事件

  • BIO，有客人来，我去跟踪这个客人为它服务（普通函数的思想）
  • NIO，安顿客人事件发生，我触发这个事件，并处理这个事件（Reactor模式）

代码

服务端代码

  1. package NIO;
  3. //NIO的服务器（餐厅）
  4. public class NIOServer{
  5.     //通道管理器（服务员）
  6.     private Selector selector;
  8.     //（餐厅开门）
  9.     //获得一个ServerSocket通道，并对该通道做一些初始化的工作
  10.     public void initServer(int port) throws IOException{
  11.         //获得一个ServerSocket通道（开一扇门）
  12.         ServerSocketChannel serverChannel = ServerSocketChannel.open();
  13.         serverChannel.configureBlocking(false);    //设置通道为非阻塞
  14.         serverChannel.socket().bind(new InetSocketAddress(port)); //将该通道对应的ServerSocket绑定到port端口
  16.         //获得一个通道管理器（服务员）
  17.         this.selector = Selector.open();
  19.         //（将此服务员安排在这个餐厅；并将餐厅的门打开）
  20.         //将通道管理器和该通道绑定，并为该通道注册SelectionKey.OP_ACCEPT事件
  21.         //当该事件到达时，selector.select()会返回，如果该事件没到达selector.select()会一直阻塞
  22.         serverChannel.register(selector, SelectionKey.OP_ACCEPT);
  23.     }
  25.     //（服务员在这里领任务：看是否有客人来；是否有人点餐；……）
  26.     //采用轮询的方式监听selector上是否有需要处理的事件，如果有，则进行处理
  27.     public void listen() throws IOException{
  28.         System.out.println("服务器启动成功！");
  30.         //轮询访问selector
  31.         while(true) {
  32.             //（服务员看看有没有事情要处理
  33.             //    1. 如果没事情，就会堵塞在这一句，等待事情来；
  34.             //    2. 如果有事情，就继续下面的代码
  35.             // ）
  36.             //当注册的事件到达时，方法返回；否则，该方法会一直阻塞
  37.             selector.select();
  39.             //（取出所有的事情，一件一件做）
  40.             //获得selector中选中的项的迭代器，选中的项为注册的事件
  41.             Iterator<?> ite = this.selector.selectedKeys().iterator();
  42.             while(ite.hasNext()){
  43.                 //（服务员要处理的事情名称）
  44.                 SelectionKey key = (SelectionKey)ite.next();
  46.                 //（删除这个事情，表示已经做了）
  47.                 //删除已选的key，以防重复处理
  48.                 ite.remove();
  50.                 //（去处理这个事情）
  51.                 //处理这个key
  52.                 handler(key);
  53.             }
  54.         }
  55.     }
  57.     //（服务员干活）
  58.     //处理请求
  59.     public void handler(SelectionKey key) throws IOException{
  60.         if(key.isAcceptable()) {
  61.             //（有客人来）
  62.             //客户端请求连接事件
  63.             handlerAccept(key);
  64.         } else if(key.isReadable()){
  65.             //（客人要点餐）
  66.             //获得可读的事件
  67.             handelerRead(key);
  68.         }
  69.     }
  71.     //（服务员如何安排客人坐下）
  72.     //处理连接请求
  73.     public void handlerAccept(SelectionKey key) throws IOException {
  74.         ServerSocketChannel server = (ServerSocketChannel) key.channel();
  75.         SocketChannel channel = server.accept();    //获得和客户端连接的通道
  76.         channel.configureBlocking(false);            //设置成非阻塞
  78.         System.out.println("新的客户端连接");
  80.         //在这里可以给客户端发送信息
  81.         //...
  83.         //（客户坐下以后，把菜单给他，给他点菜的权限）
  84.         //在和客户端连接成功之后，为了可以接收到客户端的信息，需要给通道设置读的权限
  85.         channel.register(this.selector, SelectionKey.OP_READ);
  86.     }
  88.     //（服务员如何取得客人的菜单）
  89.     //处理读的事件
  90.     public void handelerRead(SelectionKey key) throws IOException {
  91.         //服务器可读取消息：得到事件发生的Socket通道
  92.         SocketChannel channel = (SocketChannel) key.channel();
  94.         //创建读取缓冲区
  95.         ByteBuffer buffer = ByteBuffer.allocate(1024);
  96.         channel.read(buffer);
  97.         byte[] data = buffer.array();
  98.         String msg = new String(data).trim();
  99.         System.out.println("服务端收到的信息：" + msg);
  101.         ByteBuffer outBuffer = ByteBuffer.wrap(msg.getBytes());
  102.         channel.write(outBuffer); //将消息送回给客户端
  103.     }
  105.     //（正式开门）
  106.     //启动服务端测试
  107.     public static void main(String[] args) throws IOException{
  108.         NIOServer server = new NIOServer();
  109.         server.initServer(8000);
  110.         server.listen();
  111.     }
  112. }

客户端：可以使用windows的telnet 

  1. # 1. 打开Windows命令行
  2. # 2. 使用telnet连接服务端
  3. > telnet 127.0.0.1 8000 
  5. # 3. 给服务端发送消息
  6. > send wo yao chifan

参考链接

  1. NIO代码分析（讲的很不错，推荐）