JobHandler.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
namespace Tgstation.Server.Host.Jobs
{
    sealed class JobHandler : IDisposable
    {
        public bool Started => task != null;
 
        public int? Progress { get; set; }
 
        public string? Stage { get; set; }
 
        readonly CancellationTokenSource cancellationTokenSource;
 
        readonly Func<CancellationToken, Task<bool>> jobActivator;
 
        Task<bool>? task;
 
        public JobHandler(Func<CancellationToken, Task<bool>> jobActivator)
        {
            this.jobActivator = jobActivator ?? throw new ArgumentNullException(nameof(jobActivator));
            cancellationTokenSource = new CancellationTokenSource();
        }
 
        public void Dispose() => cancellationTokenSource.Dispose();
 
        public Task<bool> Wait(CancellationToken cancellationToken)
        {
            if (task == null)
                throw new InvalidOperationException("Job not started!");
 
            return task.WaitAsync(cancellationToken);
        }
 
        public void Cancel() => cancellationTokenSource.Cancel();
 
        public void Start()
        {
            lock (cancellationTokenSource)
            {
                if (task != null)
                    throw new InvalidOperationException("Job already started");
                task = jobActivator(cancellationTokenSource.Token);
            }
        }
    }
}