Add Mutex.holdsLock

Author: Francesco Vasco

kotlinx-coroutines-core/src/main/kotlin/kotlinx/coroutines/experimental/sync/Mutex.kt

@@ -91,6 +91,13 @@ public interface Mutex {
     public fun <R> registerSelectLock(select: SelectInstance<R>, owner: Any?, block: suspend () -> R)
 
     /**
+     * Checks mutex locked by owner
+     *
+     * @return `true` on mutex lock by owner, `false` if not locker or it is locked by different owner
+     */
+    public fun holdsLock(owner: Any): Boolean
+
+        /**
      * Unlocks this mutex. Throws [IllegalStateException] if invoked on a mutex that is not locked.
      *
      * @param owner Optional owner token for debugging. When `owner` is specified (non-null value) and this mutex
@@ -325,6 +332,15 @@ internal class MutexImpl(locked: Boolean) : Mutex {
         }
     }
 
+    public override fun holdsLock(owner: Any) =
+            _state.value.let { state ->
+                when (state) {
+                    is Empty -> state.locked === owner
+                    is LockedQueue -> state.owner === owner
+                    else -> false
+                }
+            }
+
     public override fun unlock(owner: Any?) {
         _state.loop { state ->
             when (state) {

kotlinx-coroutines-core/src/test/kotlin/kotlinx/coroutines/experimental/sync/MutexTest.kt

@@ -107,4 +107,37 @@ class MutexTest : TestBase() {
         mutex.unlock() // should not produce StackOverflowError
         assertThat(done, IsEqual(waiters))
     }
+
+    @Test
+    fun holdLock() = runBlocking {
+        val mutex = Mutex()
+        val firstOwner = Any()
+        val secondOwner = Any()
+
+        // no lock
+        assertFalse(mutex.holdsLock(firstOwner))
+        assertFalse(mutex.holdsLock(secondOwner))
+
+        // owner firstOwner
+        mutex.lock(firstOwner)
+        val secondLockJob = launch(CommonPool) {
+            mutex.lock(secondOwner)
+        }
+
+        assertTrue(mutex.holdsLock(firstOwner))
+        assertFalse(mutex.holdsLock(secondOwner))
+
+        // owner secondOwner
+        mutex.unlock(firstOwner)
+        secondLockJob.join()
+
+        assertFalse(mutex.holdsLock(firstOwner))
+        assertTrue(mutex.holdsLock(secondOwner))
+
+        mutex.unlock(secondOwner)
+
+        // no lock
+        assertFalse(mutex.holdsLock(firstOwner))
+        assertFalse(mutex.holdsLock(secondOwner))
+    }
 }