解锁Java性能新境界：NIO，让你的代码飞起来！

Java NIO（New Input/Output）是Java 1.4引入的一套新的I/O API，旨在提高I/O操作的性能和可扩展性。NIO通过使用非阻塞I/O、选择器（Selectors）、缓冲区（Buffers）等机制，使得应用程序能够更高效地处理大量并发连接。

以下是一个简单的Java 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;
import java.util.Set;

public class NIOServer {
    private static final int PORT = 8080;

    public static void main(String[] args) throws IOException {
        // 打开一个ServerSocketChannel
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        serverSocketChannel.socket().bind(new InetSocketAddress(PORT));
        serverSocketChannel.configureBlocking(false);

        // 打开一个Selector
        Selector selector = Selector.open();
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

        System.out.println("Server started on port " + PORT);

        while (true) {
            // 选择一组键，其相应的通道已准备好进行I/O操作
            selector.select();

            // 获取已选择的键集
            Set<SelectionKey> selectedKeys = selector.selectedKeys();
            Iterator<SelectionKey> keyIterator = selectedKeys.iterator();

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

                if (key.isAcceptable()) {
                    // 接受一个新的连接
                    ServerSocketChannel serverChannel = (ServerSocketChannel) key.channel();
                    SocketChannel socketChannel = serverChannel.accept();
                    socketChannel.configureBlocking(false);
                    socketChannel.register(selector, SelectionKey.OP_READ);
                    System.out.println("Accepted new connection from " + socketChannel.getRemoteAddress());
                } else if (key.isReadable()) {
                    // 读取数据
                    SocketChannel socketChannel = (SocketChannel) key.channel();
                    ByteBuffer buffer = ByteBuffer.allocate(256);
                    int bytesRead = socketChannel.read(buffer);
                    if (bytesRead == -1) {
                        socketChannel.close();
                    } else {
                        buffer.flip();
                        byte[] data = new byte[buffer.limit()];
                        buffer.get(data);
                        System.out.println("Received: " + new String(data));
                        buffer.clear();
                    }
                }
            }
        }
    }
}

详解

1. ServerSocketChannel:

   • ServerSocketChannel是一个可以监听新进来的TCP连接的通道。它类似于传统的ServerSocket，但具有非阻塞模式。
   • serverSocketChannel.configureBlocking(false);将通道设置为非阻塞模式。

2. Selector:

   • Selector用于检查一个或多个NIO通道的状态。它可以检测通道是否准备好进行读、写或连接操作。
   • Selector selector = Selector.open();创建一个新的选择器。
   • serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);将服务器通道注册到选择器上，并指定感兴趣的操作为“接受连接”。

3. SelectionKey:

   • SelectionKey表示选择器中的一个注册通道的特定操作。每个通道在注册时都会返回一个SelectionKey对象。
   • selector.select();阻塞直到至少有一个通道在你注册的操作上就绪。
   • Set<SelectionKey> selectedKeys = selector.selectedKeys();获取所有已选择的键。
   • keyIterator.remove();从集合中移除当前键，防止重复处理。

4. ByteBuffer:

   • ByteBuffer是一个用于存储字节数据的缓冲区。它是NIO的核心组件之一，提供了对字节数组的高效访问。
   • ByteBuffer buffer = ByteBuffer.allocate(256);分配一个新的字节缓冲区。
   • int bytesRead = socketChannel.read(buffer);从通道中读取数据到缓冲区。
   • buffer.flip();切换缓冲区的模式，从写模式切换到读模式。
   • byte[] data = new byte[buffer.limit()];创建一个字节数组来存储读取的数据。
   • buffer.get(data);将缓冲区中的数据复制到字节数组中。
   • buffer.clear();清空缓冲区，准备下一次写入。

5. 处理逻辑:

   • 如果key.isAcceptable()为真，表示有新的连接请求，需要接受该连接并将其注册到选择器上，关注读操作。
   • 如果key.isReadable()为真，表示通道中有数据可读，读取数据并打印出来。如果读取到的数据长度为-1，表示客户端关闭了连接，关闭通道。

总结

Java NIO通过非阻塞I/O、选择器和缓冲区等机制，使得应用程序能够更高效地处理大量并发连接。上述示例展示了如何使用NIO构建一个简单的服务器，能够接受客户端连接并读取数据。