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

时间:2014-09-16 17:56:38

标签: java client selector nio socketchannel

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

这是客户端的代码,它不起作用。我希望startReading()方法中的runnable实例应该响应来自服务器的回复,但事实并非如此。在这种情况下,_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);
    }
}

1 个答案:

答案 0 :(得分:1)

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

您必须执行以下操作:

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