MaxCountProcessor.cs

using System.Threading.Tasks;
using Waher.Persistence;
 
namespace Waher.Script.Persistence.SQL.Processors
{
    public class MaxCountProcessor : IProcessor<object>
    {
        private readonly IProcessor<object> processor;
        private readonly bool isAsynchronous;
        private int count;
 
        public MaxCountProcessor(IProcessor<object> Processor, int Count)
        {
            this.processor = Processor;
            this.count = Count;
            this.isAsynchronous = Processor.IsAsynchronous;
        }
 
        public bool IsAsynchronous => this.isAsynchronous;
 
        public bool Process(object Object)
        {
            if (this.count <= 0)
                return false;
 
            this.count--;
            return this.processor.Process(Object);
        }
 
        public Task<bool> ProcessAsync(object Object)
        {
            if (this.count <= 0)
                return Task.FromResult(false);
 
            this.count--;
            return this.processor.ProcessAsync(Object);
        }
 
        public bool Flush() => this.processor.Flush();
 
        public Task<bool> FlushAsync() => this.processor.FlushAsync();
    }
}