CTT/Unity/Assets/Model/Core/Async/ETTaskCompletionSource.cs

using System;
using System.Diagnostics;
using System.Runtime.CompilerServices;
using System.Runtime.ExceptionServices;

namespace ET
{
    public class ETTaskCompletionSource: ICriticalNotifyCompletion
    {
        private AwaiterStatus state;
        private ExceptionDispatchInfo exception;
        private Action continuation; // action or list

        [DebuggerHidden]
        public ETTask Task => new ETTask(this);

        [DebuggerHidden]
        public AwaiterStatus Status => state;

        [DebuggerHidden]
        public bool IsCompleted => state != AwaiterStatus.Pending;

        [DebuggerHidden]
        public void UnsafeOnCompleted(Action action)
        {
            this.continuation = action;
            if (state != AwaiterStatus.Pending)
            {
                TryInvokeContinuation();
            }
        }

        [DebuggerHidden]
        public void OnCompleted(Action action)
        {
            this.UnsafeOnCompleted(action);
        }

        [DebuggerHidden]
        public void GetResult()
        {
            switch (this.state)
            {
                case AwaiterStatus.Succeeded:
                    return;
                case AwaiterStatus.Faulted:
                    this.exception?.Throw();
                    this.exception = null;
                    return;
                default:
                    throw new NotSupportedException("ETTask does not allow call GetResult directly when task not completed. Please use 'await'.");
            }
        }

        [DebuggerHidden]
        public void SetResult()
        {
            if (this.TrySetResult())
            {
                return;
            }

            throw new InvalidOperationException("TaskT_TransitionToFinal_AlreadyCompleted");
        }

        [DebuggerHidden]
        public void SetException(Exception e)
        {
            if (this.TrySetException(e))
            {
                return;
            }

            throw new InvalidOperationException("TaskT_TransitionToFinal_AlreadyCompleted");
        }

        [DebuggerHidden]
        private void TryInvokeContinuation()
        {
            this.continuation?.Invoke();
            this.continuation = null;
        }

        [DebuggerHidden]
        private bool TrySetResult()
        {
            if (this.state != AwaiterStatus.Pending)
            {
                return false;
            }

            this.state = AwaiterStatus.Succeeded;

            this.TryInvokeContinuation();
            return true;
        }

        [DebuggerHidden]
        private bool TrySetException(Exception e)
        {
            if (this.state != AwaiterStatus.Pending)
            {
                return false;
            }

            this.state = AwaiterStatus.Faulted;

            this.exception = ExceptionDispatchInfo.Capture(e);
            this.TryInvokeContinuation();
            return true;
        }
    }

    public class ETTaskCompletionSource<T>: ICriticalNotifyCompletion
    {
        private AwaiterStatus state;
        private T value;
        private ExceptionDispatchInfo exception;
        private Action continuation; // action or list

        [DebuggerHidden]
        public ETTask<T> Task => new ETTask<T>(this);

        [DebuggerHidden]
        public ETTaskCompletionSource<T> GetAwaiter()
        {
            return this;
        }

        [DebuggerHidden]
        public T GetResult()
        {
            switch (this.state)
            {
                case AwaiterStatus.Succeeded:
                    return this.value;
                case AwaiterStatus.Faulted:
                    this.exception?.Throw();
                    this.exception = null;
                    return default;
                default:
                    throw new NotSupportedException("ETask does not allow call GetResult directly when task not completed. Please use 'await'.");
            }
        }

        [DebuggerHidden]
        public bool IsCompleted => state != AwaiterStatus.Pending;

        [DebuggerHidden]
        public AwaiterStatus Status => state;

        [DebuggerHidden]
        public void UnsafeOnCompleted(Action action)
        {
            this.continuation = action;
            if (state != AwaiterStatus.Pending)
            {
                TryInvokeContinuation();
            }
        }

        [DebuggerHidden]
        public void OnCompleted(Action action)
        {
            this.UnsafeOnCompleted(action);
        }

        [DebuggerHidden]
        public void SetResult(T result)
        {
            if (this.TrySetResult(result))
            {
                return;
            }

            throw new InvalidOperationException("TaskT_TransitionToFinal_AlreadyCompleted");
        }

        [DebuggerHidden]
        public void SetException(Exception e)
        {
            if (this.TrySetException(e))
            {
                return;
            }

            throw new InvalidOperationException("TaskT_TransitionToFinal_AlreadyCompleted");
        }

        [DebuggerHidden]
        private void TryInvokeContinuation()
        {
            this.continuation?.Invoke();
            this.continuation = null;
        }

        [DebuggerHidden]
        private bool TrySetResult(T result)
        {
            if (this.state != AwaiterStatus.Pending)
            {
                return false;
            }

            this.state = AwaiterStatus.Succeeded;

            this.value = result;
            this.TryInvokeContinuation();
            return true;
        }

        [DebuggerHidden]
        private bool TrySetException(Exception e)
        {
            if (this.state != AwaiterStatus.Pending)
            {
                return false;
            }

            this.state = AwaiterStatus.Faulted;

            this.exception = ExceptionDispatchInfo.Capture(e);
            this.TryInvokeContinuation();
            return true;
        }
    }
}