cannam@49
|
1 // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
|
cannam@49
|
2 // Licensed under the MIT License:
|
cannam@49
|
3 //
|
cannam@49
|
4 // Permission is hereby granted, free of charge, to any person obtaining a copy
|
cannam@49
|
5 // of this software and associated documentation files (the "Software"), to deal
|
cannam@49
|
6 // in the Software without restriction, including without limitation the rights
|
cannam@49
|
7 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
cannam@49
|
8 // copies of the Software, and to permit persons to whom the Software is
|
cannam@49
|
9 // furnished to do so, subject to the following conditions:
|
cannam@49
|
10 //
|
cannam@49
|
11 // The above copyright notice and this permission notice shall be included in
|
cannam@49
|
12 // all copies or substantial portions of the Software.
|
cannam@49
|
13 //
|
cannam@49
|
14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
cannam@49
|
15 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
cannam@49
|
16 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
cannam@49
|
17 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
cannam@49
|
18 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
cannam@49
|
19 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
cannam@49
|
20 // THE SOFTWARE.
|
cannam@49
|
21
|
cannam@49
|
22 // This file contains extended inline implementation details that are required along with async.h.
|
cannam@49
|
23 // We move this all into a separate file to make async.h more readable.
|
cannam@49
|
24 //
|
cannam@49
|
25 // Non-inline declarations here are defined in async.c++.
|
cannam@49
|
26
|
cannam@49
|
27 #ifndef KJ_ASYNC_H_
|
cannam@49
|
28 #error "Do not include this directly; include kj/async.h."
|
cannam@49
|
29 #include "async.h" // help IDE parse this file
|
cannam@49
|
30 #endif
|
cannam@49
|
31
|
cannam@49
|
32 #ifndef KJ_ASYNC_INL_H_
|
cannam@49
|
33 #define KJ_ASYNC_INL_H_
|
cannam@49
|
34
|
cannam@49
|
35 #if defined(__GNUC__) && !KJ_HEADER_WARNINGS
|
cannam@49
|
36 #pragma GCC system_header
|
cannam@49
|
37 #endif
|
cannam@49
|
38
|
cannam@49
|
39 namespace kj {
|
cannam@49
|
40 namespace _ { // private
|
cannam@49
|
41
|
cannam@49
|
42 template <typename T>
|
cannam@49
|
43 class ExceptionOr;
|
cannam@49
|
44
|
cannam@49
|
45 class ExceptionOrValue {
|
cannam@49
|
46 public:
|
cannam@49
|
47 ExceptionOrValue(bool, Exception&& exception): exception(kj::mv(exception)) {}
|
cannam@49
|
48 KJ_DISALLOW_COPY(ExceptionOrValue);
|
cannam@49
|
49
|
cannam@49
|
50 void addException(Exception&& exception) {
|
cannam@49
|
51 if (this->exception == nullptr) {
|
cannam@49
|
52 this->exception = kj::mv(exception);
|
cannam@49
|
53 }
|
cannam@49
|
54 }
|
cannam@49
|
55
|
cannam@49
|
56 template <typename T>
|
cannam@49
|
57 ExceptionOr<T>& as() { return *static_cast<ExceptionOr<T>*>(this); }
|
cannam@49
|
58 template <typename T>
|
cannam@49
|
59 const ExceptionOr<T>& as() const { return *static_cast<const ExceptionOr<T>*>(this); }
|
cannam@49
|
60
|
cannam@49
|
61 Maybe<Exception> exception;
|
cannam@49
|
62
|
cannam@49
|
63 protected:
|
cannam@49
|
64 // Allow subclasses to have move constructor / assignment.
|
cannam@49
|
65 ExceptionOrValue() = default;
|
cannam@49
|
66 ExceptionOrValue(ExceptionOrValue&& other) = default;
|
cannam@49
|
67 ExceptionOrValue& operator=(ExceptionOrValue&& other) = default;
|
cannam@49
|
68 };
|
cannam@49
|
69
|
cannam@49
|
70 template <typename T>
|
cannam@49
|
71 class ExceptionOr: public ExceptionOrValue {
|
cannam@49
|
72 public:
|
cannam@49
|
73 ExceptionOr() = default;
|
cannam@49
|
74 ExceptionOr(T&& value): value(kj::mv(value)) {}
|
cannam@49
|
75 ExceptionOr(bool, Exception&& exception): ExceptionOrValue(false, kj::mv(exception)) {}
|
cannam@49
|
76 ExceptionOr(ExceptionOr&&) = default;
|
cannam@49
|
77 ExceptionOr& operator=(ExceptionOr&&) = default;
|
cannam@49
|
78
|
cannam@49
|
79 Maybe<T> value;
|
cannam@49
|
80 };
|
cannam@49
|
81
|
cannam@49
|
82 class Event {
|
cannam@49
|
83 // An event waiting to be executed. Not for direct use by applications -- promises use this
|
cannam@49
|
84 // internally.
|
cannam@49
|
85
|
cannam@49
|
86 public:
|
cannam@49
|
87 Event();
|
cannam@49
|
88 ~Event() noexcept(false);
|
cannam@49
|
89 KJ_DISALLOW_COPY(Event);
|
cannam@49
|
90
|
cannam@49
|
91 void armDepthFirst();
|
cannam@49
|
92 // Enqueue this event so that `fire()` will be called from the event loop soon.
|
cannam@49
|
93 //
|
cannam@49
|
94 // Events scheduled in this way are executed in depth-first order: if an event callback arms
|
cannam@49
|
95 // more events, those events are placed at the front of the queue (in the order in which they
|
cannam@49
|
96 // were armed), so that they run immediately after the first event's callback returns.
|
cannam@49
|
97 //
|
cannam@49
|
98 // Depth-first event scheduling is appropriate for events that represent simple continuations
|
cannam@49
|
99 // of a previous event that should be globbed together for performance. Depth-first scheduling
|
cannam@49
|
100 // can lead to starvation, so any long-running task must occasionally yield with
|
cannam@49
|
101 // `armBreadthFirst()`. (Promise::then() uses depth-first whereas evalLater() uses
|
cannam@49
|
102 // breadth-first.)
|
cannam@49
|
103 //
|
cannam@49
|
104 // To use breadth-first scheduling instead, use `armBreadthFirst()`.
|
cannam@49
|
105
|
cannam@49
|
106 void armBreadthFirst();
|
cannam@49
|
107 // Like `armDepthFirst()` except that the event is placed at the end of the queue.
|
cannam@49
|
108
|
cannam@49
|
109 kj::String trace();
|
cannam@49
|
110 // Dump debug info about this event.
|
cannam@49
|
111
|
cannam@49
|
112 virtual _::PromiseNode* getInnerForTrace();
|
cannam@49
|
113 // If this event wraps a PromiseNode, get that node. Used for debug tracing.
|
cannam@49
|
114 // Default implementation returns nullptr.
|
cannam@49
|
115
|
cannam@49
|
116 protected:
|
cannam@49
|
117 virtual Maybe<Own<Event>> fire() = 0;
|
cannam@49
|
118 // Fire the event. Possibly returns a pointer to itself, which will be discarded by the
|
cannam@49
|
119 // caller. This is the only way that an event can delete itself as a result of firing, as
|
cannam@49
|
120 // doing so from within fire() will throw an exception.
|
cannam@49
|
121
|
cannam@49
|
122 private:
|
cannam@49
|
123 friend class kj::EventLoop;
|
cannam@49
|
124 EventLoop& loop;
|
cannam@49
|
125 Event* next;
|
cannam@49
|
126 Event** prev;
|
cannam@49
|
127 bool firing = false;
|
cannam@49
|
128 };
|
cannam@49
|
129
|
cannam@49
|
130 class PromiseNode {
|
cannam@49
|
131 // A Promise<T> contains a chain of PromiseNodes tracking the pending transformations.
|
cannam@49
|
132 //
|
cannam@49
|
133 // To reduce generated code bloat, PromiseNode is not a template. Instead, it makes very hacky
|
cannam@49
|
134 // use of pointers to ExceptionOrValue which actually point to ExceptionOr<T>, but are only
|
cannam@49
|
135 // so down-cast in the few places that really need to be templated. Luckily this is all
|
cannam@49
|
136 // internal implementation details.
|
cannam@49
|
137
|
cannam@49
|
138 public:
|
cannam@49
|
139 virtual void onReady(Event& event) noexcept = 0;
|
cannam@49
|
140 // Arms the given event when ready.
|
cannam@49
|
141
|
cannam@49
|
142 virtual void setSelfPointer(Own<PromiseNode>* selfPtr) noexcept;
|
cannam@49
|
143 // Tells the node that `selfPtr` is the pointer that owns this node, and will continue to own
|
cannam@49
|
144 // this node until it is destroyed or setSelfPointer() is called again. ChainPromiseNode uses
|
cannam@49
|
145 // this to shorten redundant chains. The default implementation does nothing; only
|
cannam@49
|
146 // ChainPromiseNode should implement this.
|
cannam@49
|
147
|
cannam@49
|
148 virtual void get(ExceptionOrValue& output) noexcept = 0;
|
cannam@49
|
149 // Get the result. `output` points to an ExceptionOr<T> into which the result will be written.
|
cannam@49
|
150 // Can only be called once, and only after the node is ready. Must be called directly from the
|
cannam@49
|
151 // event loop, with no application code on the stack.
|
cannam@49
|
152
|
cannam@49
|
153 virtual PromiseNode* getInnerForTrace();
|
cannam@49
|
154 // If this node wraps some other PromiseNode, get the wrapped node. Used for debug tracing.
|
cannam@49
|
155 // Default implementation returns nullptr.
|
cannam@49
|
156
|
cannam@49
|
157 protected:
|
cannam@49
|
158 class OnReadyEvent {
|
cannam@49
|
159 // Helper class for implementing onReady().
|
cannam@49
|
160
|
cannam@49
|
161 public:
|
cannam@49
|
162 void init(Event& newEvent);
|
cannam@49
|
163 // Returns true if arm() was already called.
|
cannam@49
|
164
|
cannam@49
|
165 void arm();
|
cannam@49
|
166 // Arms the event if init() has already been called and makes future calls to init() return
|
cannam@49
|
167 // true.
|
cannam@49
|
168
|
cannam@49
|
169 private:
|
cannam@49
|
170 Event* event = nullptr;
|
cannam@49
|
171 };
|
cannam@49
|
172 };
|
cannam@49
|
173
|
cannam@49
|
174 // -------------------------------------------------------------------
|
cannam@49
|
175
|
cannam@49
|
176 class ImmediatePromiseNodeBase: public PromiseNode {
|
cannam@49
|
177 public:
|
cannam@49
|
178 ImmediatePromiseNodeBase();
|
cannam@49
|
179 ~ImmediatePromiseNodeBase() noexcept(false);
|
cannam@49
|
180
|
cannam@49
|
181 void onReady(Event& event) noexcept override;
|
cannam@49
|
182 };
|
cannam@49
|
183
|
cannam@49
|
184 template <typename T>
|
cannam@49
|
185 class ImmediatePromiseNode final: public ImmediatePromiseNodeBase {
|
cannam@49
|
186 // A promise that has already been resolved to an immediate value or exception.
|
cannam@49
|
187
|
cannam@49
|
188 public:
|
cannam@49
|
189 ImmediatePromiseNode(ExceptionOr<T>&& result): result(kj::mv(result)) {}
|
cannam@49
|
190
|
cannam@49
|
191 void get(ExceptionOrValue& output) noexcept override {
|
cannam@49
|
192 output.as<T>() = kj::mv(result);
|
cannam@49
|
193 }
|
cannam@49
|
194
|
cannam@49
|
195 private:
|
cannam@49
|
196 ExceptionOr<T> result;
|
cannam@49
|
197 };
|
cannam@49
|
198
|
cannam@49
|
199 class ImmediateBrokenPromiseNode final: public ImmediatePromiseNodeBase {
|
cannam@49
|
200 public:
|
cannam@49
|
201 ImmediateBrokenPromiseNode(Exception&& exception);
|
cannam@49
|
202
|
cannam@49
|
203 void get(ExceptionOrValue& output) noexcept override;
|
cannam@49
|
204
|
cannam@49
|
205 private:
|
cannam@49
|
206 Exception exception;
|
cannam@49
|
207 };
|
cannam@49
|
208
|
cannam@49
|
209 // -------------------------------------------------------------------
|
cannam@49
|
210
|
cannam@49
|
211 class AttachmentPromiseNodeBase: public PromiseNode {
|
cannam@49
|
212 public:
|
cannam@49
|
213 AttachmentPromiseNodeBase(Own<PromiseNode>&& dependency);
|
cannam@49
|
214
|
cannam@49
|
215 void onReady(Event& event) noexcept override;
|
cannam@49
|
216 void get(ExceptionOrValue& output) noexcept override;
|
cannam@49
|
217 PromiseNode* getInnerForTrace() override;
|
cannam@49
|
218
|
cannam@49
|
219 private:
|
cannam@49
|
220 Own<PromiseNode> dependency;
|
cannam@49
|
221
|
cannam@49
|
222 void dropDependency();
|
cannam@49
|
223
|
cannam@49
|
224 template <typename>
|
cannam@49
|
225 friend class AttachmentPromiseNode;
|
cannam@49
|
226 };
|
cannam@49
|
227
|
cannam@49
|
228 template <typename Attachment>
|
cannam@49
|
229 class AttachmentPromiseNode final: public AttachmentPromiseNodeBase {
|
cannam@49
|
230 // A PromiseNode that holds on to some object (usually, an Own<T>, but could be any movable
|
cannam@49
|
231 // object) until the promise resolves.
|
cannam@49
|
232
|
cannam@49
|
233 public:
|
cannam@49
|
234 AttachmentPromiseNode(Own<PromiseNode>&& dependency, Attachment&& attachment)
|
cannam@49
|
235 : AttachmentPromiseNodeBase(kj::mv(dependency)),
|
cannam@49
|
236 attachment(kj::mv<Attachment>(attachment)) {}
|
cannam@49
|
237
|
cannam@49
|
238 ~AttachmentPromiseNode() noexcept(false) {
|
cannam@49
|
239 // We need to make sure the dependency is deleted before we delete the attachment because the
|
cannam@49
|
240 // dependency may be using the attachment.
|
cannam@49
|
241 dropDependency();
|
cannam@49
|
242 }
|
cannam@49
|
243
|
cannam@49
|
244 private:
|
cannam@49
|
245 Attachment attachment;
|
cannam@49
|
246 };
|
cannam@49
|
247
|
cannam@49
|
248 // -------------------------------------------------------------------
|
cannam@49
|
249
|
cannam@49
|
250 class PtmfHelper {
|
cannam@49
|
251 // This class is a private helper for GetFunctorStartAddress. The class represents the internal
|
cannam@49
|
252 // representation of a pointer-to-member-function.
|
cannam@49
|
253
|
cannam@49
|
254 template <typename... ParamTypes>
|
cannam@49
|
255 friend struct GetFunctorStartAddress;
|
cannam@49
|
256
|
cannam@49
|
257 #if __GNUG__
|
cannam@49
|
258 void* ptr;
|
cannam@49
|
259 ptrdiff_t adj;
|
cannam@49
|
260 // Layout of a pointer-to-member-function used by GCC and compatible compilers.
|
cannam@49
|
261 #else
|
cannam@49
|
262 #error "TODO(port): PTMF instruction address extraction"
|
cannam@49
|
263 #endif
|
cannam@49
|
264
|
cannam@49
|
265 #define BODY \
|
cannam@49
|
266 PtmfHelper result; \
|
cannam@49
|
267 static_assert(sizeof(p) == sizeof(result), "unknown ptmf layout"); \
|
cannam@49
|
268 memcpy(&result, &p, sizeof(result)); \
|
cannam@49
|
269 return result
|
cannam@49
|
270
|
cannam@49
|
271 template <typename R, typename C, typename... P, typename F>
|
cannam@49
|
272 static PtmfHelper from(F p) { BODY; }
|
cannam@49
|
273 // Create a PtmfHelper from some arbitrary pointer-to-member-function which is not
|
cannam@49
|
274 // overloaded nor a template. In this case the compiler is able to deduce the full function
|
cannam@49
|
275 // signature directly given the name since there is only one function with that name.
|
cannam@49
|
276
|
cannam@49
|
277 template <typename R, typename C, typename... P>
|
cannam@49
|
278 static PtmfHelper from(R (C::*p)(NoInfer<P>...)) { BODY; }
|
cannam@49
|
279 template <typename R, typename C, typename... P>
|
cannam@49
|
280 static PtmfHelper from(R (C::*p)(NoInfer<P>...) const) { BODY; }
|
cannam@49
|
281 // Create a PtmfHelper from some poniter-to-member-function which is a template. In this case
|
cannam@49
|
282 // the function must match exactly the containing type C, return type R, and parameter types P...
|
cannam@49
|
283 // GetFunctorStartAddress normally specifies exactly the correct C and R, but can only make a
|
cannam@49
|
284 // guess at P. Luckily, if the function parameters are template parameters then it's not
|
cannam@49
|
285 // necessary to be precise about P.
|
cannam@49
|
286 #undef BODY
|
cannam@49
|
287
|
cannam@49
|
288 void* apply(void* obj) {
|
cannam@49
|
289 #if defined(__arm__) || defined(__mips__) || defined(__aarch64__)
|
cannam@49
|
290 if (adj & 1) {
|
cannam@49
|
291 ptrdiff_t voff = (ptrdiff_t)ptr;
|
cannam@49
|
292 #else
|
cannam@49
|
293 ptrdiff_t voff = (ptrdiff_t)ptr;
|
cannam@49
|
294 if (voff & 1) {
|
cannam@49
|
295 voff &= ~1;
|
cannam@49
|
296 #endif
|
cannam@49
|
297 return *(void**)(*(char**)obj + voff);
|
cannam@49
|
298 } else {
|
cannam@49
|
299 return ptr;
|
cannam@49
|
300 }
|
cannam@49
|
301 }
|
cannam@49
|
302 };
|
cannam@49
|
303
|
cannam@49
|
304 template <typename... ParamTypes>
|
cannam@49
|
305 struct GetFunctorStartAddress {
|
cannam@49
|
306 // Given a functor (any object defining operator()), return the start address of the function,
|
cannam@49
|
307 // suitable for passing to addr2line to obtain a source file/line for debugging purposes.
|
cannam@49
|
308 //
|
cannam@49
|
309 // This turns out to be incredibly hard to implement in the presence of overloaded or templated
|
cannam@49
|
310 // functors. Therefore, we impose these specific restrictions, specific to our use case:
|
cannam@49
|
311 // - Overloading is not allowed, but templating is. (Generally we only intend to support lambdas
|
cannam@49
|
312 // anyway.)
|
cannam@49
|
313 // - The template parameters to GetFunctorStartAddress specify a hint as to the expected
|
cannam@49
|
314 // parameter types. If the functor is templated, its parameters must match exactly these types.
|
cannam@49
|
315 // (If it's not templated, ParamTypes are ignored.)
|
cannam@49
|
316
|
cannam@49
|
317 template <typename Func>
|
cannam@49
|
318 static void* apply(Func&& func) {
|
cannam@49
|
319 typedef decltype(func(instance<ParamTypes>()...)) ReturnType;
|
cannam@49
|
320 return PtmfHelper::from<ReturnType, Decay<Func>, ParamTypes...>(
|
cannam@49
|
321 &Decay<Func>::operator()).apply(&func);
|
cannam@49
|
322 }
|
cannam@49
|
323 };
|
cannam@49
|
324
|
cannam@49
|
325 template <>
|
cannam@49
|
326 struct GetFunctorStartAddress<Void&&>: public GetFunctorStartAddress<> {};
|
cannam@49
|
327 // Hack for TransformPromiseNode use case: an input type of `Void` indicates that the function
|
cannam@49
|
328 // actually has no parameters.
|
cannam@49
|
329
|
cannam@49
|
330 class TransformPromiseNodeBase: public PromiseNode {
|
cannam@49
|
331 public:
|
cannam@49
|
332 TransformPromiseNodeBase(Own<PromiseNode>&& dependency, void* continuationTracePtr);
|
cannam@49
|
333
|
cannam@49
|
334 void onReady(Event& event) noexcept override;
|
cannam@49
|
335 void get(ExceptionOrValue& output) noexcept override;
|
cannam@49
|
336 PromiseNode* getInnerForTrace() override;
|
cannam@49
|
337
|
cannam@49
|
338 private:
|
cannam@49
|
339 Own<PromiseNode> dependency;
|
cannam@49
|
340 void* continuationTracePtr;
|
cannam@49
|
341
|
cannam@49
|
342 void dropDependency();
|
cannam@49
|
343 void getDepResult(ExceptionOrValue& output);
|
cannam@49
|
344
|
cannam@49
|
345 virtual void getImpl(ExceptionOrValue& output) = 0;
|
cannam@49
|
346
|
cannam@49
|
347 template <typename, typename, typename, typename>
|
cannam@49
|
348 friend class TransformPromiseNode;
|
cannam@49
|
349 };
|
cannam@49
|
350
|
cannam@49
|
351 template <typename T, typename DepT, typename Func, typename ErrorFunc>
|
cannam@49
|
352 class TransformPromiseNode final: public TransformPromiseNodeBase {
|
cannam@49
|
353 // A PromiseNode that transforms the result of another PromiseNode through an application-provided
|
cannam@49
|
354 // function (implements `then()`).
|
cannam@49
|
355
|
cannam@49
|
356 public:
|
cannam@49
|
357 TransformPromiseNode(Own<PromiseNode>&& dependency, Func&& func, ErrorFunc&& errorHandler)
|
cannam@49
|
358 : TransformPromiseNodeBase(kj::mv(dependency),
|
cannam@49
|
359 GetFunctorStartAddress<DepT&&>::apply(func)),
|
cannam@49
|
360 func(kj::fwd<Func>(func)), errorHandler(kj::fwd<ErrorFunc>(errorHandler)) {}
|
cannam@49
|
361
|
cannam@49
|
362 ~TransformPromiseNode() noexcept(false) {
|
cannam@49
|
363 // We need to make sure the dependency is deleted before we delete the continuations because it
|
cannam@49
|
364 // is a common pattern for the continuations to hold ownership of objects that might be in-use
|
cannam@49
|
365 // by the dependency.
|
cannam@49
|
366 dropDependency();
|
cannam@49
|
367 }
|
cannam@49
|
368
|
cannam@49
|
369 private:
|
cannam@49
|
370 Func func;
|
cannam@49
|
371 ErrorFunc errorHandler;
|
cannam@49
|
372
|
cannam@49
|
373 void getImpl(ExceptionOrValue& output) override {
|
cannam@49
|
374 ExceptionOr<DepT> depResult;
|
cannam@49
|
375 getDepResult(depResult);
|
cannam@49
|
376 KJ_IF_MAYBE(depException, depResult.exception) {
|
cannam@49
|
377 output.as<T>() = handle(
|
cannam@49
|
378 MaybeVoidCaller<Exception, FixVoid<ReturnType<ErrorFunc, Exception>>>::apply(
|
cannam@49
|
379 errorHandler, kj::mv(*depException)));
|
cannam@49
|
380 } else KJ_IF_MAYBE(depValue, depResult.value) {
|
cannam@49
|
381 output.as<T>() = handle(MaybeVoidCaller<DepT, T>::apply(func, kj::mv(*depValue)));
|
cannam@49
|
382 }
|
cannam@49
|
383 }
|
cannam@49
|
384
|
cannam@49
|
385 ExceptionOr<T> handle(T&& value) {
|
cannam@49
|
386 return kj::mv(value);
|
cannam@49
|
387 }
|
cannam@49
|
388 ExceptionOr<T> handle(PropagateException::Bottom&& value) {
|
cannam@49
|
389 return ExceptionOr<T>(false, value.asException());
|
cannam@49
|
390 }
|
cannam@49
|
391 };
|
cannam@49
|
392
|
cannam@49
|
393 // -------------------------------------------------------------------
|
cannam@49
|
394
|
cannam@49
|
395 class ForkHubBase;
|
cannam@49
|
396
|
cannam@49
|
397 class ForkBranchBase: public PromiseNode {
|
cannam@49
|
398 public:
|
cannam@49
|
399 ForkBranchBase(Own<ForkHubBase>&& hub);
|
cannam@49
|
400 ~ForkBranchBase() noexcept(false);
|
cannam@49
|
401
|
cannam@49
|
402 void hubReady() noexcept;
|
cannam@49
|
403 // Called by the hub to indicate that it is ready.
|
cannam@49
|
404
|
cannam@49
|
405 // implements PromiseNode ------------------------------------------
|
cannam@49
|
406 void onReady(Event& event) noexcept override;
|
cannam@49
|
407 PromiseNode* getInnerForTrace() override;
|
cannam@49
|
408
|
cannam@49
|
409 protected:
|
cannam@49
|
410 inline ExceptionOrValue& getHubResultRef();
|
cannam@49
|
411
|
cannam@49
|
412 void releaseHub(ExceptionOrValue& output);
|
cannam@49
|
413 // Release the hub. If an exception is thrown, add it to `output`.
|
cannam@49
|
414
|
cannam@49
|
415 private:
|
cannam@49
|
416 OnReadyEvent onReadyEvent;
|
cannam@49
|
417
|
cannam@49
|
418 Own<ForkHubBase> hub;
|
cannam@49
|
419 ForkBranchBase* next = nullptr;
|
cannam@49
|
420 ForkBranchBase** prevPtr = nullptr;
|
cannam@49
|
421
|
cannam@49
|
422 friend class ForkHubBase;
|
cannam@49
|
423 };
|
cannam@49
|
424
|
cannam@49
|
425 template <typename T> T copyOrAddRef(T& t) { return t; }
|
cannam@49
|
426 template <typename T> Own<T> copyOrAddRef(Own<T>& t) { return t->addRef(); }
|
cannam@49
|
427
|
cannam@49
|
428 template <typename T>
|
cannam@49
|
429 class ForkBranch final: public ForkBranchBase {
|
cannam@49
|
430 // A PromiseNode that implements one branch of a fork -- i.e. one of the branches that receives
|
cannam@49
|
431 // a const reference.
|
cannam@49
|
432
|
cannam@49
|
433 public:
|
cannam@49
|
434 ForkBranch(Own<ForkHubBase>&& hub): ForkBranchBase(kj::mv(hub)) {}
|
cannam@49
|
435
|
cannam@49
|
436 void get(ExceptionOrValue& output) noexcept override {
|
cannam@49
|
437 ExceptionOr<T>& hubResult = getHubResultRef().template as<T>();
|
cannam@49
|
438 KJ_IF_MAYBE(value, hubResult.value) {
|
cannam@49
|
439 output.as<T>().value = copyOrAddRef(*value);
|
cannam@49
|
440 } else {
|
cannam@49
|
441 output.as<T>().value = nullptr;
|
cannam@49
|
442 }
|
cannam@49
|
443 output.exception = hubResult.exception;
|
cannam@49
|
444 releaseHub(output);
|
cannam@49
|
445 }
|
cannam@49
|
446 };
|
cannam@49
|
447
|
cannam@49
|
448 template <typename T, size_t index>
|
cannam@49
|
449 class SplitBranch final: public ForkBranchBase {
|
cannam@49
|
450 // A PromiseNode that implements one branch of a fork -- i.e. one of the branches that receives
|
cannam@49
|
451 // a const reference.
|
cannam@49
|
452
|
cannam@49
|
453 public:
|
cannam@49
|
454 SplitBranch(Own<ForkHubBase>&& hub): ForkBranchBase(kj::mv(hub)) {}
|
cannam@49
|
455
|
cannam@49
|
456 typedef kj::Decay<decltype(kj::get<index>(kj::instance<T>()))> Element;
|
cannam@49
|
457
|
cannam@49
|
458 void get(ExceptionOrValue& output) noexcept override {
|
cannam@49
|
459 ExceptionOr<T>& hubResult = getHubResultRef().template as<T>();
|
cannam@49
|
460 KJ_IF_MAYBE(value, hubResult.value) {
|
cannam@49
|
461 output.as<Element>().value = kj::mv(kj::get<index>(*value));
|
cannam@49
|
462 } else {
|
cannam@49
|
463 output.as<Element>().value = nullptr;
|
cannam@49
|
464 }
|
cannam@49
|
465 output.exception = hubResult.exception;
|
cannam@49
|
466 releaseHub(output);
|
cannam@49
|
467 }
|
cannam@49
|
468 };
|
cannam@49
|
469
|
cannam@49
|
470 // -------------------------------------------------------------------
|
cannam@49
|
471
|
cannam@49
|
472 class ForkHubBase: public Refcounted, protected Event {
|
cannam@49
|
473 public:
|
cannam@49
|
474 ForkHubBase(Own<PromiseNode>&& inner, ExceptionOrValue& resultRef);
|
cannam@49
|
475
|
cannam@49
|
476 inline ExceptionOrValue& getResultRef() { return resultRef; }
|
cannam@49
|
477
|
cannam@49
|
478 private:
|
cannam@49
|
479 Own<PromiseNode> inner;
|
cannam@49
|
480 ExceptionOrValue& resultRef;
|
cannam@49
|
481
|
cannam@49
|
482 ForkBranchBase* headBranch = nullptr;
|
cannam@49
|
483 ForkBranchBase** tailBranch = &headBranch;
|
cannam@49
|
484 // Tail becomes null once the inner promise is ready and all branches have been notified.
|
cannam@49
|
485
|
cannam@49
|
486 Maybe<Own<Event>> fire() override;
|
cannam@49
|
487 _::PromiseNode* getInnerForTrace() override;
|
cannam@49
|
488
|
cannam@49
|
489 friend class ForkBranchBase;
|
cannam@49
|
490 };
|
cannam@49
|
491
|
cannam@49
|
492 template <typename T>
|
cannam@49
|
493 class ForkHub final: public ForkHubBase {
|
cannam@49
|
494 // A PromiseNode that implements the hub of a fork. The first call to Promise::fork() replaces
|
cannam@49
|
495 // the promise's outer node with a ForkHub, and subsequent calls add branches to that hub (if
|
cannam@49
|
496 // possible).
|
cannam@49
|
497
|
cannam@49
|
498 public:
|
cannam@49
|
499 ForkHub(Own<PromiseNode>&& inner): ForkHubBase(kj::mv(inner), result) {}
|
cannam@49
|
500
|
cannam@49
|
501 Promise<_::UnfixVoid<T>> addBranch() {
|
cannam@49
|
502 return Promise<_::UnfixVoid<T>>(false, kj::heap<ForkBranch<T>>(addRef(*this)));
|
cannam@49
|
503 }
|
cannam@49
|
504
|
cannam@49
|
505 _::SplitTuplePromise<T> split() {
|
cannam@49
|
506 return splitImpl(MakeIndexes<tupleSize<T>()>());
|
cannam@49
|
507 }
|
cannam@49
|
508
|
cannam@49
|
509 private:
|
cannam@49
|
510 ExceptionOr<T> result;
|
cannam@49
|
511
|
cannam@49
|
512 template <size_t... indexes>
|
cannam@49
|
513 _::SplitTuplePromise<T> splitImpl(Indexes<indexes...>) {
|
cannam@49
|
514 return kj::tuple(addSplit<indexes>()...);
|
cannam@49
|
515 }
|
cannam@49
|
516
|
cannam@49
|
517 template <size_t index>
|
cannam@49
|
518 Promise<JoinPromises<typename SplitBranch<T, index>::Element>> addSplit() {
|
cannam@49
|
519 return Promise<JoinPromises<typename SplitBranch<T, index>::Element>>(
|
cannam@49
|
520 false, maybeChain(kj::heap<SplitBranch<T, index>>(addRef(*this)),
|
cannam@49
|
521 implicitCast<typename SplitBranch<T, index>::Element*>(nullptr)));
|
cannam@49
|
522 }
|
cannam@49
|
523 };
|
cannam@49
|
524
|
cannam@49
|
525 inline ExceptionOrValue& ForkBranchBase::getHubResultRef() {
|
cannam@49
|
526 return hub->getResultRef();
|
cannam@49
|
527 }
|
cannam@49
|
528
|
cannam@49
|
529 // -------------------------------------------------------------------
|
cannam@49
|
530
|
cannam@49
|
531 class ChainPromiseNode final: public PromiseNode, public Event {
|
cannam@49
|
532 // Promise node which reduces Promise<Promise<T>> to Promise<T>.
|
cannam@49
|
533 //
|
cannam@49
|
534 // `Event` is only a public base class because otherwise we can't cast Own<ChainPromiseNode> to
|
cannam@49
|
535 // Own<Event>. Ugh, templates and private...
|
cannam@49
|
536
|
cannam@49
|
537 public:
|
cannam@49
|
538 explicit ChainPromiseNode(Own<PromiseNode> inner);
|
cannam@49
|
539 ~ChainPromiseNode() noexcept(false);
|
cannam@49
|
540
|
cannam@49
|
541 void onReady(Event& event) noexcept override;
|
cannam@49
|
542 void setSelfPointer(Own<PromiseNode>* selfPtr) noexcept override;
|
cannam@49
|
543 void get(ExceptionOrValue& output) noexcept override;
|
cannam@49
|
544 PromiseNode* getInnerForTrace() override;
|
cannam@49
|
545
|
cannam@49
|
546 private:
|
cannam@49
|
547 enum State {
|
cannam@49
|
548 STEP1,
|
cannam@49
|
549 STEP2
|
cannam@49
|
550 };
|
cannam@49
|
551
|
cannam@49
|
552 State state;
|
cannam@49
|
553
|
cannam@49
|
554 Own<PromiseNode> inner;
|
cannam@49
|
555 // In STEP1, a PromiseNode for a Promise<T>.
|
cannam@49
|
556 // In STEP2, a PromiseNode for a T.
|
cannam@49
|
557
|
cannam@49
|
558 Event* onReadyEvent = nullptr;
|
cannam@49
|
559 Own<PromiseNode>* selfPtr = nullptr;
|
cannam@49
|
560
|
cannam@49
|
561 Maybe<Own<Event>> fire() override;
|
cannam@49
|
562 };
|
cannam@49
|
563
|
cannam@49
|
564 template <typename T>
|
cannam@49
|
565 Own<PromiseNode> maybeChain(Own<PromiseNode>&& node, Promise<T>*) {
|
cannam@49
|
566 return heap<ChainPromiseNode>(kj::mv(node));
|
cannam@49
|
567 }
|
cannam@49
|
568
|
cannam@49
|
569 template <typename T>
|
cannam@49
|
570 Own<PromiseNode>&& maybeChain(Own<PromiseNode>&& node, T*) {
|
cannam@49
|
571 return kj::mv(node);
|
cannam@49
|
572 }
|
cannam@49
|
573
|
cannam@49
|
574 // -------------------------------------------------------------------
|
cannam@49
|
575
|
cannam@49
|
576 class ExclusiveJoinPromiseNode final: public PromiseNode {
|
cannam@49
|
577 public:
|
cannam@49
|
578 ExclusiveJoinPromiseNode(Own<PromiseNode> left, Own<PromiseNode> right);
|
cannam@49
|
579 ~ExclusiveJoinPromiseNode() noexcept(false);
|
cannam@49
|
580
|
cannam@49
|
581 void onReady(Event& event) noexcept override;
|
cannam@49
|
582 void get(ExceptionOrValue& output) noexcept override;
|
cannam@49
|
583 PromiseNode* getInnerForTrace() override;
|
cannam@49
|
584
|
cannam@49
|
585 private:
|
cannam@49
|
586 class Branch: public Event {
|
cannam@49
|
587 public:
|
cannam@49
|
588 Branch(ExclusiveJoinPromiseNode& joinNode, Own<PromiseNode> dependency);
|
cannam@49
|
589 ~Branch() noexcept(false);
|
cannam@49
|
590
|
cannam@49
|
591 bool get(ExceptionOrValue& output);
|
cannam@49
|
592 // Returns true if this is the side that finished.
|
cannam@49
|
593
|
cannam@49
|
594 Maybe<Own<Event>> fire() override;
|
cannam@49
|
595 _::PromiseNode* getInnerForTrace() override;
|
cannam@49
|
596
|
cannam@49
|
597 private:
|
cannam@49
|
598 ExclusiveJoinPromiseNode& joinNode;
|
cannam@49
|
599 Own<PromiseNode> dependency;
|
cannam@49
|
600 };
|
cannam@49
|
601
|
cannam@49
|
602 Branch left;
|
cannam@49
|
603 Branch right;
|
cannam@49
|
604 OnReadyEvent onReadyEvent;
|
cannam@49
|
605 };
|
cannam@49
|
606
|
cannam@49
|
607 // -------------------------------------------------------------------
|
cannam@49
|
608
|
cannam@49
|
609 class ArrayJoinPromiseNodeBase: public PromiseNode {
|
cannam@49
|
610 public:
|
cannam@49
|
611 ArrayJoinPromiseNodeBase(Array<Own<PromiseNode>> promises,
|
cannam@49
|
612 ExceptionOrValue* resultParts, size_t partSize);
|
cannam@49
|
613 ~ArrayJoinPromiseNodeBase() noexcept(false);
|
cannam@49
|
614
|
cannam@49
|
615 void onReady(Event& event) noexcept override final;
|
cannam@49
|
616 void get(ExceptionOrValue& output) noexcept override final;
|
cannam@49
|
617 PromiseNode* getInnerForTrace() override final;
|
cannam@49
|
618
|
cannam@49
|
619 protected:
|
cannam@49
|
620 virtual void getNoError(ExceptionOrValue& output) noexcept = 0;
|
cannam@49
|
621 // Called to compile the result only in the case where there were no errors.
|
cannam@49
|
622
|
cannam@49
|
623 private:
|
cannam@49
|
624 uint countLeft;
|
cannam@49
|
625 OnReadyEvent onReadyEvent;
|
cannam@49
|
626
|
cannam@49
|
627 class Branch final: public Event {
|
cannam@49
|
628 public:
|
cannam@49
|
629 Branch(ArrayJoinPromiseNodeBase& joinNode, Own<PromiseNode> dependency,
|
cannam@49
|
630 ExceptionOrValue& output);
|
cannam@49
|
631 ~Branch() noexcept(false);
|
cannam@49
|
632
|
cannam@49
|
633 Maybe<Own<Event>> fire() override;
|
cannam@49
|
634 _::PromiseNode* getInnerForTrace() override;
|
cannam@49
|
635
|
cannam@49
|
636 Maybe<Exception> getPart();
|
cannam@49
|
637 // Calls dependency->get(output). If there was an exception, return it.
|
cannam@49
|
638
|
cannam@49
|
639 private:
|
cannam@49
|
640 ArrayJoinPromiseNodeBase& joinNode;
|
cannam@49
|
641 Own<PromiseNode> dependency;
|
cannam@49
|
642 ExceptionOrValue& output;
|
cannam@49
|
643 };
|
cannam@49
|
644
|
cannam@49
|
645 Array<Branch> branches;
|
cannam@49
|
646 };
|
cannam@49
|
647
|
cannam@49
|
648 template <typename T>
|
cannam@49
|
649 class ArrayJoinPromiseNode final: public ArrayJoinPromiseNodeBase {
|
cannam@49
|
650 public:
|
cannam@49
|
651 ArrayJoinPromiseNode(Array<Own<PromiseNode>> promises,
|
cannam@49
|
652 Array<ExceptionOr<T>> resultParts)
|
cannam@49
|
653 : ArrayJoinPromiseNodeBase(kj::mv(promises), resultParts.begin(), sizeof(ExceptionOr<T>)),
|
cannam@49
|
654 resultParts(kj::mv(resultParts)) {}
|
cannam@49
|
655
|
cannam@49
|
656 protected:
|
cannam@49
|
657 void getNoError(ExceptionOrValue& output) noexcept override {
|
cannam@49
|
658 auto builder = heapArrayBuilder<T>(resultParts.size());
|
cannam@49
|
659 for (auto& part: resultParts) {
|
cannam@49
|
660 KJ_IASSERT(part.value != nullptr,
|
cannam@49
|
661 "Bug in KJ promise framework: Promise result had neither value no exception.");
|
cannam@49
|
662 builder.add(kj::mv(*_::readMaybe(part.value)));
|
cannam@49
|
663 }
|
cannam@49
|
664 output.as<Array<T>>() = builder.finish();
|
cannam@49
|
665 }
|
cannam@49
|
666
|
cannam@49
|
667 private:
|
cannam@49
|
668 Array<ExceptionOr<T>> resultParts;
|
cannam@49
|
669 };
|
cannam@49
|
670
|
cannam@49
|
671 template <>
|
cannam@49
|
672 class ArrayJoinPromiseNode<void> final: public ArrayJoinPromiseNodeBase {
|
cannam@49
|
673 public:
|
cannam@49
|
674 ArrayJoinPromiseNode(Array<Own<PromiseNode>> promises,
|
cannam@49
|
675 Array<ExceptionOr<_::Void>> resultParts);
|
cannam@49
|
676 ~ArrayJoinPromiseNode();
|
cannam@49
|
677
|
cannam@49
|
678 protected:
|
cannam@49
|
679 void getNoError(ExceptionOrValue& output) noexcept override;
|
cannam@49
|
680
|
cannam@49
|
681 private:
|
cannam@49
|
682 Array<ExceptionOr<_::Void>> resultParts;
|
cannam@49
|
683 };
|
cannam@49
|
684
|
cannam@49
|
685 // -------------------------------------------------------------------
|
cannam@49
|
686
|
cannam@49
|
687 class EagerPromiseNodeBase: public PromiseNode, protected Event {
|
cannam@49
|
688 // A PromiseNode that eagerly evaluates its dependency even if its dependent does not eagerly
|
cannam@49
|
689 // evaluate it.
|
cannam@49
|
690
|
cannam@49
|
691 public:
|
cannam@49
|
692 EagerPromiseNodeBase(Own<PromiseNode>&& dependency, ExceptionOrValue& resultRef);
|
cannam@49
|
693
|
cannam@49
|
694 void onReady(Event& event) noexcept override;
|
cannam@49
|
695 PromiseNode* getInnerForTrace() override;
|
cannam@49
|
696
|
cannam@49
|
697 private:
|
cannam@49
|
698 Own<PromiseNode> dependency;
|
cannam@49
|
699 OnReadyEvent onReadyEvent;
|
cannam@49
|
700
|
cannam@49
|
701 ExceptionOrValue& resultRef;
|
cannam@49
|
702
|
cannam@49
|
703 Maybe<Own<Event>> fire() override;
|
cannam@49
|
704 };
|
cannam@49
|
705
|
cannam@49
|
706 template <typename T>
|
cannam@49
|
707 class EagerPromiseNode final: public EagerPromiseNodeBase {
|
cannam@49
|
708 public:
|
cannam@49
|
709 EagerPromiseNode(Own<PromiseNode>&& dependency)
|
cannam@49
|
710 : EagerPromiseNodeBase(kj::mv(dependency), result) {}
|
cannam@49
|
711
|
cannam@49
|
712 void get(ExceptionOrValue& output) noexcept override {
|
cannam@49
|
713 output.as<T>() = kj::mv(result);
|
cannam@49
|
714 }
|
cannam@49
|
715
|
cannam@49
|
716 private:
|
cannam@49
|
717 ExceptionOr<T> result;
|
cannam@49
|
718 };
|
cannam@49
|
719
|
cannam@49
|
720 template <typename T>
|
cannam@49
|
721 Own<PromiseNode> spark(Own<PromiseNode>&& node) {
|
cannam@49
|
722 // Forces evaluation of the given node to begin as soon as possible, even if no one is waiting
|
cannam@49
|
723 // on it.
|
cannam@49
|
724 return heap<EagerPromiseNode<T>>(kj::mv(node));
|
cannam@49
|
725 }
|
cannam@49
|
726
|
cannam@49
|
727 // -------------------------------------------------------------------
|
cannam@49
|
728
|
cannam@49
|
729 class AdapterPromiseNodeBase: public PromiseNode {
|
cannam@49
|
730 public:
|
cannam@49
|
731 void onReady(Event& event) noexcept override;
|
cannam@49
|
732
|
cannam@49
|
733 protected:
|
cannam@49
|
734 inline void setReady() {
|
cannam@49
|
735 onReadyEvent.arm();
|
cannam@49
|
736 }
|
cannam@49
|
737
|
cannam@49
|
738 private:
|
cannam@49
|
739 OnReadyEvent onReadyEvent;
|
cannam@49
|
740 };
|
cannam@49
|
741
|
cannam@49
|
742 template <typename T, typename Adapter>
|
cannam@49
|
743 class AdapterPromiseNode final: public AdapterPromiseNodeBase,
|
cannam@49
|
744 private PromiseFulfiller<UnfixVoid<T>> {
|
cannam@49
|
745 // A PromiseNode that wraps a PromiseAdapter.
|
cannam@49
|
746
|
cannam@49
|
747 public:
|
cannam@49
|
748 template <typename... Params>
|
cannam@49
|
749 AdapterPromiseNode(Params&&... params)
|
cannam@49
|
750 : adapter(static_cast<PromiseFulfiller<UnfixVoid<T>>&>(*this), kj::fwd<Params>(params)...) {}
|
cannam@49
|
751
|
cannam@49
|
752 void get(ExceptionOrValue& output) noexcept override {
|
cannam@49
|
753 KJ_IREQUIRE(!isWaiting());
|
cannam@49
|
754 output.as<T>() = kj::mv(result);
|
cannam@49
|
755 }
|
cannam@49
|
756
|
cannam@49
|
757 private:
|
cannam@49
|
758 ExceptionOr<T> result;
|
cannam@49
|
759 bool waiting = true;
|
cannam@49
|
760 Adapter adapter;
|
cannam@49
|
761
|
cannam@49
|
762 void fulfill(T&& value) override {
|
cannam@49
|
763 if (waiting) {
|
cannam@49
|
764 waiting = false;
|
cannam@49
|
765 result = ExceptionOr<T>(kj::mv(value));
|
cannam@49
|
766 setReady();
|
cannam@49
|
767 }
|
cannam@49
|
768 }
|
cannam@49
|
769
|
cannam@49
|
770 void reject(Exception&& exception) override {
|
cannam@49
|
771 if (waiting) {
|
cannam@49
|
772 waiting = false;
|
cannam@49
|
773 result = ExceptionOr<T>(false, kj::mv(exception));
|
cannam@49
|
774 setReady();
|
cannam@49
|
775 }
|
cannam@49
|
776 }
|
cannam@49
|
777
|
cannam@49
|
778 bool isWaiting() override {
|
cannam@49
|
779 return waiting;
|
cannam@49
|
780 }
|
cannam@49
|
781 };
|
cannam@49
|
782
|
cannam@49
|
783 } // namespace _ (private)
|
cannam@49
|
784
|
cannam@49
|
785 // =======================================================================================
|
cannam@49
|
786
|
cannam@49
|
787 template <typename T>
|
cannam@49
|
788 Promise<T>::Promise(_::FixVoid<T> value)
|
cannam@49
|
789 : PromiseBase(heap<_::ImmediatePromiseNode<_::FixVoid<T>>>(kj::mv(value))) {}
|
cannam@49
|
790
|
cannam@49
|
791 template <typename T>
|
cannam@49
|
792 Promise<T>::Promise(kj::Exception&& exception)
|
cannam@49
|
793 : PromiseBase(heap<_::ImmediateBrokenPromiseNode>(kj::mv(exception))) {}
|
cannam@49
|
794
|
cannam@49
|
795 template <typename T>
|
cannam@49
|
796 template <typename Func, typename ErrorFunc>
|
cannam@49
|
797 PromiseForResult<Func, T> Promise<T>::then(Func&& func, ErrorFunc&& errorHandler) {
|
cannam@49
|
798 typedef _::FixVoid<_::ReturnType<Func, T>> ResultT;
|
cannam@49
|
799
|
cannam@49
|
800 Own<_::PromiseNode> intermediate =
|
cannam@49
|
801 heap<_::TransformPromiseNode<ResultT, _::FixVoid<T>, Func, ErrorFunc>>(
|
cannam@49
|
802 kj::mv(node), kj::fwd<Func>(func), kj::fwd<ErrorFunc>(errorHandler));
|
cannam@49
|
803 return PromiseForResult<Func, T>(false,
|
cannam@49
|
804 _::maybeChain(kj::mv(intermediate), implicitCast<ResultT*>(nullptr)));
|
cannam@49
|
805 }
|
cannam@49
|
806
|
cannam@49
|
807 namespace _ { // private
|
cannam@49
|
808
|
cannam@49
|
809 template <typename T>
|
cannam@49
|
810 struct IdentityFunc {
|
cannam@49
|
811 inline T operator()(T&& value) const {
|
cannam@49
|
812 return kj::mv(value);
|
cannam@49
|
813 }
|
cannam@49
|
814 };
|
cannam@49
|
815 template <typename T>
|
cannam@49
|
816 struct IdentityFunc<Promise<T>> {
|
cannam@49
|
817 inline Promise<T> operator()(T&& value) const {
|
cannam@49
|
818 return kj::mv(value);
|
cannam@49
|
819 }
|
cannam@49
|
820 };
|
cannam@49
|
821 template <>
|
cannam@49
|
822 struct IdentityFunc<void> {
|
cannam@49
|
823 inline void operator()() const {}
|
cannam@49
|
824 };
|
cannam@49
|
825 template <>
|
cannam@49
|
826 struct IdentityFunc<Promise<void>> {
|
cannam@49
|
827 Promise<void> operator()() const;
|
cannam@49
|
828 // This can't be inline because it will make the translation unit depend on kj-async. Awkwardly,
|
cannam@49
|
829 // Cap'n Proto relies on being able to include this header without creating such a link-time
|
cannam@49
|
830 // dependency.
|
cannam@49
|
831 };
|
cannam@49
|
832
|
cannam@49
|
833 } // namespace _ (private)
|
cannam@49
|
834
|
cannam@49
|
835 template <typename T>
|
cannam@49
|
836 template <typename ErrorFunc>
|
cannam@49
|
837 Promise<T> Promise<T>::catch_(ErrorFunc&& errorHandler) {
|
cannam@49
|
838 // then()'s ErrorFunc can only return a Promise if Func also returns a Promise. In this case,
|
cannam@49
|
839 // Func is being filled in automatically. We want to make sure ErrorFunc can return a Promise,
|
cannam@49
|
840 // but we don't want the extra overhead of promise chaining if ErrorFunc doesn't actually
|
cannam@49
|
841 // return a promise. So we make our Func return match ErrorFunc.
|
cannam@49
|
842 return then(_::IdentityFunc<decltype(errorHandler(instance<Exception&&>()))>(),
|
cannam@49
|
843 kj::fwd<ErrorFunc>(errorHandler));
|
cannam@49
|
844 }
|
cannam@49
|
845
|
cannam@49
|
846 template <typename T>
|
cannam@49
|
847 T Promise<T>::wait(WaitScope& waitScope) {
|
cannam@49
|
848 _::ExceptionOr<_::FixVoid<T>> result;
|
cannam@49
|
849
|
cannam@49
|
850 waitImpl(kj::mv(node), result, waitScope);
|
cannam@49
|
851
|
cannam@49
|
852 KJ_IF_MAYBE(value, result.value) {
|
cannam@49
|
853 KJ_IF_MAYBE(exception, result.exception) {
|
cannam@49
|
854 throwRecoverableException(kj::mv(*exception));
|
cannam@49
|
855 }
|
cannam@49
|
856 return _::returnMaybeVoid(kj::mv(*value));
|
cannam@49
|
857 } else KJ_IF_MAYBE(exception, result.exception) {
|
cannam@49
|
858 throwFatalException(kj::mv(*exception));
|
cannam@49
|
859 } else {
|
cannam@49
|
860 // Result contained neither a value nor an exception?
|
cannam@49
|
861 KJ_UNREACHABLE;
|
cannam@49
|
862 }
|
cannam@49
|
863 }
|
cannam@49
|
864
|
cannam@49
|
865 template <typename T>
|
cannam@49
|
866 ForkedPromise<T> Promise<T>::fork() {
|
cannam@49
|
867 return ForkedPromise<T>(false, refcounted<_::ForkHub<_::FixVoid<T>>>(kj::mv(node)));
|
cannam@49
|
868 }
|
cannam@49
|
869
|
cannam@49
|
870 template <typename T>
|
cannam@49
|
871 Promise<T> ForkedPromise<T>::addBranch() {
|
cannam@49
|
872 return hub->addBranch();
|
cannam@49
|
873 }
|
cannam@49
|
874
|
cannam@49
|
875 template <typename T>
|
cannam@49
|
876 _::SplitTuplePromise<T> Promise<T>::split() {
|
cannam@49
|
877 return refcounted<_::ForkHub<_::FixVoid<T>>>(kj::mv(node))->split();
|
cannam@49
|
878 }
|
cannam@49
|
879
|
cannam@49
|
880 template <typename T>
|
cannam@49
|
881 Promise<T> Promise<T>::exclusiveJoin(Promise<T>&& other) {
|
cannam@49
|
882 return Promise(false, heap<_::ExclusiveJoinPromiseNode>(kj::mv(node), kj::mv(other.node)));
|
cannam@49
|
883 }
|
cannam@49
|
884
|
cannam@49
|
885 template <typename T>
|
cannam@49
|
886 template <typename... Attachments>
|
cannam@49
|
887 Promise<T> Promise<T>::attach(Attachments&&... attachments) {
|
cannam@49
|
888 return Promise(false, kj::heap<_::AttachmentPromiseNode<Tuple<Attachments...>>>(
|
cannam@49
|
889 kj::mv(node), kj::tuple(kj::fwd<Attachments>(attachments)...)));
|
cannam@49
|
890 }
|
cannam@49
|
891
|
cannam@49
|
892 template <typename T>
|
cannam@49
|
893 template <typename ErrorFunc>
|
cannam@49
|
894 Promise<T> Promise<T>::eagerlyEvaluate(ErrorFunc&& errorHandler) {
|
cannam@49
|
895 // See catch_() for commentary.
|
cannam@49
|
896 return Promise(false, _::spark<_::FixVoid<T>>(then(
|
cannam@49
|
897 _::IdentityFunc<decltype(errorHandler(instance<Exception&&>()))>(),
|
cannam@49
|
898 kj::fwd<ErrorFunc>(errorHandler)).node));
|
cannam@49
|
899 }
|
cannam@49
|
900
|
cannam@49
|
901 template <typename T>
|
cannam@49
|
902 Promise<T> Promise<T>::eagerlyEvaluate(decltype(nullptr)) {
|
cannam@49
|
903 return Promise(false, _::spark<_::FixVoid<T>>(kj::mv(node)));
|
cannam@49
|
904 }
|
cannam@49
|
905
|
cannam@49
|
906 template <typename T>
|
cannam@49
|
907 kj::String Promise<T>::trace() {
|
cannam@49
|
908 return PromiseBase::trace();
|
cannam@49
|
909 }
|
cannam@49
|
910
|
cannam@49
|
911 template <typename Func>
|
cannam@49
|
912 inline PromiseForResult<Func, void> evalLater(Func&& func) {
|
cannam@49
|
913 return _::yield().then(kj::fwd<Func>(func), _::PropagateException());
|
cannam@49
|
914 }
|
cannam@49
|
915
|
cannam@49
|
916 template <typename Func>
|
cannam@49
|
917 inline PromiseForResult<Func, void> evalNow(Func&& func) {
|
cannam@49
|
918 PromiseForResult<Func, void> result = nullptr;
|
cannam@49
|
919 KJ_IF_MAYBE(e, kj::runCatchingExceptions([&]() {
|
cannam@49
|
920 result = func();
|
cannam@49
|
921 })) {
|
cannam@49
|
922 result = kj::mv(*e);
|
cannam@49
|
923 }
|
cannam@49
|
924 return result;
|
cannam@49
|
925 }
|
cannam@49
|
926
|
cannam@49
|
927 template <typename T>
|
cannam@49
|
928 template <typename ErrorFunc>
|
cannam@49
|
929 void Promise<T>::detach(ErrorFunc&& errorHandler) {
|
cannam@49
|
930 return _::detach(then([](T&&) {}, kj::fwd<ErrorFunc>(errorHandler)));
|
cannam@49
|
931 }
|
cannam@49
|
932
|
cannam@49
|
933 template <>
|
cannam@49
|
934 template <typename ErrorFunc>
|
cannam@49
|
935 void Promise<void>::detach(ErrorFunc&& errorHandler) {
|
cannam@49
|
936 return _::detach(then([]() {}, kj::fwd<ErrorFunc>(errorHandler)));
|
cannam@49
|
937 }
|
cannam@49
|
938
|
cannam@49
|
939 template <typename T>
|
cannam@49
|
940 Promise<Array<T>> joinPromises(Array<Promise<T>>&& promises) {
|
cannam@49
|
941 return Promise<Array<T>>(false, kj::heap<_::ArrayJoinPromiseNode<T>>(
|
cannam@49
|
942 KJ_MAP(p, promises) { return kj::mv(p.node); },
|
cannam@49
|
943 heapArray<_::ExceptionOr<T>>(promises.size())));
|
cannam@49
|
944 }
|
cannam@49
|
945
|
cannam@49
|
946 // =======================================================================================
|
cannam@49
|
947
|
cannam@49
|
948 namespace _ { // private
|
cannam@49
|
949
|
cannam@49
|
950 template <typename T>
|
cannam@49
|
951 class WeakFulfiller final: public PromiseFulfiller<T>, private kj::Disposer {
|
cannam@49
|
952 // A wrapper around PromiseFulfiller which can be detached.
|
cannam@49
|
953 //
|
cannam@49
|
954 // There are a couple non-trivialities here:
|
cannam@49
|
955 // - If the WeakFulfiller is discarded, we want the promise it fulfills to be implicitly
|
cannam@49
|
956 // rejected.
|
cannam@49
|
957 // - We cannot destroy the WeakFulfiller until the application has discarded it *and* it has been
|
cannam@49
|
958 // detached from the underlying fulfiller, because otherwise the later detach() call will go
|
cannam@49
|
959 // to a dangling pointer. Essentially, WeakFulfiller is reference counted, although the
|
cannam@49
|
960 // refcount never goes over 2 and we manually implement the refcounting because we need to do
|
cannam@49
|
961 // other special things when each side detaches anyway. To this end, WeakFulfiller is its own
|
cannam@49
|
962 // Disposer -- dispose() is called when the application discards its owned pointer to the
|
cannam@49
|
963 // fulfiller and detach() is called when the promise is destroyed.
|
cannam@49
|
964
|
cannam@49
|
965 public:
|
cannam@49
|
966 KJ_DISALLOW_COPY(WeakFulfiller);
|
cannam@49
|
967
|
cannam@49
|
968 static kj::Own<WeakFulfiller> make() {
|
cannam@49
|
969 WeakFulfiller* ptr = new WeakFulfiller;
|
cannam@49
|
970 return Own<WeakFulfiller>(ptr, *ptr);
|
cannam@49
|
971 }
|
cannam@49
|
972
|
cannam@49
|
973 void fulfill(FixVoid<T>&& value) override {
|
cannam@49
|
974 if (inner != nullptr) {
|
cannam@49
|
975 inner->fulfill(kj::mv(value));
|
cannam@49
|
976 }
|
cannam@49
|
977 }
|
cannam@49
|
978
|
cannam@49
|
979 void reject(Exception&& exception) override {
|
cannam@49
|
980 if (inner != nullptr) {
|
cannam@49
|
981 inner->reject(kj::mv(exception));
|
cannam@49
|
982 }
|
cannam@49
|
983 }
|
cannam@49
|
984
|
cannam@49
|
985 bool isWaiting() override {
|
cannam@49
|
986 return inner != nullptr && inner->isWaiting();
|
cannam@49
|
987 }
|
cannam@49
|
988
|
cannam@49
|
989 void attach(PromiseFulfiller<T>& newInner) {
|
cannam@49
|
990 inner = &newInner;
|
cannam@49
|
991 }
|
cannam@49
|
992
|
cannam@49
|
993 void detach(PromiseFulfiller<T>& from) {
|
cannam@49
|
994 if (inner == nullptr) {
|
cannam@49
|
995 // Already disposed.
|
cannam@49
|
996 delete this;
|
cannam@49
|
997 } else {
|
cannam@49
|
998 KJ_IREQUIRE(inner == &from);
|
cannam@49
|
999 inner = nullptr;
|
cannam@49
|
1000 }
|
cannam@49
|
1001 }
|
cannam@49
|
1002
|
cannam@49
|
1003 private:
|
cannam@49
|
1004 mutable PromiseFulfiller<T>* inner;
|
cannam@49
|
1005
|
cannam@49
|
1006 WeakFulfiller(): inner(nullptr) {}
|
cannam@49
|
1007
|
cannam@49
|
1008 void disposeImpl(void* pointer) const override {
|
cannam@49
|
1009 // TODO(perf): Factor some of this out so it isn't regenerated for every fulfiller type?
|
cannam@49
|
1010
|
cannam@49
|
1011 if (inner == nullptr) {
|
cannam@49
|
1012 // Already detached.
|
cannam@49
|
1013 delete this;
|
cannam@49
|
1014 } else {
|
cannam@49
|
1015 if (inner->isWaiting()) {
|
cannam@49
|
1016 inner->reject(kj::Exception(kj::Exception::Type::FAILED, __FILE__, __LINE__,
|
cannam@49
|
1017 kj::heapString("PromiseFulfiller was destroyed without fulfilling the promise.")));
|
cannam@49
|
1018 }
|
cannam@49
|
1019 inner = nullptr;
|
cannam@49
|
1020 }
|
cannam@49
|
1021 }
|
cannam@49
|
1022 };
|
cannam@49
|
1023
|
cannam@49
|
1024 template <typename T>
|
cannam@49
|
1025 class PromiseAndFulfillerAdapter {
|
cannam@49
|
1026 public:
|
cannam@49
|
1027 PromiseAndFulfillerAdapter(PromiseFulfiller<T>& fulfiller,
|
cannam@49
|
1028 WeakFulfiller<T>& wrapper)
|
cannam@49
|
1029 : fulfiller(fulfiller), wrapper(wrapper) {
|
cannam@49
|
1030 wrapper.attach(fulfiller);
|
cannam@49
|
1031 }
|
cannam@49
|
1032
|
cannam@49
|
1033 ~PromiseAndFulfillerAdapter() noexcept(false) {
|
cannam@49
|
1034 wrapper.detach(fulfiller);
|
cannam@49
|
1035 }
|
cannam@49
|
1036
|
cannam@49
|
1037 private:
|
cannam@49
|
1038 PromiseFulfiller<T>& fulfiller;
|
cannam@49
|
1039 WeakFulfiller<T>& wrapper;
|
cannam@49
|
1040 };
|
cannam@49
|
1041
|
cannam@49
|
1042 } // namespace _ (private)
|
cannam@49
|
1043
|
cannam@49
|
1044 template <typename T>
|
cannam@49
|
1045 template <typename Func>
|
cannam@49
|
1046 bool PromiseFulfiller<T>::rejectIfThrows(Func&& func) {
|
cannam@49
|
1047 KJ_IF_MAYBE(exception, kj::runCatchingExceptions(kj::mv(func))) {
|
cannam@49
|
1048 reject(kj::mv(*exception));
|
cannam@49
|
1049 return false;
|
cannam@49
|
1050 } else {
|
cannam@49
|
1051 return true;
|
cannam@49
|
1052 }
|
cannam@49
|
1053 }
|
cannam@49
|
1054
|
cannam@49
|
1055 template <typename Func>
|
cannam@49
|
1056 bool PromiseFulfiller<void>::rejectIfThrows(Func&& func) {
|
cannam@49
|
1057 KJ_IF_MAYBE(exception, kj::runCatchingExceptions(kj::mv(func))) {
|
cannam@49
|
1058 reject(kj::mv(*exception));
|
cannam@49
|
1059 return false;
|
cannam@49
|
1060 } else {
|
cannam@49
|
1061 return true;
|
cannam@49
|
1062 }
|
cannam@49
|
1063 }
|
cannam@49
|
1064
|
cannam@49
|
1065 template <typename T, typename Adapter, typename... Params>
|
cannam@49
|
1066 Promise<T> newAdaptedPromise(Params&&... adapterConstructorParams) {
|
cannam@49
|
1067 return Promise<T>(false, heap<_::AdapterPromiseNode<_::FixVoid<T>, Adapter>>(
|
cannam@49
|
1068 kj::fwd<Params>(adapterConstructorParams)...));
|
cannam@49
|
1069 }
|
cannam@49
|
1070
|
cannam@49
|
1071 template <typename T>
|
cannam@49
|
1072 PromiseFulfillerPair<T> newPromiseAndFulfiller() {
|
cannam@49
|
1073 auto wrapper = _::WeakFulfiller<T>::make();
|
cannam@49
|
1074
|
cannam@49
|
1075 Own<_::PromiseNode> intermediate(
|
cannam@49
|
1076 heap<_::AdapterPromiseNode<_::FixVoid<T>, _::PromiseAndFulfillerAdapter<T>>>(*wrapper));
|
cannam@49
|
1077 Promise<_::JoinPromises<T>> promise(false,
|
cannam@49
|
1078 _::maybeChain(kj::mv(intermediate), implicitCast<T*>(nullptr)));
|
cannam@49
|
1079
|
cannam@49
|
1080 return PromiseFulfillerPair<T> { kj::mv(promise), kj::mv(wrapper) };
|
cannam@49
|
1081 }
|
cannam@49
|
1082
|
cannam@49
|
1083 } // namespace kj
|
cannam@49
|
1084
|
cannam@49
|
1085 #endif // KJ_ASYNC_INL_H_
|