Kotlin Coroutine Callback

OnCompleteCallback.kt

import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.channels.Channel
import kotlinx.coroutines.launch


interface OnCompleteCallback {
    fun complete()

    /**
     * Waits for complete callback
     */
    suspend fun await()
}


class DefaultOnCompleteCallback : OnCompleteCallback {
    private val channel = Channel<Unit>(1)

    override fun complete() {
        CoroutineScope(Dispatchers.Main).launch {
            channel.send(Unit)
        }
    }

    override suspend fun await() {
        channel.receive()
    }
}