feat: add all function to the async module (#305)

Author: sodiray

cdn/radash.esm.js

@@ -408,6 +408,28 @@ const parallel = async (limit, array, func) => {
   }
   return results.map((r) => r.result);
 };
+async function all(promises) {
+  const entries = isArray(promises) ? promises.map((p) => [null, p]) : Object.entries(promises);
+  const results = await Promise.all(
+    entries.map(
+      ([key, value]) => value.then((result) => ({ result, exc: null, key })).catch((exc) => ({ result: null, exc, key }))
+    )
+  );
+  const exceptions = results.filter((r) => r.exc);
+  if (exceptions.length > 0) {
+    throw new AggregateError(exceptions.map((e) => e.exc));
+  }
+  if (isArray(promises)) {
+    return results.map((r) => r.result);
+  }
+  return results.reduce(
+    (acc, item) => ({
+      ...acc,
+      [item.key]: item.result
+    }),
+    {}
+  );
+}
 const retry = async (options, func) => {
   const times = options?.times ?? 3;
   const delay = options?.delay;
@@ -888,4 +910,4 @@ const trim = (str, charsToTrim = " ") => {
   return str.replace(regex, "");
 };
 
-export { alphabetical, assign, boil, callable, camel, capitalize, chain, clone, cluster, compose, construct, counting, crush, dash, debounce, defer, diff, draw, first, flat, fork, get, group, guard, intersects, invert, isArray, isDate, isEmpty, isEqual, isFloat, isFunction, isInt, isNumber, isObject, isPrimitive, isString, isSymbol, iterate, keys, last, list, listify, lowerize, map, mapEntries, mapKeys, mapValues, max, memo, merge, min, objectify, omit, parallel, partial, partob, pascal, pick, proxied, random, range, reduce, replace, replaceOrAppend, retry, select, series, set, shake, shift, shuffle, sift, sleep, snake, sort, sum, template, throttle, title, toFloat, toInt, toggle, trim, tryit as try, tryit, uid, unique, upperize, zip, zipToObject };
+export { all, alphabetical, assign, boil, callable, camel, capitalize, chain, clone, cluster, compose, construct, counting, crush, dash, debounce, defer, diff, draw, first, flat, fork, get, group, guard, intersects, invert, isArray, isDate, isEmpty, isEqual, isFloat, isFunction, isInt, isNumber, isObject, isPrimitive, isString, isSymbol, iterate, keys, last, list, listify, lowerize, map, mapEntries, mapKeys, mapValues, max, memo, merge, min, objectify, omit, parallel, partial, partob, pascal, pick, proxied, random, range, reduce, replace, replaceOrAppend, retry, select, series, set, shake, shift, shuffle, sift, sleep, snake, sort, sum, template, throttle, title, toFloat, toInt, toggle, trim, tryit as try, tryit, uid, unique, upperize, zip, zipToObject };

cdn/radash.js

@@ -411,6 +411,28 @@ var radash = (function (exports) {
     }
     return results.map((r) => r.result);
   };
+  async function all(promises) {
+    const entries = isArray(promises) ? promises.map((p) => [null, p]) : Object.entries(promises);
+    const results = await Promise.all(
+      entries.map(
+        ([key, value]) => value.then((result) => ({ result, exc: null, key })).catch((exc) => ({ result: null, exc, key }))
+      )
+    );
+    const exceptions = results.filter((r) => r.exc);
+    if (exceptions.length > 0) {
+      throw new AggregateError(exceptions.map((e) => e.exc));
+    }
+    if (isArray(promises)) {
+      return results.map((r) => r.result);
+    }
+    return results.reduce(
+      (acc, item) => ({
+        ...acc,
+        [item.key]: item.result
+      }),
+      {}
+    );
+  }
   const retry = async (options, func) => {
     const times = options?.times ?? 3;
     const delay = options?.delay;
@@ -891,6 +913,7 @@ var radash = (function (exports) {
     return str.replace(regex, "");
   };
 
+  exports.all = all;
   exports.alphabetical = alphabetical;
   exports.assign = assign;
   exports.boil = boil;

cdn/radash.min.js

@@ -1,6 +1,6 @@
 {
   "name": "radash",
-  "version": "10.8.1",
+  "version": "10.9.0",
   "description": "Functional utility library - modern, simple, typed, powerful",
   "main": "dist/cjs/index.cjs",
   "module": "dist/esm/index.mjs",

package.json

@@ -1,4 +1,5 @@
 import { fork, list, range, sort } from './array'
+import { isArray } from './typed'
 
 /**
  * An async reduce function. Works like the
@@ -145,6 +146,75 @@ export const parallel = async <T, K>(
   return results.map(r => r.result)
 }
 
+type PromiseValues<T extends Promise<any>[]> = {
+  [K in keyof T]: T[K] extends Promise<infer U> ? U : never
+}
+
+/**
+ * Functionally similar to Promise.all or Promise.allSettled. If any
+ * errors are thrown, all errors are gathered and thrown in an
+ * AggregateError.
+ *
+ * @example
+ * const [user] = await all({
+ *   api.users.create(...),
+ *   s3.buckets.create(...),
+ *   slack.customerSuccessChannel.sendMessage(...)
+ * })
+ */
+export async function all<T extends Promise<any>[]>(
+  promises: T
+): Promise<PromiseValues<T>>
+/**
+ * Functionally similar to Promise.all or Promise.allSettled. If any
+ * errors are thrown, all errors are gathered and thrown in an
+ * AggregateError.
+ *
+ * @example
+ * const { user } = await all({
+ *   user: api.users.create(...),
+ *   bucket: s3.buckets.create(...),
+ *   message: slack.customerSuccessChannel.sendMessage(...)
+ * })
+ */
+export async function all<T extends Record<string, Promise<any>>>(
+  promises: T
+): Promise<{ [K in keyof T]: Awaited<T[K]> }>
+export async function all<
+  T extends Record<string, Promise<any>> | Promise<any>[]
+>(promises: T) {
+  const entries = isArray(promises)
+    ? promises.map(p => [null, p] as [null, Promise<any>])
+    : Object.entries(promises)
+
+  const results = await Promise.all(
+    entries.map(([key, value]) =>
+      value
+        .then(result => ({ result, exc: null, key }))
+        .catch(exc => ({ result: null, exc, key }))
+    )
+  )
+
+  const exceptions = results.filter(r => r.exc)
+  if (exceptions.length > 0) {
+    throw new AggregateError(exceptions.map(e => e.exc))
+  }
+
+  if (isArray(promises)) {
+    return results.map(r => r.result) as T extends Promise<any>[]
+      ? PromiseValues<T>
+      : unknown
+  }
+
+  return results.reduce(
+    (acc, item) => ({
+      ...acc,
+      [item.key!]: item.result
+    }),
+    {} as { [K in keyof T]: Awaited<T[K]> }
+  )
+}
+
 /**
  * Retries the given function the specified number
  * of times.

src/async.ts

@@ -30,6 +30,7 @@ export {
   zipToObject
 } from './array'
 export {
+  all,
   defer,
   guard,
   map,

src/index.ts

@@ -485,4 +485,61 @@ describe('async module', () => {
       }
     })
   })
+
+  describe('_.all', () => {
+    const promise = {
+      pass: <T>(value: T) => new Promise<T>(res => res(value)),
+      fail: (err: any) => new Promise((res, rej) => rej(err))
+    }
+    it('returns array with values in correct order when given array', async () => {
+      const result = await _.all([
+        promise.pass(22),
+        promise.pass('hello'),
+        promise.pass({ name: 'ray' })
+      ])
+      assert.deepEqual(result, [22, 'hello', { name: 'ray' }])
+    })
+    it('returns object with values in correct keys when given object', async () => {
+      const result = await _.all({
+        num: promise.pass(22),
+        str: promise.pass('hello'),
+        obj: promise.pass({ name: 'ray' })
+      })
+      assert.deepEqual(result, {
+        num: 22,
+        str: 'hello',
+        obj: { name: 'ray' }
+      })
+    })
+    it('throws aggregate error when a single promise fails (in object mode)', async () => {
+      try {
+        await _.all({
+          num: promise.pass(22),
+          str: promise.pass('hello'),
+          err: promise.fail(new Error('broken'))
+        })
+      } catch (e: any) {
+        const err = e as AggregateError
+        assert.equal(err.errors.length, 1)
+        assert.equal(err.errors[0].message, 'broken')
+        return
+      }
+      assert.fail('Expected error to be thrown but it was not')
+    })
+    it('throws aggregate error when a single promise fails (in array mode)', async () => {
+      try {
+        await _.all([
+          promise.pass(22),
+          promise.pass('hello'),
+          promise.fail(new Error('broken'))
+        ])
+      } catch (e: any) {
+        const err = e as AggregateError
+        assert.equal(err.errors.length, 1)
+        assert.equal(err.errors[0].message, 'broken')
+        return
+      }
+      assert.fail('Expected error to be thrown but it was not')
+    })
+  })
 })