private async Task <int> GetOrCreateTaskLogId(IServiceBusMessage message, CancellationToken cancellationToken, ITaskClient taskClient, Guid projectId, Guid planId, Guid jobId, Guid parentTimelineId, string timelineName, string hubName)
{
// attempt to get from message
var logIdObject = message.GetProperty(VstsMessageConstants.TaskLogIdPropertyName);
var taskLogId = 0;
var gotLogId = logIdObject != null && int.TryParse(logIdObject.ToString(), out taskLogId);
if (gotLogId)
{
return(taskLogId);
}
// attempt to find existing
var records = await taskClient.GetRecordsAsync(projectId, hubName, planId, parentTimelineId, userState : null, cancellationToken : cancellationToken).ConfigureAwait(false);
foreach (var record in records)
{
if (string.Equals(record.Name, timelineName, StringComparison.OrdinalIgnoreCase))
{
return(record.Log.Id);
}
}
// Create a new timeline
var subTimelineId = Guid.NewGuid();
// create a log file
var logsSubtimelineId = string.Format(@"logs\{0:D}", subTimelineId);
var taskLog = await taskClient.CreateLogAsync(projectId, hubName, planId, new TaskLog(logsSubtimelineId), userState : null, cancellationToken : cancellationToken).ConfigureAwait(false);
// create a sub-timeline
var timelineRecord = new TimelineRecord
{
Id = subTimelineId,
Name = timelineName,
StartTime = DateTime.UtcNow,
State = TimelineRecordState.InProgress,
RecordType = "task", // Record type can be job or task, as we will be dealing only with task here
WorkerName = this.settings.WorkerName,
Order = 1, // The job timeline record must be at order 1
Log = taskLog,
ParentId = jobId,
PercentComplete = 0,
ErrorCount = 0,
WarningCount = 0
};
await taskClient.UpdateTimelineRecordsAsync(projectId, hubName, planId, parentTimelineId, new List <TimelineRecord> {
timelineRecord
}, cancellationToken).ConfigureAwait(false);
// save the taskLogId on the message
taskLogId = taskLog.Id;
return(taskLogId);
}
internal async Task CompleteTimelineRecords(Guid projectId, Guid planId, string hubName, Guid parentTimelineId, TaskResult result, CancellationToken cancellationToken, ITaskClient taskClient)
{
// Find all existing timeline records and close them
var records = await taskClient.GetRecordsAsync(projectId, hubName, planId, parentTimelineId, userState : null, cancellationToken : cancellationToken).ConfigureAwait(false);
var recordsToUpdate = GetTimelineRecordsToUpdate(records);
foreach (var record in recordsToUpdate)
{
record.State = TimelineRecordState.Completed;
record.PercentComplete = 100;
record.Result = result;
record.FinishTime = DateTime.UtcNow;
}
await taskClient.UpdateTimelineRecordsAsync(projectId, hubName, planId, parentTimelineId, recordsToUpdate, cancellationToken).ConfigureAwait(false);
}
