commonMain.korlibs.io.stream.sync.DequeSyncStream.kt Maven / Gradle / Ivy

Go to download
@file:Suppress("PackageDirectoryMismatch")

package korlibs.io.stream

import korlibs.memory.*
import kotlinx.atomicfu.locks.reentrantLock
import kotlinx.atomicfu.locks.withLock

fun DequeSyncStream(): SyncStream = object : SyncStreamBase() {
    var closed = false
    val deque: SimpleBytesDeque = SimpleBytesDeque()
    val lock = reentrantLock()

    override val separateReadWrite: Boolean get() = true
    override val seekable get() = false

    override fun read(position: Long, buffer: ByteArray, offset: Int, len: Int): Int = lock.withLock {
        if (position != deque.read) error("Invalid DequeSyncStreamBase.position for reading $position != ${deque.read}")
        val res = deque.read(buffer, offset, len)
        if (len > 0 && res <= 0 && closed) return@withLock -1
        res
    }

    override fun write(position: Long, buffer: ByteArray, offset: Int, len: Int): Unit = lock.withLock {
        if (position != deque.written) error("Invalid DequeSyncStreamBase.position for writing $position != ${deque.written}")
        deque.write(buffer, offset, len)
    }

    override var length: Long
        get() = deque.written
        set(value) {}

    override fun close() = lock.withLock {
        closed = true
        deque.clear()
    }
}.toSyncStream()

//class DequeSyncStream : AutoCloseable, SyncInputStream, SyncOutputStream, SyncPositionStream, SyncLengthStream {
//    val deque = SimpleBytesDeque()
//    override var position: Long = 0L
//    override var length: Long = 0L
//
//    override fun write(buffer: ByteArray, offset: Int, len: Int) {
//        deque.write(buffer, offset, len)
//        length += len
//    }
//
//    override fun read(buffer: ByteArray, offset: Int, len: Int): Int {
//        return deque.read(buffer, offset, len).also {
//            position += it
//        }
//    }
//
//    override fun close() {
//        deque.clear()
//    }
//}