SemaphoreSlimContext.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
namespace Tgstation.Server.Host.Utils
{
    sealed class SemaphoreSlimContext : IDisposable
    {
        public static async ValueTask<SemaphoreSlimContext> Lock(SemaphoreSlim semaphore, CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(semaphore);
            await semaphore.WaitAsync(cancellationToken);
            return new SemaphoreSlimContext(semaphore);
        }
 
        public static SemaphoreSlimContext TryLock(SemaphoreSlim semaphore, out bool locked)
        {
            ArgumentNullException.ThrowIfNull(semaphore);
            locked = semaphore.Wait(TimeSpan.Zero);
            return locked
                ? new SemaphoreSlimContext(semaphore)
                : null;
        }
 
        readonly SemaphoreSlim lockedSemaphore;
 
        SemaphoreSlimContext(SemaphoreSlim lockedSemaphore)
        {
            this.lockedSemaphore = lockedSemaphore;
        }
 
        public void Dispose() => lockedSemaphore.Release();
    }
}