LockedObj

LockedObj.kt

import kotlinx.coroutines.NonCancellable
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock
import kotlinx.coroutines.withContext

class LockedObj<T>(
    /**
     * Use it directly only in [withLock]
     */
    var value: T,
) {

    private val lock = Mutex()

    suspend fun set(newValue: T) {
        lock.withLock {
            value = newValue
        }
    }

    suspend fun get(): T =
        lock.withLock {
            value
        }

    suspend fun update(action: (T) -> T) {
        lock.withLock {
            value = action(value)
        }
    }

    suspend fun withLock(action: LockedObj<T>.() -> Unit) {
        lock.withLock {
            action()
        }
    }

    suspend fun withNonCancellableLock(action: LockedObj<T>.() -> Unit) {
        withContext(NonCancellable) {
            lock.withLock {
                action()
            }
        }
    }
}