diff Implab/FuncChainTaskBase.cs @ 145:706fccb85524 v2

RC: cancellation support for promises + tests
author cin
date Sun, 08 Mar 2015 02:52:27 +0300
parents
children eb793fbbe4ea
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Implab/FuncChainTaskBase.cs	Sun Mar 08 02:52:27 2015 +0300
@@ -0,0 +1,60 @@
+using System;
+using System.Threading;
+
+namespace Implab {
+    public class FuncChainTaskBase<TResult> : AbstractPromise<TResult> {
+        readonly Func<Exception, IPromise<TResult>> m_error;
+        readonly Func<Exception, IPromise<TResult>> m_cancel;
+
+        int m_cancelationLock;
+
+        protected FuncChainTaskBase( Func<Exception, IPromise<TResult>> error, Func<Exception, IPromise<TResult>> cancel) {
+            m_error = error;
+            m_cancel = cancel;
+        }
+
+        public void Reject(Exception error) {
+            if (LockCancelation())
+                HandleErrorInternal(error);
+        }
+
+        public override void CancelOperation(Exception reason) {
+            if (m_cancel != null && LockCancelation()) {
+                try {
+                    Observe(m_cancel(reason));
+                } catch(Exception err) {
+                    HandleErrorInternal(err);
+                }
+            }
+
+        }
+
+        protected void HandleErrorInternal(Exception error) {
+            if (m_error != null) {
+                try {
+                    Observe(m_error(error));
+                } catch(Exception err) {
+                    SetError(err);
+                }
+            } else {
+                SetError(error);
+            }
+        }
+
+        protected void Observe(IPromise<TResult> operation) {
+            if (operation == null)
+                throw new NullReferenceException("The task returned null promise");
+
+            // pass operation results to the current promise
+            operation.On(SetResult, SetError, SetCancelled);
+
+            // pass the cancelation request
+            CancellationRequested(operation.Cancel);
+        }
+
+        protected bool LockCancelation() {
+            return 0 == Interlocked.CompareExchange(ref m_cancelationLock, 1, 0);
+        }
+    }
+}
+