java的NIO和AIO

Buffer

• position、limit、capacity
• 初始化 Buffer
• 填充 Buffer
• 提取 Buffer 中的值
• mark() & reset()
• rewind() & clear() & compact()

Channel

• FileChannel
• SocketChannel
• ServerSocketChannel
• DatagramChannel

Selector

Buffer

一个 Buffer 本质上是内存中的一块，我们可以将数据写入这块内存，之后从这块内存获取数据。

java.nio 定义了以下几个 Buffer 的实现，其实核心是最后的 ByteBuffer，其他实现类只是包装了一下它而已，我们使用最多的通常也是 ByteBuffer。

可以将 Buffer 理解为一个数组，IntBuffer、CharBuffer、DoubleBuffer 等分别对应 int[]、char[]、double[] 等。

position、limit、capacity

Buffer 中也有几个重要属性：position、limit、capacity。

capacity，它代表这个缓冲区的容量，一旦设定就不可以更改。比如 capacity 为 1024 的 IntBuffer，代表其一次可以存放 1024 个 int 类型的值。一旦 Buffer 的容量达到 capacity，需要清空 Buffer，才能重新写入值。

position 和 limit 是变化的，我们分别看下读和写操作下，它们是如何变化的。

position 的初始值是 0，每往 Buffer 中写入一个值，position 就自动加 1，代表下一次的写入位置。读操作的时候也是类似的，每读一个值，position 就自动加 1。

从写操作模式到读操作模式切换的时候（flip），position 都会归零，这样就可以从头开始读写了。

limit：写操作模式下，limit 代表的是最大能写入的数据，这个时候 limit 等于 capacity。写结束后，切换到读模式，此时的 limit 等于 Buffer 中实际的数据大小，因为 Buffer 不一定被写满了。

切换模式

调用 Buffer 的 flip() 方法，可以进行模式切换。其实这个方法也就是设置了一下 position 和 limit 值。

1public final Buffer flip() {
2    limit = position; // 将 limit 设置为实际写入的数据数量
3    position = 0; // 重置 position 为 0
4    mark = -1; // mark 之后再说
5    return this;
6}

初始化 Buffer

每个 Buffer 实现类都提供了一个静态方法 allocate(int capacity) 帮助我们快速实例化一个 Buffer。另外经常使用 wrap 方法来初始化一个 Buffer。

1ByteBuffer byteBuf = ByteBuffer.allocate(1024);
2IntBuffer intBuf = IntBuffer.allocate(1024);
3LongBuffer longBuf = LongBuffer.allocate(1024);
4public static ByteBuffer wrap(byte[] array) {
5    ...
6}

填充 Buffer

put 方法：各个 Buffer 类都提供了一些 put 方法用于将数据填充到 Buffer 中，这些方法需要自己控制 Buffer 大小，不能超过 capacity，超过会抛 java.nio.BufferOverflowException 异常。

通过Channel填充：int num = channel.read(buf);将 Channel 的数据填充到 Buffer 中。在系统层面上，这个操作我们称为读操作，因为数据是从外部（文件或网络等）读到内存中。

提取 Buffer 中的值

get 方法：读操作提供了一系列的 get 方法。

通过Channel读取：int num = channel.write(buf);将buf的内容写入channel。

mark() & reset()

mark 用于临时保存 position 的值，每次调用 mark() 方法都会将 mark 设值为当前的 position，便于后续需要的时候使用。

考虑以下场景，我们在 position 为 5 的时候，先 mark() 一下，然后继续往下读，读到第 10 的时候，我想重新回到 position 为 5 的地方重新来一遍，那只要调一下 reset() 方法，position 就回到 5 了。

rewind() & clear() & compact()

rewind()：会重置 position 为 0，通常用于重新从头读写 Buffer。

clear()：有点重置 Buffer 的意思，相当于重新实例化了一样。不会将 Buffer 中的数据清空，只不过后续的写入会覆盖掉原来的数据，也就相当于清空了数据了。

compact()：和 clear() 一样的是，它们都是在准备往 Buffer 填充新的数据之前调用。先处理还没有读取的数据，也就是 position 到 limit 之间的数据（还没有读过的数据），先将这些数据移到左边，然后在这个基础上再开始写入。很明显写入模式下，此时 limit 还是等于 capacity，position 指向原来数据的右边。

1public final Buffer clear() {
2        position = 0;
3        limit = capacity;
4        mark = -1;
5        return this;
6    }
7
8    public ByteBuffer compact() {
9
10        int pos = position();
11        int lim = limit();
12        assert (pos <= lim);
13        int rem = (pos <= lim ? lim - pos : 0);  //position 到 limit 之间的数据
14
15        unsafe.copyMemory(ix(pos), ix(0), (long)rem << 0); //移到左边
16        position(rem);
17        limit(capacity());
18        discardMark();
19        return this;
20
21    }

Channel

所有的 NIO 操作始于通道，通道是数据来源或数据写入的目的地。

• FileChannel：文件通道，用于文件的读和写
• DatagramChannel：用于 UDP 连接的接收和发送
• SocketChannel：把它理解为 TCP 连接通道，简单理解就是 TCP 客户端
• ServerSocketChannel：TCP 对应的服务端，用于监听某个端口进来的请求

