annotate win32-mingw/include/kj/async-inl.h @ 149:279b18cc7785

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