java.nio.channels.SocketChannel 用于定期写入和立即读取

Question

我想编写一个客户端应用程序，它将消息发送到服务器并接收其回复。无论回复如何，客户端都会多次发送消息(例如，每隔一秒定期发送一条消息)。当回复回来时，客户希望尽快回复。

这是客户端的代码，但不起作用。我希望 startReading() 方法中的可运行实例应该响应服务器的回复，但它没有。在这种情况下，_channel.write(buffer) 无法正确返回。

请让我知道以下代码的问题或实现上述行为的其他方式。

public class MyClient {

    private SocketChannel _channel = null;
    private Selector _selector = null;
    private InetSocketAddress _addr = new InetSocketAddress("127.0.0.1", 5555);

    public MyClient () {
        _selector = SelectorProvider.provider().openSelector();
        _channel = SocketChannel.open();
        _channel.configureBlocking(false);
        startReading();
        _channel.connect(_addr);
    }

    private void startReading () throws IOException {
        ByteBuffer buffer = ByteBuffer.allocate(1024);
        _channel.register(_selector, SelectionKey.OP_READ, buffer);
        Runnable runnable = new Runnable() {
            @Override
            public void run () {
                try {
                    while (0 < _selector.select()) {
                        Iterator<SelectionKey> keyIterator = _selector.selectedKeys().iterator();
                        while (keyIterator.hasNext()) {
                            SelectionKey key = keyIterator.next();
                            keyIterator.remove();
                            if (key.isReadable())
                                read(key);
                        }
                    }
                }
                catch (IOException e) {}
            }
        };
        ExecutorService service = Executors.newFixedThreadPool(1);
        service.execute(runnable);
    }

    private void read(SelectionKey key) throws IOException {
        // do some reading operations
    }

    @Override
    public void run() {
        ByteBuffer buffer = ByteBuffer.allocate(1024);
        // write message to buffer
        buffer.flip();
        try {
            _channel.write(buffer);
        } catch (IOException e) {}
    }

    public static void main (String[] args) {
        MyClient client = new MyClient();
        ScheduledExecutorService ex = Executors.newSingleThreadScheduledExecutor();
        ex.scheduleAtFixedRate(client, 1000, 1000, TimeUnit.MILLISECONDS);
    }
}

Answer 1

您忽略了 channel.write() 的返回代码。它没有义务写入整个缓冲区。在非阻塞模式下，它根本不需要写任何东西。

您必须执行以下操作:

1. 当它返回一个正值时，即它已经写入了一些东西，循环。
2. 如果它返回零并且 buffer.remaining() 为零，则说明您已完成:compact() 缓冲区并返回。
3. 如果它返回零并且 buffer.remaining() 为非-0，则套接字发送缓冲区已满，因此您必须 (a) 压缩缓冲区 ( b) 注册OP_WRITE而不是OP_READ并且(c)返回到选择循环。当 channel 变为可写时，重复上面的 (1) 操作，这次如果您在 (2) 操作中获得成功，则返回注册 OP_READ 而不是 OP_WRITE。换句话说，您正在等待选择器告诉您套接字发送缓冲区何时有空间；尝试使用它；如果您成功完成写入，那么您就再次完成了。