从InputStream读取所有字节

Question

我想从活动tcp连接的InputStream异步读取所有字节。 输入缓冲区中可能没有任何内容，在这种情况下，我需要一个空字节array []而不是读取/超时。

我尝试了这个：

byte[] bytes = IOUtils.toByteArray(tcpInputStream);

但是，如果InputStream中没有任何内容，则读取将挂起，直到tcp超时，然后引发异常。

Answer 1

您可以使用非阻塞SocketChannel：

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;

...

SocketChannel socketChannel = SocketChannel.open();
socketChannel.connect(new InetSocketAddress("127.0.0.1", 80));
ByteBuffer buffer = ByteBuffer.allocate(2048);
int bytesRead;
while ((bytesRead = socketChannel.read(buffer)) >= 0) {
    if (bytesRead == 0)
        Thread.sleep(1000); // do usefull stuff
    else {
        // use the bytes and prepare the buffer for next read
        buffer.clear();
    }
}
socketChannel.close();

当服务器关闭连接时，还要准备IOException 。