SemaphoreExtensions.cs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

namespace Lilith.Core.Extensions;

public static class SemaphoreExtensions
{
    public static IDisposable EnterOrThrow(this SemaphoreSlim gate, string message)
    {
        if (!gate.Wait(0)) throw new InvalidOperationException(message);
        return new SemaphoreLease(gate);
    }

    public static async Task<IDisposable> EnterAsync(this SemaphoreSlim gate, CancellationToken ct = default)
    {
        await gate.WaitAsync(ct);
        return new SemaphoreLease(gate);
    }
}