annotate Implab/FuncTaskBase.cs @ 145:706fccb85524 v2

RC: cancellation support for promises + tests
author cin
date Sun, 08 Mar 2015 02:52:27 +0300
parents 8c0b95069066
children eb793fbbe4ea
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
144
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
1 using System;
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
2 using System.Threading;
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
3
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
4 namespace Implab {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
5 public class FuncTaskBase<TResult> : AbstractPromise<TResult> {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
6 readonly Func<Exception, TResult> m_cancel;
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
7 readonly Func<Exception, TResult> m_error;
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
8
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
9 int m_cancelationLock;
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
10
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
11 protected FuncTaskBase( Func<Exception, TResult> error, Func<Exception, TResult> cancel) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
12 m_error = error;
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
13 m_cancel = cancel;
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
14 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
15
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
16 public void Reject(Exception error) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
17 Safe.ArgumentNotNull(error, "error");
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
18 if (LockCancelation())
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
19 HandleErrorInternal(error);
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
20 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
21
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
22 protected void HandleErrorInternal(Exception error) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
23 if (m_error != null) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
24 try {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
25 SetResult(m_error(error));
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
26 } catch(Exception err) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
27 SetError(err);
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
28 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
29 } else {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
30 SetError(error);
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
31 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
32 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
33
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
34 public override void CancelOperation(Exception reason) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
35 if (LockCancelation()) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
36 if (m_cancel != null) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
37 try {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
38 SetResult(m_cancel(reason));
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
39 } catch (Exception err) {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
40 HandleErrorInternal(err);
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
41 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
42 } else {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
43 SetCancelled(reason);
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
44 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
45 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
46 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
47
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
48 protected bool LockCancelation() {
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
49 return 0 == Interlocked.CompareExchange(ref m_cancelationLock, 1, 0);
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
50 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
51 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
52 }
8c0b95069066 DRAFT: refactoring
cin
parents:
diff changeset
53