Scala：输入流到数组[字节]

Question

使用 Scala，从 InputStream 读取字节数组的最佳方法是什么？

我可以看到您可以将 InputStream 转换为 char 数组

Source.fromInputStream(is).toArray()

Answer 1

怎么样：

Stream.continually(is.read).takeWhile(_ != -1).map(_.toByte).toArray

更新：使用 LazyList 而不是 Stream （因为 Stream 在 Scala 3 中已弃用）

LazyList.continually(is.read).takeWhile(_ != -1).map(_.toByte).toArray

Answer 2

Or来消除服务器代码中的瓶颈

Stream.continually(request.getInputStream.read()).takeWhile(_ != -1).map(_.toByte).toArray

刚刚通过在纯 Scala 中替换为

org.apache.commons.io.IOUtils.toByteArray(request.getInputStream)

：

def bytes(in: InputStream, initSize: Int = 8192): Array[Byte] = {
  var buf = new Array[Byte](initSize)
  val step = initSize
  var pos, n = 0
  while ({
    if (pos + step > buf.length) buf = util.Arrays.copyOf(buf, buf.length << 1)
    n = in.read(buf, pos, step)
    n != -1
  }) pos += n
  if (pos != buf.length) buf = util.Arrays.copyOf(buf, pos)
  buf
}

在任何情况下都不要忘记关闭打开的输入流：

val in = request.getInputStream
try bytes(in) finally in.close()

Answer 3

与 Eastsun 的回答类似……我一开始是作为评论，但最终变得有点长了！

我警告不要使用 Stream，如果持有对 head 元素的引用，那么流很容易消耗大量内存。

鉴于您只会读取文件一次，那么 Iterator 是一个更好的选择：

def inputStreamToByteArray(is: InputStream): Array[Byte] =
  Iterator continually is.read takeWhile (-1 !=) map (_.toByte) toArray

Answer 4

import scala.tools.nsc.io.Streamable
Streamable.bytes(is)

不记得那是多久以前的事：可能以天为单位。回到2.8，它更像是

new Streamable.Bytes { def inputStream() = is } toByteArray

Answer 5

使用 Scala IO，这应该可以工作：

def inputStreamToByteArray(is: InputStream): Array[Byte] = 
   Resource.fromInputStream(in).byteArray

Answer 6

使用 better-files，您可以简单地执行 is.bytes

Answer 7

Source.fromInputStream(is).map(_.toByte).toArray

Answer 8

基于流和 ByteArraOutputStream 的解决方案的缓冲版本如何最大限度地减少最终数组增长的样板？

val EOF: Int = -1

def readBytes(is: InputStream, bufferSize: Int): Array[Byte] = {
  val buf = Array.ofDim[Byte](bufferSize)
  val out = new ByteArrayOutputStream(bufferSize)

  Stream.continually(is.read(buf)) takeWhile { _ != EOF } foreach { n =>
    out.write(buf, 0, n)
  }

  out.toByteArray
}

Answer 9

这是使用 scalaz-stream 的方法：

import scalaz.concurrent.Task
import scalaz.stream._
import scodec.bits.ByteVector

def allBytesR(is: InputStream): Process[Task, ByteVector] =
  io.chunkR(is).evalMap(_(4096)).reduce(_ ++ _).lastOr(ByteVector.empty)

Answer 10

从 JDK 9 开始：

is.readAllBytes()

Answer 11

我们可以使用 Google API ByteStreams

com.google.common.io.ByteStreams

将流传递给 ByteStreams.toByteArray 方法进行转换

ByteStreams.toByteArray(stream)

Answer 12

def inputStreamToByteArray(is: InputStream): Array[Byte] = {
    val buf = ListBuffer[Byte]()
    var b = is.read()
    while (b != -1) {
        buf.append(b.byteValue)
        b = is.read()
    }
    buf.toArray
}