esiur-dotnet/Esyur/Core/AsyncAwaiter.cs

using System;
using System.Collections.Generic;
using System.Runtime.CompilerServices;
using System.Text;
using System.Threading.Tasks;

namespace Esyur.Core
{
    public class AsyncAwaiter : INotifyCompletion
    {
        Action callback = null;

        AsyncException exception = null;

        object result;


        public AsyncAwaiter(AsyncReply reply)
        {
            reply.Then(x =>
            {
                this.IsCompleted = true;
                this.result = x;
                this.callback?.Invoke();
            }).Error(x =>
            {
                exception = x;
                this.IsCompleted = true;
                this.callback?.Invoke();
            });
        }

        public object GetResult()
        {
            if (exception != null)
                throw exception;
            return result;
        }

        public bool IsCompleted { get; private set; }

        public void OnCompleted(Action continuation)
        {
            if (IsCompleted)
                continuation?.Invoke();
            else
                // Continue....
                callback = continuation;
        }


    }
}