mirror of
https://github.com/azaion/satellite-provider.git
synced 2026-04-22 10:46:38 +00:00
40 lines
1.1 KiB
C#
40 lines
1.1 KiB
C#
using System.Threading.Channels;
|
|
using SatelliteProvider.Common.DTO;
|
|
using SatelliteProvider.Common.Interfaces;
|
|
|
|
namespace SatelliteProvider.Services;
|
|
|
|
public class RegionRequestQueue : IRegionRequestQueue
|
|
{
|
|
private readonly Channel<RegionRequest> _queue;
|
|
|
|
public RegionRequestQueue(int capacity)
|
|
{
|
|
var options = new BoundedChannelOptions(capacity)
|
|
{
|
|
FullMode = BoundedChannelFullMode.Wait
|
|
};
|
|
_queue = Channel.CreateBounded<RegionRequest>(options);
|
|
}
|
|
|
|
public async ValueTask EnqueueAsync(RegionRequest request, CancellationToken cancellationToken = default)
|
|
{
|
|
await _queue.Writer.WriteAsync(request, cancellationToken);
|
|
}
|
|
|
|
public async ValueTask<RegionRequest?> DequeueAsync(CancellationToken cancellationToken = default)
|
|
{
|
|
if (await _queue.Reader.WaitToReadAsync(cancellationToken))
|
|
{
|
|
if (_queue.Reader.TryRead(out var request))
|
|
{
|
|
return request;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public int Count => _queue.Reader.Count;
|
|
}
|
|
|