code/KustoCopyConsole/Runner/RunnerBase.cs

using Azure.Core;
using KustoCopyConsole.Entity.State;
using KustoCopyConsole.JobParameter;
using KustoCopyConsole.Kusto;
using KustoCopyConsole.Storage;
using System.Diagnostics;

namespace KustoCopyConsole.Runner
{
    internal class RunnerBase
    {
        private static readonly TraceSource _traceSource = new(TraceConstants.TRACE_SOURCE);

        private readonly TimeSpan _wakePeriod;

        public RunnerBase(
            MainJobParameterization parameterization,
            TokenCredential credential,
            RowItemGateway rowItemGateway,
            DbClientFactory dbClientFactory,
            IStagingBlobUriProvider stagingBlobUriProvider,
            TimeSpan wakePeriod)
        {
            Parameterization = parameterization;
            Credential = credential;
            RowItemGateway = rowItemGateway;
            DbClientFactory = dbClientFactory;
            StagingBlobUriProvider = stagingBlobUriProvider;
            _wakePeriod = wakePeriod;
        }

        protected MainJobParameterization Parameterization { get; }

        protected TokenCredential Credential { get; }

        protected RowItemGateway RowItemGateway { get; }

        protected DbClientFactory DbClientFactory { get; }

        protected IStagingBlobUriProvider StagingBlobUriProvider { get; }

        protected bool AllActivitiesCompleted()
        {
            return !RowItemGateway.InMemoryCache.ActivityMap
                .Values
                .Where(a => a.RowItem.State == ActivityState.Active)
                .Any();
        }

        protected async Task SleepAsync(CancellationToken ct)
        {
            await Task.Delay(_wakePeriod, ct);
        }

        protected void TraceWarning(string text)
        {
            _traceSource.TraceEvent(TraceEventType.Warning, 0, text);
        }
    }
}