提高简单 TCP 客户端/套接字上的数据传输速度

Question

我有一个家庭作业，通过重定向标准 I/O 来创建一个带有客户端/服务器 TCP 套接字对的简单数据传输机制。我实际上可以正常工作，但是当我尝试传输大文件（比如~5g）时，速度会急剧下降。我正在使用 BufferedInputStream 和 BufferedOutputStream，我认为也许我可以在那里进行一些优化。我的服务器的代码是：

private static final int BUF_SIZE = 2047;

public static void main(String[] args) throws IOException{
   /*
    * Attempt to parse command line arguments.
    * @require args[0] is an int
    */
   int port = 0;
   try {
       port = Integer.parseInt(args[0]);
   } catch(NumberFormatException e) {
       System.err.println("Port must be an integer in range 0 - 65535.");
       System.exit(-1);
   }

   /*
    * Bind server socket to specified port number and wait for request.
    * @require port >= 0 && port <= 65535
    */
   ServerSocket welcomeSocket = null;
   welcomeSocket = new ServerSocket(port);
   System.out.println("Now listening on port: " + port);

    /*
     * Accept connection from client socket.
     */
    Socket connectionSocket = null;
    connectionSocket = welcomeSocket.accept();
    System.out.println("Client made connection");

    BufferedInputStream input;
    BufferedOutputStream output;
    if(System.in.available() > 0) {
        input = new BufferedInputStream(System.in, BUF_SIZE);
        output = new BufferedOutputStream(
                connectionSocket.getOutputStream(), BUF_SIZE);
    } else {
        input = new BufferedInputStream(
                connectionSocket.getInputStream(), BUF_SIZE);
        output = new BufferedOutputStream(System.out, BUF_SIZE);
    }

    int place;
    while((place = input.read()) != -1)
        output.write(place);

    input.close();
    output.close();
    welcomeSocket.close();
    connectionSocket.close();
}

客户端代码本质上是相同的。我尝试过使用不同的缓冲区大小，包括默认值（不指定缓冲区大小），但它们都以大致相同的速度运行。关于如何提高我的表现有什么建议吗？

谢谢您的宝贵时间！

Answer 1

while((place = input.read()) != -1)

您一次从缓冲区读取一个字节。调用该方法数百万次的开销是相当大的。

我建议使用其他版本将多个字节读取到缓冲区中（并以相同的方式写入）：

public int read(byte[] b,
            int off,
            int len)

示例：

byte[] myBuffer = new byte[BUF_SIZE];
while((place = input.read(myBuffer, 0, BUF_SIZE)) != 1)
    output.write(myBuffer, 0, place);

Answer 2

您一次读取和发送一个字节，效率不高，您应该读取数据块（空闲大小将是磁盘硬件缓冲区大小）。
当然，磁盘应该是你的瓶颈，读取5G格式的磁盘需要时间。