我如何转换流<Byte>到输入流

Question

I have java.util.stream.Stream<Byte> in Java/Kotlin(jvm).

How can I convert java.util.stream.Stream<Byte> to java.io.InputStream without keeping each buffer in memory.

I can do this conversion by

val inputStream = ByteArrayInputStream(byteStream.collect(Collectors.toList()).toByteArray())

But it will create a large byte[]/ByteArray in memory if the stream is very long.

Answer 1

You could implement it with something like this:

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
}

You would probably need to override more methods to improve performance or provide more functionality, like for example closing the stream.

Also, note that I didn't test this code thoroughly.