public void Init()
{
_cts = new CancellationTokenSource(TimeSpan.FromSeconds(15));
_loggerFactory = Substitute.For <ILoggerFactory>();
_logger = Substitute.For <ILogger <QuidjiboServer> >();
_loggerFactory.CreateLogger <QuidjiboServer>().Returns(_logger);
_quidjiboConfiguration = Substitute.For <IQuidjiboConfiguration>();
_workProviderFactory = Substitute.For <IWorkProviderFactory>();
_workProvider = Substitute.For <IWorkProvider>();
_workProviderFactory.CreateAsync(Arg.Any <string>(), Arg.Any <CancellationToken>()).Returns(Task.FromResult(_workProvider));
_workProviderFactory.PollingInterval.Returns(1);
_scheduleProviderFactory = Substitute.For <IScheduleProviderFactory>();
_scheduleProvider = Substitute.For <IScheduleProvider>();
_scheduleProviderFactory.CreateAsync(Arg.Any <string[]>(), Arg.Any <CancellationToken>()).Returns(Task.FromResult(_scheduleProvider));
_scheduleProviderFactory.PollingInterval.Returns(1);
_progressProviderFactory = Substitute.For <IProgressProviderFactory>();
_progressProvider = Substitute.For <IProgressProvider>();
_progressProviderFactory.CreateAsync(Arg.Any <string[]>(), Arg.Any <CancellationToken>()).Returns(Task.FromResult(_progressProvider));
Substitute.For <IWorkDispatcher>();
Substitute.For <IPayloadSerializer>();
_cronProvider = Substitute.For <ICronProvider>();
_pipeline = Substitute.For <IQuidjiboPipeline>();
_sut = new QuidjiboServer(_loggerFactory, _quidjiboConfiguration, _workProviderFactory, _scheduleProviderFactory, _progressProviderFactory, _cronProvider, _pipeline);
}
private async Task InvokePipelineAsync(IWorkProvider provider, WorkItem item)
{
using (var linkedTokenSource = CancellationTokenSource.CreateLinkedTokenSource(_cts.Token))
{
var progress = new QuidjiboProgress();
progress.ProgressChanged += async(sender, tracker) =>
{
var progressProvider = await _progressProviderFactory.CreateAsync(item.Queue, _cts.Token);
var progressItem = new ProgressItem
{
Id = Guid.NewGuid(),
CorrelationId = item.CorrelationId,
RecordedOn = DateTime.UtcNow,
Name = item.Name,
Queue = item.Queue,
Note = tracker.Text,
Value = tracker.Value,
WorkId = item.Id
};
await progressProvider.ReportAsync(progressItem, _cts.Token);
};
var renewTask = RenewAsync(provider, item, linkedTokenSource.Token);
var context = new QuidjiboContext
{
Item = item,
WorkProvider = provider,
Progress = progress,
State = new PipelineState()
};
try
{
await _quidjiboPipeline.StartAsync(context, linkedTokenSource.Token);
if (context.State.Success)
{
await provider.CompleteAsync(item, linkedTokenSource.Token);
_logger.LogDebug("Completed : {0}", item.Id);
}
else
{
await provider.FaultAsync(item, linkedTokenSource.Token);
_logger.LogError("Faulted : {0}", item.Id);
}
}
catch (Exception exception)
{
_logger.LogError("Faulted : {0}, {1}", item.Id, exception);
await provider.FaultAsync(item, linkedTokenSource.Token);
}
finally
{
_logger.LogDebug("Release : {0}", item.Id);
linkedTokenSource.Cancel();
await renewTask;
await _quidjiboPipeline.EndAsync(context);
}
}
}