jvmMain.kotlinx.io.streams.Input.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlinx-io-jvm Show documentation

IO support libraries for Kotlin

There is a newer version: 0.1.16

package kotlinx.io.streams

import kotlinx.io.core.*
import kotlinx.io.pool.*
import java.io.*

internal class InputStreamAsInput(
    private val stream: InputStream,
    pool: ObjectPool
) : AbstractInput(pool = pool) {

    override fun fill(): IoBuffer? {
        val buffer = ByteArrayPool.borrow()
        try {
            val rc = stream.read(buffer, 0, ByteArrayPoolBufferSize - IoBuffer.ReservedSize)
            val result = when {
                rc >= 0 -> pool.borrow().also {
                    it.reserveEndGap(IoBuffer.ReservedSize)
                    it.writeFully(buffer, 0, rc)
                }
                else -> null
            }

            ByteArrayPool.recycle(buffer)
            return result
        } catch (t: Throwable) {
            ByteArrayPool.recycle(buffer)
            throw t
        }
    }

    override fun closeSource() {
        stream.close()
    }
}

fun InputStream.asInput(pool: ObjectPool = IoBuffer.Pool): Input = InputStreamAsInput(this, pool)