Channel 经常翻译为通道，类似 IO 中的流，用于读取和写入。它与前面介绍的 Buffer 打交道。

读操作的时候将 Channel 中的数据填充到 Buffer 中，而写操作时将 Buffer 中的数据写入到 Channel 中。

FileChannel

FileChannel 是不支持非阻塞的。

1// 读取
2FileInputStream inputStream = new FileInputStream(new File("/data.txt"));
3FileChannel fileChannel = inputStream.getChannel();
4
5ByteBuffer buffer = ByteBuffer.allocate(1024);
6int num = fileChannel.read(buffer);
7
8// 写入
9ByteBuffer buffer = ByteBuffer.allocate(1024);
10buffer.put("随机写入一些内容到 Buffer 中".getBytes());
11// Buffer 切换为读模式
12buffer.flip();
13while(buffer.hasRemaining()) {
14    // 将 Buffer 中的内容写入文件
15    fileChannel.write(buffer);
16}

SocketChannel

SocketChannel 是 TCP 客户端，它代表的是一个网络通道，可读可写。

1//打开一个 TCP 连接：
2// 打开一个通道
3SocketChannel socketChannel = SocketChannel.open();
4// 发起连接
5socketChannel.connect(new InetSocketAddress("https://www.baidu.com", 80));
6
7// 读取数据
8socketChannel.read(buffer);
9
10// 写入数据到网络连接中
11while(buffer.hasRemaining()) {
12    socketChannel.write(buffer);   
13}

ServerSocketChannel

ServerSocketChannel 就是对应的服务端。ServerSocketChannel 不和 Buffer 打交道了，因为它并不实际处理数据，它一旦接收到请求后，实例化 SocketChannel，之后在这个连接通道上的数据传递它就不管了，因为它需要继续监听端口，等待下一个连接。

ServerSocketChannel 用于监听机器端口，管理从这个端口进来的 TCP 连接。

1// 实例化
2ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
3// 监听 8080 端口
4serverSocketChannel.socket().bind(new InetSocketAddress(8080));
5
6while (true) {
7    // 一旦有一个 TCP 连接进来，就对应创建一个 SocketChannel 进行处理
8    SocketChannel socketChannel = serverSocketChannel.accept();
9}

DatagramChannel

UDP 和 TCP 不一样，DatagramChannel 一个类处理了服务端和客户端。

UDP 是面向无连接的，不需要和对方握手，不需要通知对方，就可以直接将数据包投出去，至于能不能送达，它是不知道的。

1//监听端口
2DatagramChannel channel = DatagramChannel.open();
3channel.socket().bind(new InetSocketAddress(9090));
4
5ByteBuffer buf = ByteBuffer.allocate(48);
6buf.clear();
7
8channel.receive(buf);
9
10//发送数据
11String newData = "New String to write to file..."
12                    + System.currentTimeMillis();
13
14ByteBuffer buf = ByteBuffer.allocate(48);
15buf.clear();
16buf.put(newData.getBytes());
17buf.flip();
18
19int bytesSent = channel.send(buf, new InetSocketAddress("localhost.com", 80));

Selector

Selector 建立在非阻塞的基础之上，多路复用指的就是它，用于实现一个线程管理多个 Channel。

1//开启一个 Selector  多路复用器
2Selector selector = Selector.open();
3
4//将 Channel 注册到 Selector 上, 注册到 Selector 的 Channel 必须要支持非阻塞模式.FileChannel 不支持非阻塞，最常见的 SocketChannel 和 ServerSocketChannel。
5// 将通道设置为非阻塞模式，因为默认都是阻塞模式的
6channel.configureBlocking(false);
7// 注册
8SelectionKey key = channel.register(selector, SelectionKey.OP_READ);
9//注册方法返回值是 SelectionKey 实例，它包含了 Channel 和 Selector 信息，也包括了一个叫做 Interest Set 的信息，即我们设置的我们感兴趣的正在监听的事件集合。
10//监听感兴趣的事件,可以同时监听一个 Channel 中的发生的多个事件,共以下四种事件：
11//        SelectionKey.OP_READ
12//        SelectionKey.OP_WRITE
13//        SelectionKey.OP_CONNECT
14//        SelectionKey.OP_ACCEPT
15
16//调用 select() 方法获取通道信息。用于判断是否有我们感兴趣的事件已经发生了。
17while(true) {
18  // 判断是否有事件准备好
19  int readyChannels = selector.select();
20  if(readyChannels == 0) continue;
21
22  // 遍历
23  Set<SelectionKey> selectedKeys = selector.selectedKeys();
24  Iterator<SelectionKey> keyIterator = selectedKeys.iterator();
25  while(keyIterator.hasNext()) {
26    SelectionKey key = keyIterator.next();
27
28    if(key.isAcceptable()) {
29        // a connection was accepted by a ServerSocketChannel.
30
31    } else if (key.isConnectable()) {
32        // a connection was established with a remote server.
33
34    } else if (key.isReadable()) {
35        // a channel is ready for reading
36
37    } else if (key.isWritable()) {
38        // a channel is ready for writing
39    }
40
41    keyIterator.remove();
42  }
43}