如何将流<Byte>转换为输入流

Question

我有java.util.stream.Stream<Byte>在Java/Kotlin(jvm)。

如何将java.util.stream.Stream<Byte>转换为java.io.InputStream而不保留每个缓冲区在内存中

我可以通过

进行转换

fun convertToInputStream(byteStream: Stream<Byte>) = ByteArrayInputStream(byteStream.collect(Collectors.toList()).toByteArray())

在我的例子中，如果我使用一个大的限制，我将得到一个OutOfMemoryError，因为它在内存中创建了一个大字节[]

import java.io.ByteArrayInputStream
import java.io.File
import java.util.stream.Collectors
import java.util.stream.IntStream
import java.util.stream.Stream
fun convertToInputStream(byteStream: Stream<Byte>) = ByteArrayInputStream(byteStream.collect(Collectors.toList()).toByteArray())

fun main() {
val file = File("/tmp/data.txt")
val inputStream = convertToInputStream(
IntStream.iterate(1) { 65 + it % 3 }.limit(1000000000).mapToObj { it.toByte() }
)
inputStream.copyTo(file.outputStream())
}

Answer 1

你可以这样实现它:

fun main() {
val stream = Stream.of<Byte>(10, 15, -50, 20, 50)
val buf = ByteArrayOutputStream()
stream.asInputStream().transferTo(buf)
val bytes = buf.toByteArray() // [10, 15, -50, 20, 50]
}
fun Stream<Byte>.asInputStream() = object : InputStream() {
private val iter = iterator()
override fun read() = if (iter.hasNext()) iter.next().toUByte().toInt() else -1
}

你可能需要重写更多的方法来提高性能或提供更多的功能，比如关闭流。

另外，请注意，我没有彻底测试这段代码。