io.wavebeans.execution.pod.AbstractStreamOutputPod.kt Maven / Gradle / Ivy

Go to download
package io.wavebeans.execution.pod

import io.wavebeans.lib.AnyBean
import io.wavebeans.lib.BeanParams
import io.wavebeans.lib.NoParams
import io.wavebeans.lib.Sample
import io.wavebeans.lib.io.StreamOutput
import io.wavebeans.lib.io.Writer

abstract class AbstractStreamOutputPod : StreamOutput, TickPod {

    abstract val bean: StreamOutput

    abstract val sampleRate: Float

    @Volatile
    private var isFinished = false

    private var writer: Writer? = null

    override fun start() {
        writer = bean.writer(sampleRate)
    }

    override fun tick(): Boolean {
        check(writer != null) { "Pod should be started first" }
        return if (!isFinished) {
            val isSomethingLeft = writer!!.write()
            if (!isSomethingLeft) isFinished = true
            true
        } else {
            false
        }
    }

    override fun close() {
        isFinished = true
        writer?.close()
    }

    override fun isFinished(): Boolean = isFinished

    override fun inputs(): List = listOf(bean)

    override fun iteratorStart(sampleRate: Float, partitionIdx: Int): Long = throw UnsupportedOperationException("You can't read from this pod")

    override fun iteratorNext(iteratorKey: Long, buckets: Int): List? = throw UnsupportedOperationException("You can't read from this pod")

    override fun writer(sampleRate: Float): Writer = throw UnsupportedOperationException("Not required by pod")

    override fun desiredSampleRate(): Float? = throw UnsupportedOperationException("Not required by pod")

    override val parameters: BeanParams
        get() = NoParams()

    override fun toString(): String = "[$podKey]${this::class.simpleName}"
}