55 lines
1.2 KiB
C#
55 lines
1.2 KiB
C#
using System.Collections.Concurrent;
|
|
using System.Diagnostics.CodeAnalysis;
|
|
|
|
namespace MesETL.App.Services;
|
|
|
|
/// <summary>
|
|
/// 数据队列
|
|
/// </summary>
|
|
public class DataRecordQueue : IDisposable
|
|
{
|
|
private readonly BlockingCollection<DataRecord> _queue;
|
|
|
|
public int Count => _queue.Count;
|
|
public bool IsCompleted => _queue.IsCompleted;
|
|
public bool IsAddingCompleted => _queue.IsAddingCompleted;
|
|
|
|
public event Action? OnRecordWrite;
|
|
public event Action? OnRecordRead;
|
|
|
|
public DataRecordQueue() : this(500_000) // 默认容量最大500K
|
|
{
|
|
}
|
|
|
|
public DataRecordQueue(int boundedCapacity)
|
|
{
|
|
_queue = new BlockingCollection<DataRecord>(boundedCapacity);
|
|
}
|
|
|
|
public bool TryDequeue([MaybeNullWhen(false)] out DataRecord record)
|
|
{
|
|
if (_queue.TryTake(out record))
|
|
{
|
|
OnRecordRead?.Invoke();
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public DataRecord Dequeue() => _queue.Take();
|
|
|
|
public void CompleteAdding() => _queue.CompleteAdding();
|
|
|
|
public void Enqueue(DataRecord record)
|
|
{
|
|
_queue.Add(record);
|
|
OnRecordWrite?.Invoke();
|
|
}
|
|
|
|
|
|
public void Dispose()
|
|
{
|
|
_queue.Dispose();
|
|
}
|
|
} |