快速了解Java中NIO核心组件

背景知识

同步、异步、阻塞、非阻塞

首先，这几个概念非常容易搞混淆，但NIO中又有涉及，所以总结一下。

同步：API调用返回时调用者就知道操作的结果如何了（实际读取/写入了多少字节）。

异步：相对于同步，API调用返回时调用者不知道操作的结果，后面才会回调通知结果。

阻塞：当无数据可读，或者不能写入所有数据时，挂起当前线程等待。

非阻塞：读取时，可以读多少数据就读多少然后返回，写入时，可以写入多少数据就写入多少然后返回。

对于I/O操作，根据Oracle官网的文档，同步异步的划分标准是“调用者是否需要等待I/O操作完成”，这个“等待I/O操作完成”的意思不是指一定要读取到数据或者说写入所有数据，而是指真正进行I/O操作时，比如数据在TCP/IP协议栈缓冲区和JVM缓冲区之间传输的这段时间，调用者是否要等待。

所以，我们常用的read()和write()方法都是同步I/O，同步I/O又分为阻塞和非阻塞两种模式，如果是非阻塞模式，检测到无数据可读时，直接就返回了，并没有真正执行I/O操作。

总结就是，Java中实际上只有同步阻塞I/O、同步非阻塞I/O与异步I/O三种机制，我们下文所说的是前两种，JDK1.7才开始引入异步I/O，那称之为NIO.2。

传统IO

我们知道，一个新技术的出现总是伴随着改进和提升，JavaNIO的出现亦如此。

传统I/O是阻塞式I/O，主要问题是系统资源的浪费。比如我们为了读取一个TCP连接的数据，调用InputStream的read()方法，这会使当前线程被挂起，直到有数据到达才被唤醒，那该线程在数据到达这段时间内，占用着内存资源（存储线程栈）却无所作为，也就是俗话说的占着茅坑不拉屎，为了读取其他连接的数据，我们不得不启动另外的线程。在并发连接数量不多的时候，这可能没什么问题，然而当连接数量达到一定规模，内存资源会被大量线程消耗殆尽。另一方面，线程切换需要更改处理器的状态，比如程序计数器、寄存器的值，因此非常频繁的在大量线程之间切换，同样是一种资源浪费。

随着技术的发展，现代操作系统提供了新的I/O机制，可以避免这种资源浪费。基于此，诞生了JavaNIO，NIO的代表性特征就是非阻塞I/O。紧接着我们发现，简单的使用非阻塞I/O并不能解决问题，因为在非阻塞模式下，read()方法在没有读取到数据时就会立即返回，不知道数据何时到达的我们，只能不停的调用read()方法进行重试，这显然太浪费CPU资源了，从下文可以知道，Selector组件正是为解决此问题而生。

JavaNIO核心组件

1.Channel

概念

JavaNIO中的所有I/O操作都基于Channel对象，就像流操作都要基于Stream对象一样，因此很有必要先了解Channel是什么。以下内容摘自JDK1.8的文档

Achannelrepresentsanopenconnectiontoanentitysuchasahardwaredevice,afile,anetworksocket,oraprogramcomponentthatiscapableofperformingoneormoredistinctI/Ooperations,forexamplereadingorwriting.

从上述内容可知，一个Channel（通道）代表和某一实体的连接，这个实体可以是文件、网络套接字等。也就是说，通道是JavaNIO提供的一座桥梁，用于我们的程序和操作系统底层I/O服务进行交互。

通道是一种很基本很抽象的描述，和不同的I/O服务交互，执行不同的I/O操作，实现不一样，因此具体的有FileChannel、SocketChannel等。

通道使用起来跟Stream比较像，可以读取数据到Buffer中，也可以把Buffer中的数据写入通道。

当然，也有区别，主要体现在如下两点：

一个通道，既可以读又可以写，而一个Stream是单向的（所以分InputStream和OutputStream）

通道有非阻塞I/O模式

实现

JavaNIO中最常用的通道实现是如下几个，可以看出跟传统的I/O操作类是一一对应的。

FileChannel：读写文件

DatagramChannel:UDP协议网络通信

SocketChannel：TCP协议网络通信

ServerSocketChannel：监听TCP连接

2.Buffer

NIO中所使用的缓冲区不是一个简单的byte数组，而是封装过的Buffer类，通过它提供的API，我们可以灵活的操纵数据，下面细细道来。

与Java基本类型相对应，NIO提供了多种Buffer类型，如ByteBuffer、CharBuffer、IntBuffer等，区别就是读写缓冲区时的单位长度不一样（以对应类型的变量为单位进行读写）。

Buffer中有3个很重要的变量，它们是理解Buffer工作机制的关键，分别是

capacity（总容量）

position（指针当前位置）

limit（读/写边界位置）

