一线程一ByteBuffer NIO

One thread one ByteBuffer NIO

我想知道用一个线程和一个缓冲区实现多用户 NIO 服务器的最佳方法。目前我使用选择器来实现这一点,但我只是想出了如何对所有客户端进行阅读。我在使用一个缓冲区实现写入时遇到问题。我需要第二个缓冲区来写入吗?或者我必须(不幸的是)每个客户端都有一个写缓冲区?我写这个例子是为了轻松地展示我是如何做的,只是让你知道所有的处理都发生在调度程序的线程中,我没有通过另一个线程与任何状态交互!

ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor();

ServerSocketChannel server = ServerSocketChannel.open();
Selector selector = Selector.open();
ByteBuffer buffer = ByteBuffer.allocateDirect(8192);
Map<SelectionKey, SocketChannel> clients = new HashMap<>();

server.configureBlocking(false);
server.register(selector, SelectionKey.OP_ACCEPT);
server.bind(new InetSocketAddress(43594));

scheduler.scheduleAtFixedRate(() -> {
    try {
        selector.selectNow();
        Iterator<SelectionKey> keys = selector.selectedKeys().iterator();
        while (keys.hasNext()) {
            SelectionKey key = keys.next();
            keys.remove();
            if (key.isAcceptable()) {
                for (int i = 0; i < 16; i++) {
                    SocketChannel client = server.accept();
                    if (client == null)
                        break;
                    client.configureBlocking(false);
                    SelectionKey clientKey = client.register(selector, SelectionKey.OP_READ);
                    clients.put(clientKey, client);
                }
            }
            if (key.isReadable()) {
                SocketChannel client = clients.get(key);
                if (client != null) {
                    buffer.clear();
                    client.read(buffer);
                    buffer.flip();
                    // do stuff with buffer
                }
            }
        }
    } catch (IOException e) {
        e.printStackTrace();
    }
}, 600, 600, TimeUnit.MILLISECONDS);

你不能。对于任何比回显服务器更简单的东西,每个通道至少需要一个缓冲区,可能两个(读和写)。否则您将无法处理部分读取的请求或部分写入的响应。

您可以通过 SelectionKey 附件将缓冲区与频道相关联。