com.fireflysource.common.coroutine.ChannelExtension.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of firefly-common Show documentation

There is a newer version: 5.0.2

package com.fireflysource.common.coroutine

import kotlinx.coroutines.channels.Channel
import java.util.concurrent.atomic.AtomicBoolean

inline fun  Channel.pollAll(crossinline block: (T) -> Unit) {
    while (true) {
        val message = this.poll()
        if (message != null) block(message) else break
    }
}

class Signal {
    private val channel: Channel = Channel(Channel.UNLIMITED)
    private val notified = AtomicBoolean(false)

    suspend fun wait(): T = channel.receive()

    fun notify(e: T) {
        if (notified.compareAndSet(false, true)) {
            channel.offer(e)
        }
    }

    fun reset() {
        notified.set(false)
        channel.pollAll { }
    }
}