New awaiters

Esyur.Stores.MongoDB/Esyur.Stores.MongoDB.csproj

@@ -14,6 +14,16 @@
     <Version>1.3.0</Version>
   </PropertyGroup>
 
+  <ItemGroup>
+    <Compile Remove="MongoDBStore.cs" />
+    <Compile Remove="MongoDBStoreGeneric.cs" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <None Include="MongoDBStore.cs" />
+    <None Include="MongoDBStoreGeneric.cs" />
+  </ItemGroup>
+
   <ItemGroup>
     <PackageReference Include="MongoDB.Bson" Version="2.9.1" />
     <PackageReference Include="MongoDB.Driver" Version="2.9.1" />

Esyur/Core/AsyncAwaiter.cs

@@ -6,20 +6,21 @@ using System.Threading.Tasks;
 
 namespace Esyur.Core
 {
-    public class AsyncAwaiter<T> : INotifyCompletion
+    public class AsyncAwaiter : INotifyCompletion
     {
         Action callback = null;
 
         AsyncException exception = null;
 
-        T result;
+        object result;
+
 
         public AsyncAwaiter(AsyncReply reply)
         {
             reply.Then(x =>
             {
                 this.IsCompleted = true;
-                this.result = (T)x;
+                this.result = x;
                 this.callback?.Invoke();
             }).Error(x =>
             {
@@ -29,7 +30,7 @@ namespace Esyur.Core
             });
         }
 
-        public T GetResult()
+        public object GetResult()
         {
             if (exception != null)
                 throw exception;

Esyur/Core/AsyncAwaiterGeneric.cs

@@ -0,0 +1,53 @@
+using System;
+using System.Collections.Generic;
+using System.Runtime.CompilerServices;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Esyur.Core
+{
+    public class AsyncAwaiter<T> : INotifyCompletion
+    {
+        Action callback = null;
+
+        AsyncException exception = null;
+
+        T result;
+
+        public AsyncAwaiter(AsyncReply<T> reply)
+        {
+            reply.Then(x =>
+            {
+                this.IsCompleted = true;
+                this.result = (T)x;
+                this.callback?.Invoke();
+            }).Error(x =>
+            {
+                exception = x;
+                this.IsCompleted = true;
+                this.callback?.Invoke();
+            });
+        }
+
+        public T 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;
+        }
+
+
+    }
+}
+

Esyur/Core/AsyncBag.cs

@@ -46,9 +46,9 @@ namespace Esyur.Core
             return this;
         }
 
-        public new AsyncAwaiter<object[]> GetAwaiter()
+        public new AsyncBagAwaiter GetAwaiter()
         {
-            return new AsyncAwaiter<object[]>(this);
+            return new AsyncBagAwaiter(this);
         }
 
         public new object[] Wait()

Esyur/Core/AsyncBagAwaiter.cs

@@ -0,0 +1,53 @@
+using System;
+using System.Collections.Generic;
+using System.Runtime.CompilerServices;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Esyur.Core
+{
+    public class AsyncBagAwaiter : INotifyCompletion
+    {
+        Action callback = null;
+
+        AsyncException exception = null;
+
+        object[] result;
+
+        public AsyncBagAwaiter(AsyncBag 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;
+        }
+
+
+    }
+}
+

Esyur/Core/AsyncBagAwaiterGeneric.cs

@@ -0,0 +1,53 @@
+using System;
+using System.Collections.Generic;
+using System.Runtime.CompilerServices;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Esyur.Core
+{
+    public class AsyncBagAwaiter<T> : INotifyCompletion
+    {
+        Action callback = null;
+
+        AsyncException exception = null;
+
+        T[] result;
+
+        public AsyncBagAwaiter(AsyncBag<T> reply)
+        {
+            reply.Then(x =>
+            {
+                this.IsCompleted = true;
+                this.result = x;
+                this.callback?.Invoke();
+            }).Error(x =>
+            {
+                exception = x;
+                this.IsCompleted = true;
+                this.callback?.Invoke();
+            });
+        }
+
+        public T[] 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;
+        }
+
+
+    }
+}
+

Esyur/Core/AsyncBagGeneric.cs

@@ -51,9 +51,9 @@ namespace Esyur.Core
         }
 
 
-        public new AsyncAwaiter<T[]> GetAwaiter()
+        public new AsyncBagAwaiter<T> GetAwaiter()
         {
-            return new AsyncAwaiter<T[]>(this);
+            return new AsyncBagAwaiter<T>(this);
         }
 
         public new T[] Wait()

Esyur/Core/AsyncReply.cs

@@ -255,9 +255,9 @@ namespace Esyur.Core
             //}
         }
 
-        public AsyncAwaiter<object> GetAwaiter()
+        public AsyncAwaiter GetAwaiter()
         {
-            return new AsyncAwaiter<object>(this);
+            return new AsyncAwaiter(this);
         }