Mercurial > pub > ImplabNet
comparison Implab/ActionTaskBase.cs @ 145:706fccb85524 v2
RC: cancellation support for promises + tests
| author | cin |
|---|---|
| date | Sun, 08 Mar 2015 02:52:27 +0300 |
| parents | |
| children | eb793fbbe4ea |
comparison
equal
deleted
inserted
replaced
| 144:8c0b95069066 | 145:706fccb85524 |
|---|---|
| 1 using System; | |
| 2 using System.Threading; | |
| 3 | |
| 4 namespace Implab { | |
| 5 public class ActionTaskBase : AbstractPromise { | |
| 6 readonly Action<Exception> m_cancel; | |
| 7 readonly Action<Exception> m_error; | |
| 8 | |
| 9 int m_cancelationLock; | |
| 10 | |
| 11 protected ActionTaskBase( Action<Exception> error, Action<Exception> cancel) { | |
| 12 m_error = error; | |
| 13 m_cancel = cancel; | |
| 14 } | |
| 15 | |
| 16 public void Reject(Exception error) { | |
| 17 Safe.ArgumentNotNull(error, "error"); | |
| 18 if (LockCancelation()) | |
| 19 HandleErrorInternal(error); | |
| 20 } | |
| 21 | |
| 22 protected void HandleErrorInternal(Exception error) { | |
| 23 if (m_error != null) { | |
| 24 try { | |
| 25 m_error(error); | |
| 26 SetResult(); | |
| 27 } catch(Exception err) { | |
| 28 SetError(err); | |
| 29 } | |
| 30 } else { | |
| 31 SetError(error); | |
| 32 } | |
| 33 } | |
| 34 | |
| 35 public override void CancelOperation(Exception reason) { | |
| 36 if (LockCancelation()) { | |
| 37 if (m_cancel != null) { | |
| 38 try { | |
| 39 m_cancel(reason); | |
| 40 SetResult(); | |
| 41 } catch (Exception err) { | |
| 42 HandleErrorInternal(err); | |
| 43 } | |
| 44 } else { | |
| 45 SetCancelled(reason); | |
| 46 } | |
| 47 } | |
| 48 } | |
| 49 | |
| 50 protected bool LockCancelation() { | |
| 51 return 0 == Interlocked.CompareExchange(ref m_cancelationLock, 1, 0); | |
| 52 } | |
| 53 } | |
| 54 } | |
| 55 |
