lib: add abortSignal.throwIfAborted()

Refs: whatwg/dom#1034 Signed-off-by: James M Snell <jasnell@gmail.com>
nodejs · Nov 24, 2021 · 02cc0ee · 02cc0ee
1 parent 7633c86
commit 02cc0ee
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 2 deletions.
diff --git a/doc/api/globals.md b/doc/api/globals.md
@@ -180,6 +180,14 @@ ac.abort(new Error('boom!'));
 console.log(ac.signal.reason);  // Error('boom!');
 ```
 
+#### `abortSignal.throwIfAborted()`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+If `abortSignal.aborted` is `true`, throws `abortSignal.reason`.
+
 ## Class: `Buffer`
 
 <!-- YAML

diff --git a/lib/internal/abort_controller.js b/lib/internal/abort_controller.js
@@ -29,6 +29,10 @@ const {
   }
 } = require('internal/errors');
 
+const {
+  DOMException,
+} = internalBinding('messaging');
+
 const kAborted = Symbol('kAborted');
 const kReason = Symbol('kReason');
 
@@ -69,6 +73,12 @@ class AbortSignal extends EventTarget {
     return this[kReason];
   }
 
+  throwIfAborted() {
+    if (this.aborted) {
+      throw this.reason;
+    }
+  }
+
   [customInspectSymbol](depth, options) {
     return customInspect(this, {
       aborted: this.aborted
@@ -79,7 +89,8 @@ class AbortSignal extends EventTarget {
    * @param {any} reason
    * @returns {AbortSignal}
    */
-  static abort(reason) {
+  static abort(
+      reason = new DOMException('This operation was aborted', 'AbortError')) {
     return createAbortSignal(true, reason);
   }
 }
@@ -141,7 +152,7 @@ class AbortController {
   /**
    * @param {any} reason
    */
-  abort(reason) {
+  abort(reason = new DOMException('This operation was aborted', 'AbortError')) {
     validateAbortController(this);
     abortSignal(this[kSignal], reason);
   }

diff --git a/test/parallel/test-abortcontroller.js b/test/parallel/test-abortcontroller.js
@@ -153,3 +153,20 @@ const { ok, strictEqual, throws } = require('assert');
   const signal = AbortSignal.abort('reason');
   strictEqual(signal.reason, 'reason');
 }
+
+{
+  // Test AbortSignal.reason default
+  const signal = AbortSignal.abort();
+  ok(signal.reason instanceof DOMException);
+  strictEqual(signal.reason.code, 20);
+
+  const ac = new AbortController();
+  ac.abort();
+  ok(ac.signal.reason instanceof DOMException);
+  strictEqual(ac.signal.reason.code, 20);
+}
+
+{
+  // Test abortSignal.throwIfAborted()
+  throws(() => AbortSignal.abort().throwIfAborted(), { code: 20 });
+}