如何有效地阅读 Java（在 Scala 中使用 Java 库）InputStream？

Question

我的 Scala 服务器通过socket.getInputStream 从套接字获取InputStream 对象（从我的套接字客户端发送的一些字节，字节的大小打印在下面）

下面的代码尝试将其读入数组

  var buffer: Array[Byte] = null
  def read(stream: InputStream, size: Int) = {
    val start = System.nanoTime()
    buffer = new Array[Byte](size)
    var value: Int = 0
    (0 until size).foreach(i => {
      value = stream.read()
      buffer(i) = value.toByte
    })
    val end = System.nanoTime()
    println(s"Getting buffer from InputStream, size: $size, cost: ${(end - start)/1e6} ms")
    buffer
  }

部分输出是

Getting buffer from InputStream, size: 4, cost: 174.923596 ms
Getting buffer from InputStream, size: 2408728, cost: 919.207885 ms

但是，对于相同的数据大小，一些现有的服务器可能会更快，例如Redis 可以在 ~10ms 内发送字节，所以

是否可以提高此程序的性能？

Answer 1

stream.read() 是对这个概念的最慢理解。

相反，您需要read(byte[]) 变体或read(byte[], int offset, int length) 变体（一个非常简单且性能方面基本上免费的 3 参数方法包装器）。

使用read() 的“开销”从“轻微”（如果涉及缓冲区）到“1000 倍”（如果没有缓冲区）不等。如果是第二个，您可以通过将输入流包装在 BufferedInputStream 中并从中读取来回到“轻微”开销。

但不管发生什么，这个：

int toRead = 1000;
byte[] data = new byte[toRead];
int readSoFar = 0;
while (readSoFar < toRead) {
  int read = in.read(data, readSoFar, toRead - readSoFar);
  if (read == -1) throw new IOException("Expected more data");
  toRead += read;
}

比远快：

int toRead = 1000;
byte[] data = new byte[toRead];
while (toRead > 0) {
  data[toRead--] = in.read();
}

使用 scala 对这些示例的性能没有影响。