javaIO 模型之 NIO

有的时候博客内容会有变动，首发博客是最新的，其他博客地址可能会未同步, 认准https://blog.zysicyj.top

全网最细面试题手册，支持艾宾浩斯记忆法。这是一份最全面、最详细、最高质量的 java
面试题，不建议你死记硬背，只要每天复习一遍，有个大概印象就行了。 https://store.amazingmemo.com/chapterDetail/1685324709017001

1. 什么是 NIO?

NIO（New I/O）是 Java 提供的一种非阻塞 I/O 模型，它在 JDK 1.4 中引入。与传统的 I/O 模型相比，NIO 提供了更高效、更灵活的 I/O 操作方式。

2. 为什么需要 NIO?

传统的 I/O 模型使用阻塞式 I/O，在进行读写操作时会导致线程被阻塞，直到数据准备好或者写入完成。这种模型对于并发处理能力较弱，当有大量连接同时请求时，每个连接都需要一个独立的线程来处理，造成资源浪费和性能下降。

而 NIO 采用了事件驱动的方式，通过 Selector 轮询注册的通道，只有在通道真正有读写事件发生时才会进行处理，避免了线程被阻塞的情况，提高了系统的并发处理能力。

3. NIO 的实现原理?

NIO 的核心组件包括：Channel、Buffer 和 Selector。

  • Channel: 类似于传统 I/O 中的流，可以通过 Channel 进行数据的读取和写入。常见的 Channel 类型有 SocketChannel、ServerSocketChannel、FileChannel 等。
  • Buffer: 缓冲区，用于存储数据。在 NIO 中，所有的数据都是通过 Buffer 进行读写的。常见的 Buffer 类型有 ByteBuffer、CharBuffer、IntBuffer 等。
  • Selector: 选择器，用于监听多个 Channel 的事件。通过 Selector 可以实现单线程处理多个通道的读写操作。

NIO 的工作原理如下：

  1. 创建一个 Selector，并将其注册到需要监听的 Channel 上。
  2. 当有数据准备好时，Selector 会轮询已注册的 Channel，发现有事件发生的 Channel 后进行处理。
  3. 根据不同的事件类型（读、写、连接、接收），使用对应的方法进行处理。

4. NIO 的使用示例

以下是一个简单的 NIO 服务器示例，用于接收客户端发送的消息并返回相同的消息给客户端：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Iterator;

public class NIOServer {
    public static void main(String[] args) throws IOException {
        // 创建 ServerSocketChannel
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        serverSocketChannel.socket().bind(new InetSocketAddress(8888));
        serverSocketChannel.configureBlocking(false);

        // 创建 Selector
        Selector selector = Selector.open();
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

        while (true) {
            // 阻塞等待就绪的 Channel
            int readyChannels = selector.select();

            if (readyChannels == 0) {
                continue;
            }

            // 获取就绪的 SelectionKey 集合
            Iterator<SelectionKey> keyIterator = selector.selectedKeys().iterator();

            while (keyIterator.hasNext()) {
                SelectionKey key = keyIterator.next();

                if (key.isAcceptable()) {
                    // 接收新的连接
                    ServerSocketChannel serverChannel = (ServerSocketChannel) key.channel();
                    SocketChannel socketChannel = serverChannel.accept();
                    socketChannel.configureBlocking(false);
                    socketChannel.register(selector, SelectionKey.OP_READ);
                } else if (key.isReadable()) {
                    // 读取数据
                    SocketChannel socketChannel = (SocketChannel) key.channel();
                    ByteBuffer buffer = ByteBuffer.allocate(1024);
                    int bytesRead = socketChannel.read(buffer);

                    if (bytesRead > 0) {
                        buffer.flip();
                        byte[] data = new byte[buffer.remaining()];
                        buffer.get(data);
                        String message = new String(data, "UTF-8");
                        System.out.println("Received message: " + message);

                        // 返回相同的消息给客户端
                        ByteBuffer responseBuffer = ByteBuffer.wrap(data);
                        socketChannel.write(responseBuffer);
                    } else if (bytesRead < 0) {
                        // 客户端关闭连接
                        socketChannel.close();
                    }
                }

                // 移除处理过的 SelectionKey
                keyIterator.remove();
            }
        }
    }
}

5. NIO 的优点

  • 高并发性：NIO 采用了事件驱动的方式，可以使用单线程处理多个通道的读写操作，提高系统的并发处理能力。
  • 非阻塞式 I/O：NIO 模型中的通道是非阻塞的，不会因为某个通道的读写操作而导致线程被阻塞，提高了系统的响应速度和吞吐量。
  • 内存管理优化：NIO 使用了直接内存缓冲区，可以减少数据在 Java 堆和操作系统之间的拷贝次数，提高了 I/O 性能。

6. NIO 的缺点

  • 复杂性较高：相比传统的阻塞式 I/O 模型，NIO 的编程模型更加复杂，需要处理事件驱动、多路复用等概念。
  • 对编程人员要求较高：由于 NIO 的复杂性，对编程人员的技术要求较高，需要熟悉 NIO 相关的 API 和底层原理。

7. NIO 的使用注意事项

  • 需要合理设置缓冲区大小：过小的缓冲区可能导致频繁的读写操作，影响性能；过大的缓冲区可能造成资源浪费。
  • 注意正确释放资源：在使用完 Channel 和 Buffer 后，需要及时关闭并释放资源，避免出现资源泄露问题。
  • 谨慎处理异常情况：NIO 中的异常处理相对复杂，需要仔细处理各种异常情况，以保证程序的稳定性和可靠性。

8. 总结

NIO 是一种非阻塞 I/O 模型，通过事件驱动和选择器机制实现高效的 I/O 操作。它具有高并发性、非阻塞式 I/O 和内存管理优化等优点，但也存在复杂性较高和对编程人员要求较高的缺点。在使用 NIO 时需要注意合理设置缓冲区大小、正确释放资源和谨慎处理异常情况。