Buffer的工作方式跟C语言里的字符数组非常的像，类比一下，capacity就是数组的总长度，position就是我们读/写字符的下标变量，limit就是结束符的位置。Buffer初始时3个变量的情况如下图

在对Buffer进行读/写的过程中，position会往后移动，而limit就是position移动的边界。由此不难想象，在对Buffer进行写入操作时，limit应当设置为capacity的大小，而对Buffer进行读取操作时，limit应当设置为数据的实际结束位置。（注意：将Buffer数据写入通道是Buffer读取操作，从通道读取数据到Buffer是Buffer写入操作）

在对Buffer进行读/写操作前，我们可以调用Buffer类提供的一些辅助方法来正确设置position和limit的值，主要有如下几个

flip():设置limit为position的值，然后position置为0。对Buffer进行读取操作前调用。

rewind():仅仅将position置0。一般是在重新读取Buffer数据前调用，比如要读取同一个Buffer的数据写入多个通道时会用到。

clear():回到初始状态，即limit等于capacity，position置0。重新对Buffer进行写入操作前调用。

compact():将未读取完的数据（position与limit之间的数据）移动到缓冲区开头，并将position设置为这段数据末尾的下一个位置。其实就等价于重新向缓冲区中写入了这么一段数据。

然后，看一个实例，使用FileChannel读写文本文件，通过这个例子验证通道可读可写的特性以及Buffer的基本用法（注意FileChannel不能设置为非阻塞模式）。

FileChannel channel = new RandomAccessFile("test.txt", "rw").getChannel();
channel.position(channel.size());
// 移动文件指针到末尾（追加写入）
ByteBuffer byteBuffer = ByteBuffer.allocate(20);
// 数据写入Buffer
byteBuffer.put("你好，世界！\n".getBytes(StandardCharsets.UTF_8));
// Buffer -> Channel
byteBuffer.flip();
while (byteBuffer.hasRemaining()) {
	channel.write(byteBuffer);
}
channel.position(0);
// 移动文件指针到开头（从头读取）
CharBuffer charBuffer = CharBuffer.allocate(10);
CharsetDecoder decoder = StandardCharsets.UTF_8.newDecoder();
// 读出所有数据
byteBuffer.clear();
while (channel.read(byteBuffer) != -1 || byteBuffer.position() > 0) {
	byteBuffer.flip();
	// 使用UTF-8解码器解码
	charBuffer.clear();
	decoder.decode(byteBuffer, charBuffer, false);
	System.out.print(charBuffer.flip().toString());
	byteBuffer.compact();
	// 数据可能有剩余
}
channel.close();

这个例子中使用了两个Buffer，其中 byteBuffer 作为通道读写的数据缓冲区，charBuffer 用于存储解码后的字符。clear() 和 flip() 的用法正如上文所述，需要注意的是最后那个 compact() 方法，即使 charBuffer 的大小完全足以容纳 byteBuffer 解码后的数据，这个 compact() 也必不可少，这是因为常用中文字符的UTF-8编码占3个字节，因此有很大概率出现在中间截断的情况，请看下图：

当 Decoder 读取到缓冲区末尾的 0xe4 时，无法将其映射到一个 Unicode，decode()方法第三个参数 false 的作用就是让 Decoder 把无法映射的字节及其后面的数据都视作附加数据，因此 decode() 方法会在此处停止，并且 position 会回退到 0xe4 的位置。如此一来， 缓冲区中就遗留了“中”字编码的第一个字节，必须将其 compact 到前面，以正确的和后序数据拼接起来。关于字符编码,大家可以参阅《ANSI,Unicode,BMP,UTF等编码概念实例讲解》

BTW，例子中的CharsetDecoder也是JavaNIO的一个新特性，所以大家应该发现了一点哈，NIO的操作是面向缓冲区的（传统I/O是面向流的）。

至此，我们了解了Channel与Buffer的基本用法。接下来要说的是让一个线程管理多个Channel的重要组件。

3.Selector

Selector是什么

Selector（选择器）是一个特殊的组件，用于采集各个通道的状态（或者说事件）。我们先将通道注册到选择器，并设置好关心的事件，然后就可以通过调用select()方法，静静地等待事件发生。

通道有如下4个事件可供我们监听：

Accept：有可以接受的连接

Connect：连接成功

Read：有数据可读

Write：可以写入数据了

为什么要用Selector

前文说了，如果用阻塞I/O，需要多线程（浪费内存），如果用非阻塞I/O，需要不断重试（耗费CPU）。Selector的出现解决了这尴尬的问题，非阻塞模式下，通过Selector，我们的线程只为已就绪的通道工作，不用盲目的重试了。比如，当所有通道都没有数据到达时，也就没有Read事件发生，我们的线程会在select()方法处被挂起，从而让出了CPU资源。

使用方法

如下所示，创建一个Selector，并注册一个Channel。

注意：要将Channel注册到Selector，首先需要将Channel设置为非阻塞模式，否则会抛异常。

