reactive/AsyncRx.NET/System.Reactive.Async/Concurrency/TaskPoolAsyncScheduler.cs

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT License.
// See the LICENSE file in the project root for more information. 

using System.Threading;
using System.Threading.Tasks;

namespace System.Reactive.Concurrency
{
    public sealed class TaskPoolAsyncScheduler : AsyncSchedulerBase
    {
        private readonly TaskFactory _factory;

        public static TaskPoolAsyncScheduler Current { get; } = new TaskPoolAsyncScheduler(TaskScheduler.Current);
        public static TaskPoolAsyncScheduler Default { get; } = new TaskPoolAsyncScheduler(TaskScheduler.Default);

        public TaskPoolAsyncScheduler(TaskScheduler scheduler)
        {
            if (scheduler == null)
                throw new ArgumentNullException(nameof(scheduler));

            _factory = new TaskFactory(scheduler);
        }

        public TaskPoolAsyncScheduler(TaskFactory factory)
        {
            _factory = factory ?? throw new ArgumentNullException(nameof(factory));
        }

        protected override ValueTask Delay(TimeSpan dueTime, CancellationToken token) => new(Task.Delay(dueTime, token));

        protected override ValueTask ScheduleAsyncCore(Func<CancellationToken, ValueTask> action, CancellationToken token)
        {
            var task = _factory.StartNew(() => action(token).AsTask(), token);

            task.Unwrap().ContinueWith(t =>
            {
                if (!t.IsCanceled && t.Exception != null)
                {
                    // TODO: Call event handler.
                }
            });

            return default;
        }
    }
}