Skip to content

Commit

Permalink
feat: Reduce allocations when using AsyncLock (+ remove all dependenc…
Browse files Browse the repository at this point in the history
…ies)
  • Loading branch information
dr1rrb committed Sep 30, 2024
1 parent 7e7f31d commit b2ea548
Showing 1 changed file with 25 additions and 14 deletions.
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
#nullable enable
// ******************************************************************
// Copyright � 2015-2018 Uno Platform Inc. All rights reserved.
//
Expand All @@ -18,27 +19,37 @@
using System.Threading;
using System.Threading.Tasks;

using Uno.Disposables;
namespace Uno.Threading;

namespace Uno.Threading
/// <summary>
/// An asynchronous lock, that can be used in conjuction with C# async/await
/// </summary>
internal sealed class AsyncLock
{
private readonly SemaphoreSlim _semaphore = new(1, 1);
private ulong _handleId;

/// <summary>
/// An asynchronous lock, that can be used in conjuction with C# async/await
/// Acquires the lock, then provides a disposable to release it.
/// WARNING: This DOES NOT support reentrancy.
/// </summary>
internal sealed class AsyncLock
/// <param name="ct">A cancellation token to cancel the lock</param>
/// <returns>An IDisposable instance that allows the release of the lock.</returns>
public async ValueTask<Handle> LockAsync(CancellationToken ct)
{
private readonly SemaphoreSlim _semaphore = new SemaphoreSlim(1, 1);
await _semaphore.WaitAsync(ct);

/// <summary>
/// Acquires the lock, then provides a disposable to release it.
/// </summary>
/// <param name="ct">A cancellation token to cancel the lock</param>
/// <returns>An IDisposable instance that allows the release of the lock.</returns>
public async Task<IDisposable> LockAsync(CancellationToken ct)
{
await _semaphore.WaitAsync(ct);
return new Handle(this, _handleId);
}

return Disposable.Create(() => _semaphore.Release());
public record struct Handle(AsyncLock Lock, ulong Id) : IDisposable
{
public void Dispose()
{
if (Interlocked.CompareExchange(ref Lock._handleId, Id + 1, Id) == Id) // This avoids (concurrent) double dispose / release
{
Lock._semaphore.Release();
}
}
}
}

0 comments on commit b2ea548

Please sign in to comment.