Selector selector = Selector.open();
channel.configureBlocking(false);
SelectionKey key = channel.register(selector, SelectionKey.OP_READ);

register()方法的第二个参数名叫“interest set”，也就是你所关心的事件集合。如果你关心多个事件，用一个“按位或运算符”分隔，比如

SelectionKey.OP_READ | SelectionKey.OP_WRITE

这种写法一点都不陌生，支持位运算的编程语言里都这么玩，用一个整型变量可以标识多种状态，它是怎么做到的呢，其实很简单，举个例子，首先预定义一些常量，它们的值（二进制）如下

可以发现，它们值为1的位都是错开的，因此对它们进行按位或运算之后得出的值就没有二义性，可以反推出是由哪些变量运算而来。怎么判断呢，没错，就是“按位与”运算。比如，现在有一个状态集合变量值为0011，我们只需要判断“0011&OP_READ”的值是1还是0就能确定集合是否包含OP_READ状态。

然后，注意register()方法返回了一个SelectionKey的对象，这个对象包含了本次注册的信息，我们也可以通过它修改注册信息。从下面完整的例子中可以看到，select()之后，我们也是通过获取一个SelectionKey的集合来获取到那些状态就绪了的通道。

一个完整实例

概念和理论的东西阐述完了（其实写到这里，我发现没写出多少东西，好尴尬(⊙ˍ⊙)），看一个完整的例子吧。

这个例子使用JavaNIO实现了一个单线程的服务端，功能很简单，监听客户端连接，当连接建立后，读取客户端的消息，并向客户端响应一条消息。

需要注意的是，我用字符‘\0′（一个值为0的字节）来标识消息结束。

单线程Server

public class NioServer {
	public static void main(String[] args) throws IOException {
		// 创建一个selector
		Selector selector = Selector.open();
		// 初始化TCP连接监听通道
		ServerSocketChannel listenChannel = ServerSocketChannel.open();
		listenChannel.bind(new InetSocketAddress(9999));
		listenChannel.configureBlocking(false);
		// 注册到selector（监听其ACCEPT事件）
		listenChannel.register(selector, SelectionKey.OP_ACCEPT);
		// 创建一个缓冲区
		ByteBuffer buffer = ByteBuffer.allocate(100);
		while (true) {
			selector.select();
			//阻塞，直到有监听的事件发生
			Iterator<SelectionKey> keyIter = selector.selectedKeys().iterator();
			// 通过迭代器依次访问select出来的Channel事件
			while (keyIter.hasNext()) {
				SelectionKey key = keyIter.next();
				if (key.isAcceptable()) {
					// 有连接可以接受
					SocketChannel channel = ((ServerSocketChannel) key.channel()).accept();
					channel.configureBlocking(false);
					channel.register(selector, SelectionKey.OP_READ);
					System.out.println("与【" + channel.getRemoteAddress() + "】建立了连接！");
				} else if (key.isReadable()) {
					// 有数据可以读取
					buffer.clear();
					// 读取到流末尾说明TCP连接已断开，
					// 因此需要关闭通道或者取消监听READ事件
					// 否则会无限循环
					if (((SocketChannel) key.channel()).read(buffer) == -1) {
						key.channel().close();
						continue;
					}
					// 按字节遍历数据
					buffer.flip();
					while (buffer.hasRemaining()) {
						byte b = buffer.get();
						if (b == 0) {
							// 客户端消息末尾的\0
							System.out.println();
							// 响应客户端
							buffer.clear();
							buffer.put("Hello, Client!\0".getBytes());
							buffer.flip();
							while (buffer.hasRemaining()) {
								((SocketChannel) key.channel()).write(buffer);
							}
						} else {
							System.out.print((char) b);
						}
					}
				}
				// 已经处理的事件一定要手动移除
				keyIter.remove();
			}
		}
	}
}

Client

这个客户端纯粹测试用，为了看起来不那么费劲，就用传统的写法了，代码很简短。

要严谨一点测试的话，应该并发运行大量Client，统计服务端的响应时间，而且连接建立后不要立刻发送数据，这样才能发挥出服务端非阻塞I/O的优势。

public class Client {
	public static void main(String[] args) throws Exception {
		Socket socket = new Socket("localhost", 9999);
		InputStream is = socket.getInputStream();
		OutputStream os = socket.getOutputStream();
		// 先向服务端发送数据
		os.write("Hello, Server!\0".getBytes());
		// 读取服务端发来的数据
		int b;
		while ((b = is.read()) != 0) {
			System.out.print((char) b);
		}
		System.out.println();
		socket.close();
	}
}

总结

以上就是本文关于快速了解Java中NIO核心组件的全部内容，希望对大家有所帮助。感兴趣的朋友可以继续参阅本站其他相关内容，如有不足之处，欢迎留言指出。感谢朋友们对本站的支持！