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
    {
        readonly CancellationTokenSource cancellationTokenSource;

        readonly Task task;

        public JobHandler(Func<CancellationToken, Task> job)
        {
            if (job == null)
                throw new ArgumentNullException(nameof(job));
            cancellationTokenSource = new CancellationTokenSource();
            task = job(cancellationTokenSource.Token);
        }

        public void Dispose() => cancellationTokenSource.Dispose();

        public int? Progress { get; set; }

        public async Task Wait(CancellationToken cancellationToken)
        {
            TaskCompletionSource<object> tcs = new TaskCompletionSource<object>();
            using (cancellationToken.Register(() => tcs.SetCanceled()))
                await Task.WhenAny(tcs.Task, task).ConfigureAwait(false);
            cancellationToken.ThrowIfCancellationRequested();
        }

        public void Cancel() => cancellationTokenSource.Cancel();
    }
}