Reentrance checking coroutine Mutex

gistfile1.txt

suspend fun <ReturnT> Mutex.checkedWithLock(
  block: suspend () -> ReturnT
): ReturnT {
  val element = LockedMutexesElement(this).key
  check (currentCoroutineContext()[element] == null) {
    "Reentered Mutex"
  }
  
  return withContext(element) { withLock { block() } }
}

private data class LockedMutexesElement(val mutex: Mutex) :
  CoroutineContext.Element, CoroutineContext.Key<LockedMutexesElement> {
  override val key = this
}