Allow to update worker and status in a single operation; ensure the worker is updated when an announced task is started

Author: dei79

dotNet/CoreHelpers.Extensions.Logging.Tasks/TaskLogger.cs

@@ -56,7 +56,7 @@ public TaskLogger(ITaskLoggerFactory taskLoggerFactory)
             _currentTaskLogger = _taskLoggerFactory.CreateTaskLogger(castedState.TaskId);
 
             // ensure the task is running now
-            _taskLoggerFactory.UpdateTaskStatus(castedState.TaskId, TaskStatus.Running).GetAwaiter().GetResult();
+            _taskLoggerFactory.UpdateTaskStatus(castedState.TaskId, TaskStatus.Running, castedState.TaskWorker).GetAwaiter().GetResult();
 
             // done
             return new TaskLoggerScope(castedState, this);

dotNet/CoreHelpers.Extensions.Logging.Tasks/TaskLoggerExtension.cs

@@ -16,9 +16,9 @@ public static ILoggingBuilder AddTaskLogger(this ILoggingBuilder builder)
 
         public static ITaskLoggerTypedScope? BeginTaskScope(this ILogger logger, string taskId)
             => BeginTypedTaskScope(logger, new TaskLoggerState() { TaskId = taskId, IsTaskAnnounced = true });
-
-        public static ITaskLoggerTypedScope? BeginTaskScope(this ILogger logger, string taskId, string metaDataString)
-            => BeginTypedTaskScope(logger, new TaskLoggerState() { TaskId = taskId, IsTaskAnnounced = true, MetaData = metaDataString });
+        
+        public static ITaskLoggerTypedScope? BeginTaskScope(this ILogger logger, string taskId, string taskWorker)
+            => BeginTypedTaskScope(logger, new TaskLoggerState() { TaskId = taskId, IsTaskAnnounced = true, TaskWorker = taskWorker });
 
         public static ITaskLoggerTypedScope? BeginNewTaskScope(this ILogger logger, string taskType, string taskSource, string taskWorker)
             => BeginTypedTaskScope(logger, new TaskLoggerState() { TaskId = string.Empty, TaskType = taskType, TaskSource = taskSource, TaskWorker = taskWorker, IsTaskAnnounced = false });

dotNet/CoreHelpers.TaskLogging.Abstractions/ITaskLoggerFactory.cs

@@ -64,6 +64,15 @@ public interface ITaskLoggerFactory
         /// <returns></returns>
         Task UpdateTaskStatus(string taskId, TaskStatus taskStatus);
 
+        /// <summary>
+        /// Update the task status of a given task including the worker
+        /// </summary>
+        /// <param name="taskId"></param>
+        /// <param name="taskStatus"></param>
+        /// <param name="taskWorker"></param>
+        /// <returns></returns>
+        Task UpdateTaskStatus(string taskId, TaskStatus taskStatus, string taskWorker);
+        
         /// <summary>
         /// Allows to update the worker of a task, especially useful for task reassignment
         /// </summary>

dotNet/CoreHelpers.TaskLogging/AzureStorageTableTaskLoggerFactory.cs

@@ -64,13 +64,22 @@ public Task<string> AnnounceTask(string taskType, string taskSource, string task
 
         public async Task UpdateTaskStatus(string taskKey, TaskStatus taskStatus)
         {            
+            await UpdateTaskStatus(taskKey, taskStatus, string.Empty);
+        }
+
+        public async Task UpdateTaskStatus(string taskKey, TaskStatus taskStatus, string taskWorker)
+        {
             // build the task entity
             var taskEntity = new AzureTableTaskEntity()
             {
                 PartitionKey = taskKey,
                 RowKey = taskKey,                
                 TaskState = taskStatus.ToString()                
             };
+            
+            // check the worker
+            if (!string.IsNullOrEmpty(taskWorker))
+                taskEntity.TaskWorker = taskWorker;
 
             // adjust the timings
             if (taskStatus == TaskStatus.Running)