MewtocolNet/MewtocolNet/Queue/SerialQueue.cs

using System;
using System.Threading.Tasks;

namespace MewtocolNet.Queue {

    internal class SerialQueue {

        readonly object _locker = new object();
        readonly WeakReference<Task> _lastTask = new WeakReference<Task>(null);

        internal Task Enqueue (Action action) {
            return Enqueue<bool>(() => {
                action();
                return true;
            });
        }

        internal Task<T> Enqueue<T> (Func<T> function) {
            lock (_locker) {
                Task lastTask;
                Task<T> resultTask;

                if (_lastTask.TryGetTarget(out lastTask)) {
                    resultTask = lastTask.ContinueWith(_ => function(), TaskContinuationOptions.ExecuteSynchronously);
                } else {
                    resultTask = Task.Run(function);
                }

                _lastTask.SetTarget(resultTask);

                return resultTask;
            }
        }

        internal Task Enqueue (Func<Task> asyncAction) {
            lock (_locker) {
                Task lastTask;
                Task resultTask;

                if (_lastTask.TryGetTarget(out lastTask)) {
                    resultTask = lastTask.ContinueWith(_ => asyncAction(), TaskContinuationOptions.ExecuteSynchronously).Unwrap();
                } else {
                    resultTask = Task.Run(asyncAction);
                }

                _lastTask.SetTarget(resultTask);

                return resultTask;
            }
        }

        internal Task<T> Enqueue<T> (Func<Task<T>> asyncFunction) {
            lock (_locker) {
                Task lastTask;
                Task<T> resultTask;

                if (_lastTask.TryGetTarget(out lastTask)) {
                    resultTask = lastTask.ContinueWith(_ => asyncFunction(), TaskContinuationOptions.ExecuteSynchronously).Unwrap();
                } else {
                    resultTask = Task.Run(asyncFunction);
                }

                _lastTask.SetTarget(resultTask);

                return resultTask;
            }
        }

    }

}