Selector模式的核心原理
Java NIO中的Selector是一个多路复用器，它可以监控多个SelectableChannel的事件状态，当通道上有感兴趣的事件发生时，Selector会通知对应的处理逻辑。这种模式避免了传统BIO中每个连接需要对应一个线程的资源消耗，单线程就能轮询管理大量通道的事件，是支撑高并发连接管理的基础。
Selector的工作依赖于操作系统的底层多路复用机制，比如Linux的epoll、Windows的select，这些机制可以在内核层面监控多个文件描述符的状态变化，当有任何描述符就绪时，内核会通知用户态程序，避免了用户态频繁轮询所有描述符的开销。
实现单线程管理十万级连接的步骤
1. 创建Selector和ServerSocketChannel
首先需要创建Selector实例，同时打开ServerSocketChannel并配置为非阻塞模式，绑定监听端口。
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
public class SelectorDemo {
public static void main(String[] args) throws IOException {
// 创建Selector实例
Selector selector = Selector.open();
// 打开ServerSocketChannel
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
// 配置为非阻塞模式
serverSocketChannel.configureBlocking(false);
// 绑定监听端口
serverSocketChannel.bind(new InetSocketAddress(8080));
System.out.println("服务端启动，监听端口8080");
}
}
2. 注册通道到Selector
将ServerSocketChannel注册到Selector上，关注接收连接事件，后续新的客户端连接会通过这个事件触发处理。
import java.nio.channels.SelectionKey; // 接上面的代码，注册通道到Selector // 注册时关注OP_ACCEPT事件，不附加额外附件 serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
3. 轮询处理就绪事件
通过Selector的select方法阻塞等待就绪事件，当有事件发生时，获取对应的SelectionKey集合，遍历处理每个就绪的事件。
import java.nio.channels.SocketChannel;
import java.util.Iterator;
import java.util.Set;
// 接上面的代码，启动轮询逻辑
while (true) {
// 阻塞等待就绪事件，超时时间设置为1000毫秒
int readyChannels = selector.select(1000);
if (readyChannels == 0) {
continue;
}
// 获取所有就绪的SelectionKey
Set<SelectionKey> selectedKeys = selector.selectedKeys();
Iterator<SelectionKey> keyIterator = selectedKeys.iterator();
while (keyIterator.hasNext()) {
SelectionKey key = keyIterator.next();
// 处理对应的事件
if (key.isAcceptable()) {
// 处理新连接接入
handleAccept(key, selector);
} else if (key.isReadable()) {
// 处理读事件
handleRead(key);
}
// 移除已经处理的Key，避免重复处理
keyIterator.remove();
}
}
4. 处理连接接入和读写事件
新连接接入时，需要将新的SocketChannel配置为非阻塞模式，注册到Selector上关注读事件；读事件触发时，读取通道中的数据并处理。
import java.nio.ByteBuffer;
import java.nio.charset.StandardCharsets;
private static void handleAccept(SelectionKey key, Selector selector) throws IOException {
// 获取ServerSocketChannel
ServerSocketChannel serverChannel = (ServerSocketChannel) key.channel();
// 接收新的客户端连接
SocketChannel socketChannel = serverChannel.accept();
if (socketChannel != null) {
// 配置新的通道为非阻塞模式
socketChannel.configureBlocking(false);
// 注册到Selector，关注读事件
socketChannel.register(selector, SelectionKey.OP_READ);
System.out.println("新连接接入：" + socketChannel.getRemoteAddress());
}
}
private static void handleRead(SelectionKey key) throws IOException {
// 获取对应的SocketChannel
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, StandardCharsets.UTF_8);
System.out.println("收到客户端消息：" + message);
// 回写响应数据
ByteBuffer responseBuffer = ByteBuffer.wrap(("已收到消息：" + message).getBytes(StandardCharsets.UTF_8));
socketChannel.write(responseBuffer);
} else if (bytesRead == -1) {
// 客户端关闭连接，取消注册并关闭通道
System.out.println("客户端断开连接：" + socketChannel.getRemoteAddress());
key.cancel();
socketChannel.close();
}
}
高效管理十万级连接的关键优化点
- 合理设置select方法的超时时间，避免过长时间阻塞或者频繁空轮询，示例中设置为1000毫秒，可根据实际场景调整。
- 及时移除已经处理过的SelectionKey，避免重复处理导致逻辑错误和性能损耗。
- 缓冲区大小根据实际业务场景调整，避免过大浪费内存或者过小导致多次读写。
- 如果读写逻辑较重，可以将处理逻辑放到线程池中执行，避免阻塞轮询线程，但是通道的注册和事件监听仍然保持在单线程中。
- 避免在非阻塞通道上进行阻塞操作，所有IO操作都需要基于NIO的非阻塞特性实现。
注意事项
Selector的select方法在Linux系统下默认使用epoll机制，能够高效处理大量文件描述符，但是需要注意操作系统的文件描述符限制，十万级连接需要调大系统的文件描述符上限，避免连接数达到上限后无法创建新的连接。
同时，单线程管理十万级连接的前提是每个连接上的事件处理足够轻量，如果单个连接的读写逻辑非常耗时，会阻塞整个轮询流程，导致其他连接的事件无法及时处理，这种情况下需要结合线程池将耗时的业务逻辑异步处理。