chore: continue porting code and adding test coverage
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
namespace StevesBot.Worker.Threading;
|
||||
|
||||
internal sealed class AsyncLock : IAsyncLock
|
||||
{
|
||||
private readonly SemaphoreSlim _semaphore = new(1, 1);
|
||||
private bool _disposed;
|
||||
|
||||
public async Task<IDisposable> LockAsync(CancellationToken cancellationToken = default)
|
||||
{
|
||||
await _semaphore.WaitAsync(cancellationToken);
|
||||
return new LockReleaser(_semaphore);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
if (_disposed)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_semaphore.Dispose();
|
||||
_disposed = true;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,6 @@
|
||||
namespace StevesBot.Worker.Threading;
|
||||
|
||||
internal interface IAsyncLock : IDisposable
|
||||
{
|
||||
Task<IDisposable> LockAsync(CancellationToken cancellationToken = default);
|
||||
}
|
||||
@@ -0,0 +1,23 @@
|
||||
namespace StevesBot.Worker.Threading;
|
||||
|
||||
internal sealed class LockReleaser : IDisposable
|
||||
{
|
||||
private readonly SemaphoreSlim _semaphore;
|
||||
private bool _released;
|
||||
|
||||
public LockReleaser(SemaphoreSlim semaphore)
|
||||
{
|
||||
_semaphore = semaphore;
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
if (_released)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_semaphore.Release();
|
||||
_released = true;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user