feat(hw): raceWithSignal

reduxjs/redux-toolkit#3021
changshou83 · Feb 1, 2023 · a0abd37 · a0abd37
1 parent 6cbdf3d
commit a0abd37
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/HandWriting/tools/raceWithSignal.js b/HandWriting/tools/raceWithSignal.js
@@ -0,0 +1,23 @@
+const noop = () => {};
+
+export const raceWithSignal = (signal, promise) => {
+  let cleanup = noop;
+  return new Promise((resolve, reject) => {
+    const notifyRejection = () => reject(new TaskAbortError(signal.reason));
+
+    if (signal.aborted) {
+      notifyRejection();
+      return;
+    }
+
+    cleanup = addAbortSignalListener(signal, notifyRejection);
+    promise.finally(() => cleanup()).then(resolve, reject);
+  }).finally(() => {
+    cleanup = noop;
+  });
+};
+
+export const addAbortSignalListener = (signal, callback) => {
+  signal.addEventListener("abort", callback, { once: true });
+  return () => signal.removeEventListener("abort", callback);
+};