BufferedBytesToStringProducer.cs

Go to the documentation of this file.

using System.Text;
 
namespace AnalyzeRe.LargeDataUpload
{
    public class BufferedBytesToStringProducer :
        ProducerConsumerBuffer<BufferedBytesToStringProducer.StringWithByteCount>
    {
        public static readonly Encoding DEFAULT_ENCODING = Encoding.UTF8;
 
        public Encoding Encoding { get; }
 
        public struct StringWithByteCount
        {
            public string String;
 
            public int ByteLength;
        }
 
        private readonly IProducerConsumerBuffer<BufferedBytes> _source;
 
        public BufferedBytesToStringProducer(
            IProducerConsumerBuffer<BufferedBytes> source,
            Encoding encoding = null, int max_queue_size = 3)
            : base(max_queue_size)
        {
            Encoding = encoding ?? DEFAULT_ENCODING;
            _source = source;
        }
 
        protected override void OnStart()
        {
            base.OnStart();
            // Start the source producer if it hasn't already started.
            if (!_source.IsRunning)
                _source.Start(Cancellation.Token);
        }
 
        protected override bool IsProducerFinished() => !_source.CanTake;
 
        protected override bool TryProduceNext(out StringWithByteCount next)
        {
            next = new StringWithByteCount();
 
            BufferedBytes buffer = _source.TryTake(out bool success, Cancellation.Token);
            if (!success) return false;
 
            next.String = Encoding.GetString(buffer.Bytes, 0, buffer.LengthFilled);
            next.ByteLength = buffer.LengthFilled;
            return true;
        }
    }
}