public async Task ShouldRunAllAsyncTasks(int maxDegreeOfParallelism, int taskCount)
{
// Setup
var testAsyncTaskStatuses = new Dictionary <int, bool>();
var asyncTaskRunner = new AsyncTaskRunner <TestAsyncTask>(maxDegreeOfParallelism, CancellationToken.None);
for (int i = 0; i < taskCount; i++)
{
testAsyncTaskStatuses.Add(i, false);
asyncTaskRunner.AddTask(new TestAsyncTask(i, task =>
{
lock (_lock)
{
testAsyncTaskStatuses[task.TaskId] = true;
}
}));
}
// Act
await asyncTaskRunner.WaitForCompletion();
// Assert
testAsyncTaskStatuses.ShouldAllBe(s => s.Value);
}
public async Task DeployDacPac(IReadOnlyCollection <DatabaseInfo> databases, CancellationToken cancellationToken)
{
var buildServer = _buildServerResolver.Resolve();
_log.Debug("Starting DacPac Deployment Tasks with {0} {1}.", _arguments.Threads, _arguments.Threads == 1 ? "thread" : "threads");
int completedTasks = 0;
int failedTasks = 0;
int totalTasks = databases.Count;
var asyncTaskRunner = new AsyncTaskRunner <DacPacDeployAsyncTask>(_arguments.Threads, cancellationToken);
foreach (var database in databases)
{
var dacPacDeployAsyncTask = _dacPacDeployAsyncTaskFactory.CreateAsyncTask();
dacPacDeployAsyncTask.Setup(database, (task, successful, elapsedMilliseconds) =>
{
Interlocked.Increment(ref completedTasks);
_log.Debug("Completed {0} out of {1} tasks.", completedTasks, totalTasks);
if (!successful)
{
Interlocked.Increment(ref failedTasks);
_log.Warning("{0} out of {1} tasks have failed.", failedTasks, totalTasks);
}
if (buildServer != null)
{
_log.WriteRaw(LogLevel.Info, buildServer.GenerateSetProgressMessage(completedTasks, totalTasks, "Progress Update"));
}
});
asyncTaskRunner.AddTask(dacPacDeployAsyncTask);
}
await asyncTaskRunner.WaitForCompletion();
if (failedTasks == totalTasks) // If all Tasks Failed
{
_log.Error("All Deployment Tasks Failed");
if (buildServer != null)
{
_log.WriteRaw(LogLevel.Error, buildServer.GenerateSetStatusFailMessage("All Deployment Tasks Failed"));
}
}
else if (failedTasks > 0) // If any Tasks Failed
{
_log.Warning("Some Deployment Tasks Failed");
if (buildServer != null)
{
_log.WriteRaw(LogLevel.Warn, buildServer.GenerateSetStatusSucceededWithIssuesMessage("Some Deployment Tasks Failed"));
}
}
else // If all Tasks Succeeded
{
_log.Info("All Deployment Tasks Succeeded");
if (buildServer != null)
{
_log.WriteRaw(LogLevel.Info, buildServer.GenerateSetStatusSucceededMessage("All Deployment Tasks Succeeded"));
}
}
}
