public TimerTasks(ILogTracer logger, ITaskOperations taskOperations, IJobOperations jobOperations, IScheduler scheduler)
{
_logger = logger;
_taskOperations = taskOperations;
_jobOperations = jobOperations;
_scheduler = scheduler;
}
/// <summary>
/// Lists all of the subtasks that are associated with the specified
/// multi-instance Task.
/// </summary>
/// <remarks>
/// If the Task is not a multi-instance Task then this returns an empty
/// collection.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job.
/// </param>
/// <param name='taskId'>
/// The ID of the Task.
/// </param>
/// <param name='taskListSubtasksOptions'>
/// Additional parameters for the operation
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <CloudTaskListSubtasksResult> ListSubtasksAsync(this ITaskOperations operations, string jobId, string taskId, TaskListSubtasksOptions taskListSubtasksOptions = default(TaskListSubtasksOptions), CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.ListSubtasksWithHttpMessagesAsync(jobId, taskId, taskListSubtasksOptions, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
public async Async.Task Stopping(Job job, ITaskOperations taskOperations)
{
job = job with {
State = JobState.Stopping
};
var tasks = await taskOperations.QueryAsync(filter : $"job_id eq '{job.JobId}'").ToListAsync();
var taskNotStopped = tasks.ToLookup(task => task.State != TaskState.Stopped);
var notStopped = taskNotStopped[true];
var stopped = taskNotStopped[false];
if (notStopped.Any())
{
foreach (var task in notStopped)
{
await taskOperations.MarkStopping(task);
}
}
else
{
job = job with {
State = JobState.Stopped
};
var taskInfo = stopped.Select(t => new JobTaskStopped(t.TaskId, t.Config.Task.Type, t.Error)).ToList();
await _context.Events.SendEvent(new EventJobStopped(job.JobId, job.Config, job.UserInfo, taskInfo));
}
await Replace(job);
}
}
/// <summary>
/// Reactivates a Task, allowing it to run again even if its retry count has
/// been exhausted.
/// </summary>
/// <remarks>
/// Reactivation makes a Task eligible to be retried again up to its maximum
/// retry count. The Task's state is changed to active. As the Task is no
/// longer in the completed state, any previous exit code or failure
/// information is no longer available after reactivation. Each time a Task is
/// reactivated, its retry count is reset to 0. Reactivation will fail for
/// Tasks that are not completed or that previously completed successfully
/// (with an exit code of 0). Additionally, it will fail if the Job has
/// completed (or is terminating or deleting).
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job containing the Task.
/// </param>
/// <param name='taskId'>
/// The ID of the Task to reactivate.
/// </param>
/// <param name='taskReactivateOptions'>
/// Additional parameters for the operation
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <TaskReactivateHeaders> ReactivateAsync(this ITaskOperations operations, string jobId, string taskId, TaskReactivateOptions taskReactivateOptions = default(TaskReactivateOptions), CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.ReactivateWithHttpMessagesAsync(jobId, taskId, taskReactivateOptions, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Headers);
}
}
/// <summary>
/// Adds a Task to the specified Job.
/// </summary>
/// <remarks>
/// The maximum lifetime of a Task from addition to completion is 180 days. If
/// a Task has not completed within 180 days of being added it will be
/// terminated by the Batch service and left in whatever state it was in at
/// that time.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job to which the Task is to be added.
/// </param>
/// <param name='task'>
/// The Task to be added.
/// </param>
/// <param name='taskAddOptions'>
/// Additional parameters for the operation
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <TaskAddHeaders> AddAsync(this ITaskOperations operations, string jobId, TaskAddParameter task, TaskAddOptions taskAddOptions = default(TaskAddOptions), CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.AddWithHttpMessagesAsync(jobId, task, taskAddOptions, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Headers);
}
}
/// <summary>
/// Lists all of the Tasks that are associated with the specified Job.
/// </summary>
/// <remarks>
/// For multi-instance Tasks, information such as affinityId, executionInfo and
/// nodeInfo refer to the primary Task. Use the list subtasks API to retrieve
/// information about subtasks.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='nextPageLink'>
/// The NextLink from the previous successful call to List operation.
/// </param>
/// <param name='taskListNextOptions'>
/// Additional parameters for the operation
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <IPage <CloudTask> > ListNextAsync(this ITaskOperations operations, string nextPageLink, TaskListNextOptions taskListNextOptions = default(TaskListNextOptions), CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.ListNextWithHttpMessagesAsync(nextPageLink, taskListNextOptions, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
/// <summary>
/// Gets information about the specified task.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The id of the job that contains the task.
/// </param>
/// <param name='taskId'>
/// The id of the task to get information about.
/// </param>
/// <param name='taskGetOptions'>
/// Additional parameters for the operation
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async System.Threading.Tasks.Task <CloudTask> GetAsync(this ITaskOperations operations, string jobId, string taskId, TaskGetOptions taskGetOptions = default(TaskGetOptions), System.Threading.CancellationToken cancellationToken = default(System.Threading.CancellationToken))
{
using (var _result = await operations.GetWithHttpMessagesAsync(jobId, taskId, taskGetOptions, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
/// <summary>
/// Adds a collection of Tasks to the specified Job.
/// </summary>
/// <remarks>
/// Note that each Task must have a unique ID. The Batch service may not return
/// the results for each Task in the same order the Tasks were submitted in
/// this request. If the server times out or the connection is closed during
/// the request, the request may have been partially or fully processed, or not
/// at all. In such cases, the user should re-issue the request. Note that it
/// is up to the user to correctly handle failures when re-issuing a request.
/// For example, you should use the same Task IDs during a retry so that if the
/// prior operation succeeded, the retry will not create extra Tasks
/// unexpectedly. If the response contains any Tasks which failed to add, a
/// client can retry the request. In a retry, it is most efficient to resubmit
/// only Tasks that failed to add, and to omit Tasks that were successfully
/// added on the first attempt. The maximum lifetime of a Task from addition to
/// completion is 180 days. If a Task has not completed within 180 days of
/// being added it will be terminated by the Batch service and left in whatever
/// state it was in at that time.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job to which the Task collection is to be added.
/// </param>
/// <param name='value'>
/// The collection of Tasks to add. The maximum count of Tasks is 100. The
/// total serialized size of this collection must be less than 1MB. If it is
/// greater than 1MB (for example if each Task has 100's of resource files or
/// environment variables), the request will fail with code
/// 'RequestBodyTooLarge' and should be retried again with fewer Tasks.
/// </param>
/// <param name='taskAddCollectionOptions'>
/// Additional parameters for the operation
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <TaskAddCollectionResult> AddCollectionAsync(this ITaskOperations operations, string jobId, IList <TaskAddParameter> value, TaskAddCollectionOptions taskAddCollectionOptions = default(TaskAddCollectionOptions), CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.AddCollectionWithHttpMessagesAsync(jobId, value, taskAddCollectionOptions, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
/// <summary>
/// Lists all of the tasks that are associated with the specified job.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The id of the job.
/// </param>
/// <param name='taskListOptions'>
/// Additional parameters for the operation
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <Microsoft.Rest.Azure.IPage <CloudTask> > ListAsync(this ITaskOperations operations, string jobId, TaskListOptions taskListOptions = default(TaskListOptions), System.Threading.CancellationToken cancellationToken = default(System.Threading.CancellationToken))
{
using (var _result = await operations.ListWithHttpMessagesAsync(jobId, taskListOptions, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
public AgentCanSchedule(ILogTracer log, IStorage storage, INodeOperations nodeOperations, ITaskOperations taskOperations, IScalesetOperations scalesetOperations)
{
_log = log;
_storage = storage;
_nodeOperations = nodeOperations;
_taskOperations = taskOperations;
_scalesetOperations = scalesetOperations;
}
public Scheduler(ITaskOperations taskOperations, IConfig config, IPoolOperations poolOperations, ILogTracer logTracer, IJobOperations jobOperations, IContainers containers)
{
_taskOperations = taskOperations;
_config = config;
_poolOperations = poolOperations;
_logTracer = logTracer;
_jobOperations = jobOperations;
_containers = containers;
}
public TimerRetention(
ILogTracer log,
ITaskOperations taskOps,
INotificationOperations notificaitonOps,
IJobOperations jobOps,
IReproOperations reproOps)
{
_log = log;
_taskOps = taskOps;
_notificaitonOps = notificaitonOps;
_jobOps = jobOps;
_reproOps = reproOps;
}
/// <summary>
/// Gets information about the specified Task.
/// </summary>
/// <remarks>
/// For multi-instance Tasks, information such as affinityId, executionInfo and
/// nodeInfo refer to the primary Task. Use the list subtasks API to retrieve
/// information about subtasks.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job that contains the Task.
/// </param>
/// <param name='taskId'>
/// The ID of the Task to get information about.
/// </param>
/// <param name='taskGetOptions'>
/// Additional parameters for the operation
/// </param>
public static CloudTask Get(this ITaskOperations operations, string jobId, string taskId, TaskGetOptions taskGetOptions = default(TaskGetOptions))
{
return(operations.GetAsync(jobId, taskId, taskGetOptions).GetAwaiter().GetResult());
}
/// <summary>
/// Reactivates the specified task.
/// </summary>
/// <remarks>
/// Reactivation makes a task eligible to be retried again up to its maximum
/// retry count. This will fail for tasks that are not completed or that
/// previously completed successfully (with an exit code of 0). Additionally,
/// this will fail if the job has completed (or is terminating or deleting).
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The id of the job containing the task.
/// </param>
/// <param name='taskId'>
/// The id of the task to reactivate.
/// </param>
/// <param name='taskReactivateOptions'>
/// Additional parameters for the operation
/// </param>
public static TaskReactivateHeaders Reactivate(this ITaskOperations operations, string jobId, string taskId, TaskReactivateOptions taskReactivateOptions = default(TaskReactivateOptions))
{
return(System.Threading.Tasks.Task.Factory.StartNew(s => ((ITaskOperations)s).ReactivateAsync(jobId, taskId, taskReactivateOptions), operations, System.Threading.CancellationToken.None, System.Threading.Tasks.TaskCreationOptions.None, System.Threading.Tasks.TaskScheduler.Default).Unwrap().GetAwaiter().GetResult());
}
/// <summary>
/// Terminates the specified task.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The id of the job containing the task.
/// </param>
/// <param name='taskId'>
/// The id of the task to terminate.
/// </param>
/// <param name='taskTerminateOptions'>
/// Additional parameters for the operation
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async System.Threading.Tasks.Task <TaskTerminateHeaders> TerminateAsync(this ITaskOperations operations, string jobId, string taskId, TaskTerminateOptions taskTerminateOptions = default(TaskTerminateOptions), System.Threading.CancellationToken cancellationToken = default(System.Threading.CancellationToken))
{
using (var _result = await operations.TerminateWithHttpMessagesAsync(jobId, taskId, taskTerminateOptions, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Headers);
}
}
/// <summary>
/// Lists all of the tasks that are associated with the specified job.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The id of the job.
/// </param>
/// <param name='taskListOptions'>
/// Additional parameters for the operation
/// </param>
public static Microsoft.Rest.Azure.IPage <CloudTask> List(this ITaskOperations operations, string jobId, TaskListOptions taskListOptions = default(TaskListOptions))
{
return(System.Threading.Tasks.Task.Factory.StartNew(s => ((ITaskOperations)s).ListAsync(jobId, taskListOptions), operations, System.Threading.CancellationToken.None, System.Threading.Tasks.TaskCreationOptions.None, System.Threading.Tasks.TaskScheduler.Default).Unwrap().GetAwaiter().GetResult());
}
/// <summary>
/// Adds a task to the specified job.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The id of the job to which the task is to be added.
/// </param>
/// <param name='task'>
/// The task to be added.
/// </param>
/// <param name='taskAddOptions'>
/// Additional parameters for the operation
/// </param>
public static TaskAddHeaders Add(this ITaskOperations operations, string jobId, TaskAddParameter task, TaskAddOptions taskAddOptions = default(TaskAddOptions))
{
return(System.Threading.Tasks.Task.Factory.StartNew(s => ((ITaskOperations)s).AddAsync(jobId, task, taskAddOptions), operations, System.Threading.CancellationToken.None, System.Threading.Tasks.TaskCreationOptions.None, System.Threading.Tasks.TaskScheduler.Default).Unwrap().GetAwaiter().GetResult());
}
public QueueTaskHearbeat(ILoggerFactory loggerFactory, ITaskOperations tasks, IEvents events)
{
_logger = loggerFactory.CreateLogger <QueueTaskHearbeat>();
_tasks = tasks;
_events = events;
}
/// <summary>
/// Lists all of the Tasks that are associated with the specified Job.
/// </summary>
/// <remarks>
/// For multi-instance Tasks, information such as affinityId, executionInfo and
/// nodeInfo refer to the primary Task. Use the list subtasks API to retrieve
/// information about subtasks.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='nextPageLink'>
/// The NextLink from the previous successful call to List operation.
/// </param>
/// <param name='taskListNextOptions'>
/// Additional parameters for the operation
/// </param>
public static IPage <CloudTask> ListNext(this ITaskOperations operations, string nextPageLink, TaskListNextOptions taskListNextOptions = default(TaskListNextOptions))
{
return(operations.ListNextAsync(nextPageLink, taskListNextOptions).GetAwaiter().GetResult());
}
/// <summary>
/// Adds a Task to the specified Job.
/// </summary>
/// <remarks>
/// The maximum lifetime of a Task from addition to completion is 180 days. If
/// a Task has not completed within 180 days of being added it will be
/// terminated by the Batch service and left in whatever state it was in at
/// that time.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job to which the Task is to be added.
/// </param>
/// <param name='task'>
/// The Task to be added.
/// </param>
/// <param name='taskAddOptions'>
/// Additional parameters for the operation
/// </param>
public static TaskAddHeaders Add(this ITaskOperations operations, string jobId, TaskAddParameter task, TaskAddOptions taskAddOptions = default(TaskAddOptions))
{
return(operations.AddAsync(jobId, task, taskAddOptions).GetAwaiter().GetResult());
}
/// <summary>
/// Reactivates a Task, allowing it to run again even if its retry count has
/// been exhausted.
/// </summary>
/// <remarks>
/// Reactivation makes a Task eligible to be retried again up to its maximum
/// retry count. The Task's state is changed to active. As the Task is no
/// longer in the completed state, any previous exit code or failure
/// information is no longer available after reactivation. Each time a Task is
/// reactivated, its retry count is reset to 0. Reactivation will fail for
/// Tasks that are not completed or that previously completed successfully
/// (with an exit code of 0). Additionally, it will fail if the Job has
/// completed (or is terminating or deleting).
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job containing the Task.
/// </param>
/// <param name='taskId'>
/// The ID of the Task to reactivate.
/// </param>
/// <param name='taskReactivateOptions'>
/// Additional parameters for the operation
/// </param>
public static TaskReactivateHeaders Reactivate(this ITaskOperations operations, string jobId, string taskId, TaskReactivateOptions taskReactivateOptions = default(TaskReactivateOptions))
{
return(operations.ReactivateAsync(jobId, taskId, taskReactivateOptions).GetAwaiter().GetResult());
}
/// <summary>
/// Lists all of the Tasks that are associated with the specified Job.
/// </summary>
/// <remarks>
/// For multi-instance Tasks, information such as affinityId, executionInfo and
/// nodeInfo refer to the primary Task. Use the list subtasks API to retrieve
/// information about subtasks.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job.
/// </param>
/// <param name='taskListOptions'>
/// Additional parameters for the operation
/// </param>
public static IPage <CloudTask> List(this ITaskOperations operations, string jobId, TaskListOptions taskListOptions = default(TaskListOptions))
{
return(operations.ListAsync(jobId, taskListOptions).GetAwaiter().GetResult());
}
/// <summary>
/// Lists all of the subtasks that are associated with the specified
/// multi-instance Task.
/// </summary>
/// <remarks>
/// If the Task is not a multi-instance Task then this returns an empty
/// collection.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job.
/// </param>
/// <param name='taskId'>
/// The ID of the Task.
/// </param>
/// <param name='taskListSubtasksOptions'>
/// Additional parameters for the operation
/// </param>
public static CloudTaskListSubtasksResult ListSubtasks(this ITaskOperations operations, string jobId, string taskId, TaskListSubtasksOptions taskListSubtasksOptions = default(TaskListSubtasksOptions))
{
return(operations.ListSubtasksAsync(jobId, taskId, taskListSubtasksOptions).GetAwaiter().GetResult());
}
/// <summary>
/// Updates the properties of the specified Task.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job containing the Task.
/// </param>
/// <param name='taskId'>
/// The ID of the Task to update.
/// </param>
/// <param name='constraints'>
/// Constraints that apply to this Task. If omitted, the Task is given the
/// default constraints. For multi-instance Tasks, updating the retention time
/// applies only to the primary Task and not subtasks.
/// </param>
/// <param name='taskUpdateOptions'>
/// Additional parameters for the operation
/// </param>
public static TaskUpdateHeaders Update(this ITaskOperations operations, string jobId, string taskId, TaskConstraints constraints = default(TaskConstraints), TaskUpdateOptions taskUpdateOptions = default(TaskUpdateOptions))
{
return(operations.UpdateAsync(jobId, taskId, constraints, taskUpdateOptions).GetAwaiter().GetResult());
}
/// <summary>
/// Adds a collection of tasks to the specified job.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The id of the job to which the task collection is to be added.
/// </param>
/// <param name='value'>
/// The collection of tasks to add.
/// </param>
/// <param name='taskAddCollectionOptions'>
/// Additional parameters for the operation
/// </param>
public static TaskAddCollectionResult AddCollection(this ITaskOperations operations, string jobId, System.Collections.Generic.IList <TaskAddParameter> value, TaskAddCollectionOptions taskAddCollectionOptions = default(TaskAddCollectionOptions))
{
return(System.Threading.Tasks.Task.Factory.StartNew(s => ((ITaskOperations)s).AddCollectionAsync(jobId, value, taskAddCollectionOptions), operations, System.Threading.CancellationToken.None, System.Threading.Tasks.TaskCreationOptions.None, System.Threading.Tasks.TaskScheduler.Default).Unwrap().GetAwaiter().GetResult());
}
/// <summary>
/// Adds a collection of Tasks to the specified Job.
/// </summary>
/// <remarks>
/// Note that each Task must have a unique ID. The Batch service may not return
/// the results for each Task in the same order the Tasks were submitted in
/// this request. If the server times out or the connection is closed during
/// the request, the request may have been partially or fully processed, or not
/// at all. In such cases, the user should re-issue the request. Note that it
/// is up to the user to correctly handle failures when re-issuing a request.
/// For example, you should use the same Task IDs during a retry so that if the
/// prior operation succeeded, the retry will not create extra Tasks
/// unexpectedly. If the response contains any Tasks which failed to add, a
/// client can retry the request. In a retry, it is most efficient to resubmit
/// only Tasks that failed to add, and to omit Tasks that were successfully
/// added on the first attempt. The maximum lifetime of a Task from addition to
/// completion is 180 days. If a Task has not completed within 180 days of
/// being added it will be terminated by the Batch service and left in whatever
/// state it was in at that time.
/// </remarks>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='jobId'>
/// The ID of the Job to which the Task collection is to be added.
/// </param>
/// <param name='value'>
/// The collection of Tasks to add. The maximum count of Tasks is 100. The
/// total serialized size of this collection must be less than 1MB. If it is
/// greater than 1MB (for example if each Task has 100's of resource files or
/// environment variables), the request will fail with code
/// 'RequestBodyTooLarge' and should be retried again with fewer Tasks.
/// </param>
/// <param name='taskAddCollectionOptions'>
/// Additional parameters for the operation
/// </param>
public static TaskAddCollectionResult AddCollection(this ITaskOperations operations, string jobId, IList <TaskAddParameter> value, TaskAddCollectionOptions taskAddCollectionOptions = default(TaskAddCollectionOptions))
{
return(operations.AddCollectionAsync(jobId, value, taskAddCollectionOptions).GetAwaiter().GetResult());
}
