annotate win32-mingw/include/kj/async-inl.h @ 142:75bf92aa2d1f

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