【面试题精讲】javaIO模型之NIO

程序员朱永胜

发布于 2023-10-22 08:45:17

1590

发布于 2023-10-22 08:45:17

文章被收录于专栏：朱永胜的私房菜朱永胜的私房菜

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的工作原理如下：

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

4. NIO的使用示例

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

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();
            }
        }
    }
}