ru.inforion.lab403.common.extensions.threading.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-extensions Show documentation

Kotlin extension methods and function for different Java libraries

There is a newer version: 0.3.5

package ru.inforion.lab403.common.extensions

import kotlinx.coroutines.*
import java.util.concurrent.TimeUnit
import java.util.concurrent.locks.ReentrantLock
import kotlin.concurrent.withLock
import kotlin.coroutines.CoroutineContext
import kotlin.coroutines.EmptyCoroutineContext


fun  async(
    context: CoroutineContext = EmptyCoroutineContext,
    scope: CoroutineScope = GlobalScope,
    block: suspend CoroutineScope.() -> T
) = scope.async(context, block = block)

fun launch(
    context: CoroutineContext = EmptyCoroutineContext,
    scope: CoroutineScope = GlobalScope,
    block: suspend CoroutineScope.() -> Unit
) = scope.launch(context, block = block)

fun  Collection>.wait() = runBlocking { awaitAll() }

fun Collection.wait() = runBlocking { joinAll() }

fun  Deferred.wait() = runBlocking { await() }

fun Job.wait() = runBlocking { join() }

class BlockingValue  {
    private lateinit var value: T

    private val lock = ReentrantLock()
    private val condition = lock.newCondition()

    private var hasValue = false

    private fun setWithSignal(newValue: T) {
        value = newValue
        hasValue = true
        condition.signal()
    }

    private fun valueOrNull() = if (hasValue) value else null

    fun hasValue() = lock.withLock { hasValue }

    /**
     * Try to get value without waiting, returns false if value already set else true
     *
     * i.e. NON-BLOCKING get-method
     *
     * @param newValue value to set
     */
    fun offer(newValue: T) = lock.withLock {
        if (hasValue) false else {
            setWithSignal(newValue)
            true
        }
    }

    /**
     * Try to get value infinitely
     *
     * i.e. BLOCKING get-method
     */
    fun take() = lock.withLock {
        if (!hasValue) condition.await()
        value
    }

    /**
     * Try to get value during specified timeout
     *
     * i.e. TIMEOUT-BLOCKING get-method
     *
     * @param millis timeout in milliseconds
     */
    fun poll(millis: Long) = lock.withLock {
        if (!hasValue) {
            condition.await(millis, TimeUnit.MILLISECONDS)
            valueOrNull()
        } else value
    }

    /**
     * Try to get value without waiting, returns null if value not set
     *
     * i.e. NON-BLOCKING get-method
     */
    fun poll() = lock.withLock { valueOrNull() }
}