Java NIO介绍及入门案例

Java NIO介绍及入门案例

1. Java I/O通信模型

1.1 I/O模型介绍

1.1.1 介绍

I/O模型就是用什么样的通道进行数据的发送和接收，很大程度上决定了程序通信的性能

1.1.2 分类

Java共支持3中网络编程模型：BIO、NIO、AIO

• BIO：同步阻塞，服务器实现模式为一个连接一个线程，即客户端有连接请求时，服务端就需要启动一个线程进行处理，如果这个连接不做任何事情会造成不必要的线程开销
  

• NIO：同步非阻塞，服务器实现模式为一个线程处理多个请求（连接），即客户端发送的请求都会注册到多路复用器上，多路复用器轮询到连接有I/O请求就进行处理

• AIO（NIO.2）：异步非阻塞，AIO引入异步通道的概念，采用了Proactor模式，简化了程序编写，有效的请求才启动线程，它的特点是先由操作系统完成后才通知服务端程序启动线程去处理

1.1.3 应用场景

• BIO适用于连接数目比较小且固定的架构，这种方式对服务器资源要求比较高，并发局限于应用中，编程比较简单JDK1.4以前唯一的选择
• NIO适用于连接数目多且连接比较短的架构，比如聊天服务器，弹幕系统，服务器间通讯等，编程比较复杂，JDK1.4开始支持
• AIO适用于连接数目比较多且连接比较长的架构，比如相册服务器，充分调用OS参与并发操作，编程比较复杂，JDK7开始支持

2. Java NIO基本介绍

2.1 介绍

Java NIO是JDK提供的API，相关类都被放在java.nio包及其子包下，并对原java.io包中的很多类进行改写；NIO有三大核心：Selector、Channel、Buffer；NIO是面向缓冲区或者面向块编程的，数据读取到一个它稍后处理的缓冲区，需要时可在缓冲区中前后移动，这就增加了处理过程中的灵活性，使用他可以提供非阻塞式的高伸缩性网络；Java NIO的非阻塞模式，使一个线程从某通道发送请求或读取数据，但他仅能得到目前可用的数据，如果目前没有数据可用时，就什么都不获取，而不是保持线程阻塞，所以直至数据变得可读之前，该线程可以继续做其他事情，非阻塞写也是如此，一个线程请求写入一些数据到某通道，但不需要等待它完全写入，这个线程同时可以去做别的事情

2.2 NIO三大核心

• Buffer（缓冲区）： 本质上是一个可以读写数据的内存块，可以理解成一个容器对象，该对象提供了一些方法，可以更轻松地使用内存块。Channel提供从文件、网络读取数据的渠道，但是读取或写入的数据必须经由Buffer。
• Channel（通道）： NIO的通道类似于流，不同于流的是channel可以同时进行读写；可以异步读写；可以从缓冲区读数据，也可以写数据到缓冲区
• Selector（选择器）： 用一个线程处理多个客户端连接就会使用Selector，它能够检测多个注册的通道上是否有事件发生（多个Channel以事件的方式可以注册到同一个Selector），如果有事件发生，便获取事件然后针对每个事件进行相应处理。这样就可以只用一个单线程去管理多个通道（即多个连接和请求）。只有在连接/通道真正有读写事件发生时，才会进行读写，这就大大地减少了系统开销

每个channel都会对应一个buffer，selector对应一个线程，一个线程对应多个channel（连接），该图有3个channel注册到selector，程序切换到哪个channel是由事件决定的，Event就是一个重要的概念，selector会根据不同的事件，在各个通道上切换，buffer就是一个内存块，底层是有一个数组，数据的读写通过buffer，nio的buffer是可以读也可以写的，需要使用flip方法切换

2.3 执行流程

1. 当客户端连接时，会通过ServerSocketChannel得到SocketChannel

2. Selector进行监听select方法，返回有事件发生的通道个数

3. 将socketChannel注册到Selector上，一个selector上可以注册多个socketChannel

4. 注册后返回一个SelectionKey，会和该Selector关联

5. 进一步得到各个SelectionKey（有事件发生）

6. 在通过SelectionKey反向获取socketChannel

7. 可以通过得到的channel完成业务处理

3. 入门案例

public class NIOServer {
    public static void main(String[] args) throws IOException {
        //1.创建ServerSocketChannel
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        //2.创建Selector
        Selector selector = Selector.open();
        //3.服务端监听端口
        serverSocketChannel.socket().bind(new InetSocketAddress(6666));
        //4.设置为非阻塞
        serverSocketChannel.configureBloking(false);
        //5.把serverSocketChannel注册到selector，关心事件为OP_ACCEPT
        serverSocketChannel.register(selector, SelectionKey.OP_KEEP);
        //循环等待客户端连接
        while (true) {
            //等待1s，如果没事事件发生就返回
            if (selector.select(1000) == 0){
                continue;
            }
            //返回关注事件的集合
            Set<SelectionKey> selectionKeys = selector.selectedKeys();
            Iterator<SelectionKey> keyIterator = selectionKeys.iterator();
            while (keyIterator.hasNext()) {
                //获取SelectionKey
                SelectionKey key = keyIterator.next();
                //根据key对应的通道发生的事件做相应的处理
                if (key.isAcceptable()){//有新的客户端连接
                    SocketChannel socketChannel = serverSocketChannel.accept();
                    System.out.println("客户端连接成功，生成了一个socketChannel：" + socketChannel.hashCode());
                    //将socketChannel设置为非阻塞
                    socketChannel.configureBlocking(false);
                    //将socketChannel注册到selector，关注事件为OP_READ，同时给socketChannel关联一个buffer
                    socketChannel.register(selector, SelectionKey.OP_READ, ByteBuffer.allocate(1024)); 
                }
                if (key.isReadable()) {//发生OP_READ
                    //通过key反向获取通道
                    SocketChannel channel = (SocketChannel) key.channel();
                    //获取到该channel关联的buffer
                    ByteBuffer buffer = (ByteBuffer) key.attachment();
                    while (true) {
                        channel.read(buffer);
                        System.out.println("从客户端读取到：" + new String(buffer.array()));
                    }
                }
                //手动从集合中移除当前的selectionKey，防止重复操作
                keyIterator.remove();
            }
        }
    }
}

public class NIOClient {
    public static void main(String[] args) throws IOException {
        //1.获取通道
        SocketChannel socketChannel = SocketChannel.open();
        //2.设置为非阻塞
        socketChannel.configureBlocking(false);
        //3.设置服务端地址
        InetSocketAddress inetSocketAddress = new InetSocketAddress("127.0.0.1", 6666);
        //4.连接服务器
        if (!socketChannel.connect(inetSocketAddress)) {
            while (!socketChannel.finishConnect()) {
                System.out.println("因为连接需要事件，客户端不会阻塞，可以做其他工作");
            }
        }
        //5.连接成功就发送数据
        String msg = "hello,服务端";
        ByteBuffer buffer = ByteBuffer.wrap(msg.getBytes());
        //将buffer写入通道
		socketChannel.write(buffer);
        System.out.println("发送数据成功");
    }
}

本篇博客参考了韩顺平老师的笔记