|
248
|
1 using System;
|
|
|
2 using System.Collections.Generic;
|
|
|
3 using System.Linq;
|
|
|
4 using System.Threading;
|
|
|
5
|
|
|
6 namespace Implab {
|
|
|
7 class PromiseAll<T> : IResolvable {
|
|
|
8
|
|
|
9 int m_count;
|
|
|
10
|
|
|
11 readonly List<IPromise<T>> m_promises = new List<IPromise<T>>();
|
|
|
12
|
|
|
13 readonly Deferred<T[]> m_deferred;
|
|
|
14
|
|
|
15 IPromise<T[]> m_result;
|
|
|
16
|
|
|
17 readonly Func<T, IPromise> m_cleanup;
|
|
|
18
|
|
|
19 readonly Action m_cancel;
|
|
|
20
|
|
|
21 public bool Done {
|
|
|
22 get { return m_deferred.Promise.IsResolved && m_cleanup == null; }
|
|
|
23 }
|
|
|
24
|
|
|
25 public IPromise<T[]> ResultPromise {
|
|
|
26 get { return m_result; }
|
|
|
27 }
|
|
|
28
|
|
|
29 public void AddPromise(IPromise<T> promise) {
|
|
|
30 Interlocked.Increment(ref m_count);
|
|
|
31 promise.Then(this);
|
|
|
32 }
|
|
|
33
|
|
|
34 public PromiseAll(Deferred<T[]> deferred, Func<T, IPromise> cleanup, Action cancel) {
|
|
|
35 m_deferred = deferred;
|
|
|
36 m_cancel = cancel;
|
|
|
37 m_cleanup = cleanup;
|
|
|
38 }
|
|
|
39
|
|
|
40 public void Resolve() {
|
|
|
41 if (Interlocked.Decrement(ref m_count) == 0)
|
|
|
42 m_deferred.Resolve(GetResults());
|
|
|
43 }
|
|
|
44
|
|
|
45 public void Reject(Exception error) {
|
|
|
46 m_deferred.Reject(error);
|
|
|
47 }
|
|
|
48
|
|
|
49 public void Complete() {
|
|
|
50 if (m_cancel != null || m_cleanup != null)
|
|
|
51 m_result = m_deferred.Promise.Catch(CleanupResults);
|
|
|
52 else
|
|
|
53 m_result = m_deferred.Promise;
|
|
|
54 }
|
|
|
55
|
|
|
56 IPromise<T[]> CleanupResults(Exception reason) {
|
|
|
57 var errors = new List<Exception>();
|
|
|
58 errors.Add(reason);
|
|
|
59
|
|
|
60 if (m_cancel != null)
|
|
|
61 try {
|
|
|
62 m_cancel();
|
|
|
63 } catch (Exception e) {
|
|
|
64 errors.Add(e);
|
|
|
65 }
|
|
|
66
|
|
|
67 if (m_cleanup != null) {
|
|
|
68 return Promise.All(
|
|
|
69 m_promises.Select(p => p
|
|
|
70 .Then(m_cleanup, e => { })
|
|
|
71 .Catch(e => {
|
|
|
72 errors.Add(e);
|
|
|
73 })
|
|
|
74 )
|
|
|
75 ).Then<T[]>(new Func<T[]>(() => {
|
|
|
76 throw new AggregateException(errors);
|
|
|
77 }), (Func<Exception, T[]>)null);
|
|
|
78 } else {
|
|
|
79 return Promise.Reject<T[]>(errors.Count > 1 ? new AggregateException(errors) : reason);
|
|
|
80 }
|
|
|
81 }
|
|
|
82
|
|
|
83 T[] GetResults() {
|
|
|
84 var results = new T[m_promises.Count];
|
|
|
85 for (var i = 0; i < results.Length; i++)
|
|
|
86 results[i] = m_promises[i].Join();
|
|
|
87 return results;
|
|
|
88 }
|
|
|
89 }
|
|
|
90 } |
