Java NIO如何处理并发连接

Java NIO（New I/O）是一个用于处理并发连接的强大框架，它提供了非阻塞I/O操作和选择器（Selector）来处理多个连接。以下是使用Java NIO处理并发连接的基本步骤：

1. 导入必要的包：

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;

2. 创建一个ServerSocketChannel实例并设置为非阻塞模式：

ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.configureBlocking(false);

3. 创建一个Selector实例：

Selector selector = Selector.open();

4. 将ServerSocketChannel注册到Selector上，监听连接事件（OP_ACCEPT）：

serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

5. 创建一个循环来处理Selector上的事件：

while (true) {
    selector.select(); // 阻塞，直到有事件发生
    Set<SelectionKey> selectedKeys = selector.selectedKeys(); // 获取已触发的事件
    Iterator<SelectionKey> keyIterator = selectedKeys.iterator();

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

        if (key.isAcceptable()) {
            // 处理连接事件
            handleAccept(key);
        } else if (key.isReadable()) {
            // 处理读取事件
            handleRead(key);
        }

        keyIterator.remove();
    }
}

6. 实现handleAccept方法，用于处理连接事件：

private void handleAccept(SelectionKey key) throws IOException {
    ServerSocketChannel serverSocketChannel = (ServerSocketChannel) key.channel();
    SocketChannel socketChannel = serverSocketChannel.accept();
    socketChannel.configureBlocking(false);
    socketChannel.register(key.selector(), SelectionKey.OP_READ);
}

7. 实现handleRead方法，用于处理读取事件：

private void handleRead(SelectionKey key) throws IOException {
    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).trim();
        System.out.println("Received message: " + message);

        // 可以继续注册其他事件，如OP_WRITE
    } else if (bytesRead == -1) {
        // 连接已关闭，取消SelectionKey
        key.cancel();
        socketChannel.close();
    }
}

通过以上步骤，Java NIO可以有效地处理并发连接。需要注意的是，这里的示例仅用于演示基本概念，实际应用中可能需要根据具体需求进行调整和优化。