如何在服务器套接字 JAVA 中读取所有输入流

我正在我的一个项目中使用 Java.net。我写了一个从客户端获取输入流的应用服务器。但有时我的(缓冲的)InputStream无法获取客户端发送到我的服务器的所有输出流。我该如何编写一个等待或类似的东西,让我的输入流获得客户端的所有输出流?

(我的输入流不是字符串)

private Socket clientSocket;
private ServerSocket server;
private BufferedOutputStream outputS;
private BufferedInputStream inputS;
private InputStream inBS;
private OutputStream outBS;

server = new ServerSocket(30501, 100);
clientSocket = server.accept();

public void getStreamFromClient()  {
    try {
        outBS = clientSocket.getOutputStream();
        outputS = new BufferedOutputStream( outBS);
        outputS.flush();

        inBS = clientSocket.getInputStream();
        inputS = new BufferedInputStream( inBS );

    } catch (Exception e) {
        e.printStackTrace();
    }
}

谢谢。


答案 1

您遇到的问题与TCP流性质有关。

您从服务器发送了 100 个字节(例如)这一事实并不意味着您第一次读取时将在客户端中读取 100 个字节。也许从服务器发送的字节以多个 TCP 段的形式到达客户端。

您需要实现一个循环,在该循环中读取直到收到整个消息。让我用 代替 提供一个示例。一些非常简单的事情,给你一个例子。DataInputStreamBufferedinputStream

假设您事先知道服务器将发送100字节的数据。

在客户端中,您需要编写:

byte[] messageByte = new byte[1000];
boolean end = false;
String dataString = "";

try 
{
    DataInputStream in = new DataInputStream(clientSocket.getInputStream());

    while(!end)
    {
        int bytesRead = in.read(messageByte);
        dataString += new String(messageByte, 0, bytesRead);
        if (dataString.length == 100)
        {
            end = true;
        }
    }
    System.out.println("MESSAGE: " + dataString);
}
catch (Exception e)
{
    e.printStackTrace();
}

现在,通常一个节点(此处为服务器)发送的数据大小事先是未知的。然后,您需要为服务器和客户端(或任何两个节点)与TCP通信之间的通信定义自己的小协议。

最常见和最简单的是定义TLV:类型,长度,值。因此,您定义表单服务器发送到客户端的每条消息都附带:

  • 1 个字节表示类型(例如,它也可以是 2 或其他)。
  • 1 字节(或其他)表示消息长度
  • N 个字节作为值(N 以长度表示)。

因此,您知道必须至少接收2个字节,并且对于第二个字节,您知道需要读取多少个后续字节。

这只是一个可能协议的建议。您也可以摆脱“类型”。

所以它是这样的:

byte[] messageByte = new byte[1000];
boolean end = false;
String dataString = "";

try 
{
    DataInputStream in = new DataInputStream(clientSocket.getInputStream());
    int bytesRead = 0;

    messageByte[0] = in.readByte();
    messageByte[1] = in.readByte();

    int bytesToRead = messageByte[1];

    while(!end)
    {
        bytesRead = in.read(messageByte);
        dataString += new String(messageByte, 0, bytesRead);
        if (dataString.length == bytesToRead )
        {
            end = true;
        }
    }
    System.out.println("MESSAGE: " + dataString);
}
catch (Exception e)
{
    e.printStackTrace();
}

以下代码可编译并看起来更好。它假设提供长度的前两个字节以二进制格式到达,以网络字节序(大字节序)表示。不关注消息其余部分的不同编码类型。

import java.nio.ByteBuffer;
import java.io.DataInputStream;
import java.net.ServerSocket;
import java.net.Socket;

class Test
{
    public static void main(String[] args)
    {
        byte[] messageByte = new byte[1000];
        boolean end = false;
        String dataString = "";

        try 
        {
            Socket clientSocket;
            ServerSocket server;

            server = new ServerSocket(30501, 100);
            clientSocket = server.accept();

            DataInputStream in = new DataInputStream(clientSocket.getInputStream());
            int bytesRead = 0;

            messageByte[0] = in.readByte();
            messageByte[1] = in.readByte();
            ByteBuffer byteBuffer = ByteBuffer.wrap(messageByte, 0, 2);

            int bytesToRead = byteBuffer.getShort();
            System.out.println("About to read " + bytesToRead + " octets");

            //The following code shows in detail how to read from a TCP socket

            while(!end)
            {
                bytesRead = in.read(messageByte);
                dataString += new String(messageByte, 0, bytesRead);
                if (dataString.length() == bytesToRead )
                {
                    end = true;
                }
            }

            //All the code in the loop can be replaced by these two lines
            //in.readFully(messageByte, 0, bytesToRead);
            //dataString = new String(messageByte, 0, bytesToRead);

            System.out.println("MESSAGE: " + dataString);
        }
        catch (Exception e)
        {
            e.printStackTrace();
        }
    }
}

答案 2

您可以像这样读取缓冲输入流。它将读取数据,直到到达由 -1 指示的流结束。

inputS = new BufferedInputStream(inBS);
byte[] buffer = new byte[1024];    //If you handle larger data use a bigger buffer size
int read;
while((read = inputS.read(buffer)) != -1) {
    System.out.println(read);
    // Your code to handle the data
}

推荐