手动设置DataInputStream的超时

Question

我正在两个站（A，B）之间建立标准 TCP 连接 A 正在发送消息，B 接收并发送回响应，然后我关闭连接。

• B 站是“黑匣子”，我无法在那里找零或做任何事情。

有时会出现B没有发回响应的情况，然后我需要重新尝试整个过程。

我想在A站的接收时间上设置超时（等待B站的答复）。 所以基本上当等待时间到期时，我会调度重试。

我没有找到如何为 DataInputStream 设置超时的方法。 （仅适用于整个套接字连接 - 我不想要）

一些代码：

 /**
 * Method receives the Server Response
 */
public byte[] receive(DataInputStream is) throws Exception
{
    logger.debug(TAG + " Client Recieving...");

    try
    {

        byte[] inputData = new byte[1024];
                     // here I want to set timeout for the "receiving mode"
        is.read(inputData);
        return inputData;
    } catch (Exception e)
    {
        throw new Exception(TAG + " Couldnt receive data from modem: " + e.getMessage());
    }
}

谢谢， 射线。

Answer 1

首先查看 socket.setSoTimeout(timeout)

其次，请参阅此讨论：是否可以从具有超时的输入流中读取？

读取方法被阻止。检查流是否包含数据而不阻塞的唯一方法是使用 available() 方法。所以你可以尝试以一定的延迟调用这个方法，如果n秒后没有可用的则退出。您可以编写自己的输入流来包装实现此逻辑的任何其他输入流。上面的参考显示了如何执行此操作。

Answer 2

考虑使用非阻塞 SocketChannel 而不是 DataInputStream。

例子：

private static final long TIMEOUT = 500;

/**
 * Method receives the Server Response
 */
public <C extends SelectableChannel & ReadableByteChannel>byte[]
receive(C chan) throws IOException
{
    logger.debug(TAG + " Client Recieving...");
    try
    {
        Selector sel = Selector.open();
        SelectionKey key = chan.register(sel, SelectionKey.OP_READ);
        ByteBuffer inputData = ByteBuffer.allocate(1024);
        long timeout = TIMEOUT;
        while (inputData.hasRemaining()) {
            if (timeout < 0L) {
                throw new IOException(String.format("Timed out, %d of %d bytes read", inputData.position(), inputData.limit()));
            }
            long startTime = System.nanoTime();
            sel.select(timeout);
            long endTime = System.nanoTime();
            timeout -= TimeUnit.NANOSECONDS.toMillis(endTime - startTime);
            if (sel.selectedKeys().contains(key)) {
                chan.read(inputData);
            }
            sel.selectedKeys().clear();
        }
        return inputData.array();
    } catch (Exception e)
    {
        throw new Exception(TAG + " Couldnt receive data from modem: " + e.getMessage());
    }
}