一个线程一个ByteBuffer NIO

时间:2015-06-15 02:51:42

标签: java nio bytebuffer

我想知道实现具有一个线程和一个缓冲区的多用户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);

1 个答案:

答案 0 :(得分:1)

你做不到。对于任何比echo服务器更简单的事情,每个通道至少需要一个缓冲区,可能需要两个(读写)。否则,您无法处理部分读取请求或部分书面响应。

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