从 InputStreamReader 读取字节进入无限循环

Question

我的测试失败了，因为从 InputStreamReader 读取字节会进入无限循环。

@Test
public void testSingleRequest() throws IOException {
        Server server = new Server(9000, 100);
        new Thread(server).start();
        Socket clientSocket = new Socket("localhost", 9000);
        OutputStream out = clientSocket.getOutputStream();
        InputStream in = clientSocket.getInputStream();
        String payload = "a";
        byte[] load = buildPushPayload(payload);
        out.write(load);
        byte[] response = IOUtils.toByteArray(in);
        server.stop();

        assertEquals(0, response[0]);


    }

这是写入 InputStream 的代码

byte[] resp = buildPushResponse();
output.write(resp);

private byte[] buildPushResponse() throws UnsupportedEncodingException {
        ByteBuffer buffer = ByteBuffer.allocate(1);
        buffer.order(ByteOrder.BIG_ENDIAN);
        buffer.putInt(0);
        return buffer.array();
    }

服务端和客户端都写入clientSocket输入输出流。 我似乎无法理解为什么会出现无限循环？

Answer 1

您在这里缺少的是如何知道何时停止阅读。您正在调用IOUtils.toByteArray(in)，它将一直读取到文件结束，这永远不会发生（除非另一方关闭其套接字末端）。

在处理完整的消​​息包之前，您可以使用一些方案来了解要从套接字读取多少字节作为消息包。例如，您可以说所有数据包都是 8 个字节（在这种情况下，您的 read 调用将指定读取 8 个字节），您可以说第一个字节是数据包中的字节数，您可以说一个数据包是可变长度，并以回车字节结束。

一旦您读取了一个数据包，您通常会处理该数据包，然后返回读取另一个数据包或关闭套接字（如果您已完成）。

这是一个很好的套接字编程示例：https://www.geeksforgeeks.org/socket-programming-in-java/

在这种情况下，他们使用 End Of Line 标记终止数据包或消息 - 他们一次读取一行文本，使用 DataInputStream.readUTF(socket)。

server = new ServerSocket(port); 
socket = server.accept(); 

// Create a DataInputStream object that takes input from the socket
in = new DataInputStream(new BufferedInputStream(socket.getInputStream())); 

String message = ""; 

// reads messages from client and prints them
// The loop exits if the message "Over" is received.
try
{ 
    while (!"Over".equals(message))
    { 
        message = in.readUTF(); 
        System.out.println(message); 
    }
}
catch(IOException i) 
{ 
    // handle error
} 
// close connection 
socket.close(); 
in.close();