annotate osx/include/kj/async-inl.h @ 49:3ab5a40c4e3b

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