public static Task<IDisposable> Lookup(RecursiveAsyncLock mutex)
{
Tuple<int, Task<IDisposable>> value;
if (OwnedLocks.TryGetValue(mutex, out value))
return value.Item2;
return null;
}
public static void Add(RecursiveAsyncLock mutex, Task<IDisposable> key)
{
Tuple<int, Task<IDisposable>> value;
if (!OwnedLocks.TryGetValue(mutex, out value))
value = Tuple.Create(0, key);
OwnedLocks = OwnedLocks.SetItem(mutex, Tuple.Create(value.Item1 + 1, value.Item2));
}
public void Dispose()
{
if (_mutex == null)
{
return;
}
AsyncLockTracker.Remove(_mutex);
_mutex = null;
}
public static void Add(RecursiveAsyncLock mutex, Task <IDisposable> key)
{
Tuple <int, Task <IDisposable> > value;
if (!OwnedLocks.TryGetValue(mutex, out value))
{
value = Tuple.Create(0, key);
}
OwnedLocks = OwnedLocks.SetItem(mutex, Tuple.Create(value.Item1 + 1, value.Item2));
}
public static Task <IDisposable> Lookup(RecursiveAsyncLock mutex)
{
Tuple <int, Task <IDisposable> > value;
if (OwnedLocks.TryGetValue(mutex, out value))
{
return(value.Item2);
}
return(null);
}
public static void Remove(RecursiveAsyncLock mutex)
{
var value = OwnedLocks[mutex];
if (value.Item1 == 1)
{
OwnedLocks = OwnedLocks.Remove(mutex);
value.Item2.Result.Dispose();
}
else
{
OwnedLocks = OwnedLocks.SetItem(mutex, Tuple.Create(value.Item1 - 1, value.Item2));
}
}
public static void Remove(RecursiveAsyncLock mutex)
{
var value = OwnedLocks[mutex];
if (value.Item1 == 1)
{
OwnedLocks = OwnedLocks.Remove(mutex);
value.Item2.Result.Dispose();
}
else
{
OwnedLocks = OwnedLocks.SetItem(mutex, Tuple.Create(value.Item1 - 1, value.Item2));
}
}
public async Task RecursiveAsyncLock_DoesNotPermitIndependentWaits()
{
var mutex0 = new RecursiveAsyncLock();
using (await mutex0.LockAsync())
{
var mutex = new RecursiveAsyncLock();
Func <TaskCompletionSource, TaskCompletionSource, Task> taker = async(ready, finish) =>
{
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
using (await mutex.LockAsync())
{
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
ready.TrySetResult();
await finish.Task;
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
}
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
};
var ready1 = new TaskCompletionSource();
var finish1 = new TaskCompletionSource();
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
var task1 = taker(ready1, finish1);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
await Task.WhenAny(ready1.Task, task1);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
var ready2 = new TaskCompletionSource();
var finish2 = new TaskCompletionSource();
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
var task2 = taker(ready2, finish2);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
Assert.IsFalse(ready2.Task.Wait(1000));
finish1.SetResult();
await Task.WhenAny(ready2.Task, task2);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
finish2.SetResult();
await Task.WhenAll(task1, task2);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
}
}
public async Task RecursiveAsyncLock_PermitsRecursiveWaits()
{
var mutex = new RecursiveAsyncLock();
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
var task1 = mutex.LockAsync();
var key1 = await task1;
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
var task2 = mutex.LockAsync();
var key2 = await task2;
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
key2.Dispose();
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
key1.Dispose();
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
}
public static bool Contains(RecursiveAsyncLock mutex)
{
return OwnedLocks.ContainsKey(mutex);
}
public async Task RecursiveAsyncLock_DoesNotPermitIndependentWaits()
{
var mutex0 = new RecursiveAsyncLock();
using (await mutex0.LockAsync())
{
var mutex = new RecursiveAsyncLock();
Func<TaskCompletionSource, TaskCompletionSource, Task> taker = async (ready, finish) =>
{
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
using (await mutex.LockAsync())
{
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
ready.TrySetResult();
await finish.Task;
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
}
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
};
var ready1 = new TaskCompletionSource();
var finish1 = new TaskCompletionSource();
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
var task1 = taker(ready1, finish1);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
await Task.WhenAny(ready1.Task, task1);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
var ready2 = new TaskCompletionSource();
var finish2 = new TaskCompletionSource();
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
var task2 = taker(ready2, finish2);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
Assert.IsFalse(ready2.Task.Wait(1000));
finish1.SetResult();
await Task.WhenAny(ready2.Task, task2);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
finish2.SetResult();
await Task.WhenAll(task1, task2);
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
}
}
public async Task RecursiveAsyncLock_PermitsRecursiveWaits()
{
var mutex = new RecursiveAsyncLock();
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
var task1 = mutex.LockAsync();
var key1 = await task1;
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
var task2 = mutex.LockAsync();
var key2 = await task2;
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
key2.Dispose();
Assert.IsTrue(AsyncLockTracker.Contains(mutex));
key1.Dispose();
Assert.IsFalse(AsyncLockTracker.Contains(mutex));
}
public void Dispose()
{
if (_mutex == null)
return;
AsyncLockTracker.Remove(_mutex);
_mutex = null;
}
public KeyDisposable(Task<IDisposable> keyTask, RecursiveAsyncLock mutex)
{
_mutex = mutex;
AsyncLockTracker.Add(mutex, keyTask);
}
public KeyDisposable(Task <IDisposable> keyTask, RecursiveAsyncLock mutex)
{
_mutex = mutex;
AsyncLockTracker.Add(mutex, keyTask);
}
public static bool Contains(RecursiveAsyncLock mutex)
{
return(OwnedLocks.ContainsKey(mutex));
}
public RecursiveLockAwaitable(Task<IDisposable> key, RecursiveAsyncLock mutex)
{
_key = key;
_awaiter = key.GetAwaiter();
_mutex = mutex;
}
public RecursiveLockAwaitable(Task <IDisposable> key, RecursiveAsyncLock mutex)
{
_key = key;
_awaiter = key.GetAwaiter();
_mutex = mutex;
}