add queue.all (#2)

Author: henrygd

bun.lock

@@ -35,6 +35,7 @@ let newQueue = (concurrency) => {
   let tail;
   let resolveDonePromise;
   let donePromise;
+  let queue;
   let afterRun = () => {
     active--;
     if (--size) {
@@ -54,7 +55,7 @@ let newQueue = (concurrency) => {
       );
     }
   };
-  return {
+  return queue = {
     add(p) {
       let node = { p: import_async_hooks.AsyncResource.bind(p) };
       let promise = new Promize((res, rej) => {
@@ -84,6 +85,7 @@ let newQueue = (concurrency) => {
       size = active;
     },
     active: () => active,
-    size: () => size
+    size: () => size,
+    all: (fns) => Promize.all(fns.map((fn) => queue.add(typeof fn === "function" ? fn : () => fn)))
   };
 };

dist/index.async-storage.cjs

@@ -13,6 +13,7 @@ let newQueue = (concurrency) => {
   let tail;
   let resolveDonePromise;
   let donePromise;
+  let queue;
   let afterRun = () => {
     active--;
     if (--size) {
@@ -32,7 +33,7 @@ let newQueue = (concurrency) => {
       );
     }
   };
-  return {
+  return queue = {
     add(p) {
       let node = { p: AsyncResource.bind(p) };
       let promise = new Promize((res, rej) => {
@@ -62,7 +63,8 @@ let newQueue = (concurrency) => {
       size = active;
     },
     active: () => active,
-    size: () => size
+    size: () => size,
+    all: (fns) => Promize.all(fns.map((fn) => queue.add(typeof fn === "function" ? fn : () => fn)))
   };
 };
 export {

dist/index.async-storage.js

@@ -15,11 +15,11 @@ var __copyProps = (to, from, except, desc) => {
   return to;
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var async_queue_exports = {};
-__export(async_queue_exports, {
+var queue_exports = {};
+__export(queue_exports, {
   newQueue: () => newQueue
 });
-module.exports = __toCommonJS(async_queue_exports);
+module.exports = __toCommonJS(queue_exports);
 let Promize = Promise;
 let newQueue = (concurrency) => {
   let active = 0;
@@ -28,6 +28,7 @@ let newQueue = (concurrency) => {
   let tail;
   let resolveDonePromise;
   let donePromise;
+  let queue;
   let afterRun = () => {
     active--;
     if (--size) {
@@ -47,7 +48,7 @@ let newQueue = (concurrency) => {
       );
     }
   };
-  return {
+  return queue = {
     add(p) {
       let node = { p };
       let promise = new Promize((res, rej) => {
@@ -77,6 +78,7 @@ let newQueue = (concurrency) => {
       size = active;
     },
     active: () => active,
-    size: () => size
+    size: () => size,
+    all: (fns) => Promize.all(fns.map((fn) => queue.add(typeof fn === "function" ? fn : () => fn)))
   };
 };

dist/index.cjs

@@ -10,6 +10,8 @@ export interface Queue {
     active(): number;
     /** Returns the total number of promises in the queue */
     size(): number;
+    /** Adds promises (or wrappers) to the queue and resolves like Promise.all */
+    all<T>(promiseFunctions: Array<PromiseLike<T> | (() => PromiseLike<T>)>): Promise<T[]>;
 }
 /**
  * Creates a new queue with the specified concurrency level.

dist/index.d.ts

@@ -6,6 +6,7 @@ let newQueue = (concurrency) => {
   let tail;
   let resolveDonePromise;
   let donePromise;
+  let queue;
   let afterRun = () => {
     active--;
     if (--size) {
@@ -25,7 +26,7 @@ let newQueue = (concurrency) => {
       );
     }
   };
-  return {
+  return queue = {
     add(p) {
       let node = { p };
       let promise = new Promize((res, rej) => {
@@ -55,7 +56,8 @@ let newQueue = (concurrency) => {
       size = active;
     },
     active: () => active,
-    size: () => size
+    size: () => size,
+    all: (fns) => Promize.all(fns.map((fn) => queue.add(typeof fn === "function" ? fn : () => fn)))
   };
 };
 export {

dist/index.js

@@ -31,6 +31,8 @@ interface Queue {
 	active(): number
 	/** Returns the total number of promises in the queue */
 	size(): number
+	/** Adds promises (or wrappers) to the queue and resolves like Promise.all */
+	all<T>(promiseFunctions: Array<PromiseLike<T> | (() => PromiseLike<T>)>): Promise<T[]>
 }
 
 // this just saves a few bytes
@@ -49,6 +51,7 @@ export let newQueue = (concurrency: number): Queue => {
 	let tail: Node<PromiseLike<any>> | undefined | null
 	let resolveDonePromise: (value: void | PromiseLike<void>) => void
 	let donePromise: Promise<void> | void
+	let queue: Queue
 
 	let afterRun = () => {
 		active--
@@ -66,12 +69,12 @@ export let newQueue = (concurrency: number): Queue => {
 			head = head.next
 			curHead.p().then(
 				(v) => (curHead.res(v), afterRun()),
-				(e) => (curHead.rej(e), afterRun())
+				(e) => (curHead.rej(e), afterRun()),
 			)
 		}
 	}
 
-	return {
+	return (queue = {
 		add<T>(p: () => PromiseLike<T>) {
 			let node = { p: AsyncResource.bind(p) } as unknown as Node<PromiseLike<T>>
 			let promise = new Promize((res, rej) => {
@@ -102,5 +105,7 @@ export let newQueue = (concurrency: number): Queue => {
 		},
 		active: () => active,
 		size: () => size,
-	}
+		all: <T>(fns: Array<PromiseLike<T> | (() => PromiseLike<T>)>): Promise<T[]> =>
+			Promize.all(fns.map((fn) => queue.add(typeof fn === 'function' ? fn : () => fn))),
+	})
 }

dist/index.min.js

@@ -22,6 +22,8 @@ export interface Queue {
 	active(): number
 	/** Returns the total number of promises in the queue */
 	size(): number
+	/** Adds promises (or wrappers) to the queue and resolves like Promise.all */
+	all<T>(promiseFunctions: Array<PromiseLike<T> | (() => PromiseLike<T>)>): Promise<T[]>
 }
 
 // this just saves a few bytes
@@ -40,6 +42,7 @@ export let newQueue = (concurrency: number): Queue => {
 	let tail: Node<PromiseLike<any>> | undefined | null
 	let resolveDonePromise: (value: void | PromiseLike<void>) => void
 	let donePromise: Promise<void> | void
+	let queue: Queue
 
 	let afterRun = () => {
 		active--
@@ -62,7 +65,7 @@ export let newQueue = (concurrency: number): Queue => {
 		}
 	}
 
-	return {
+	return (queue = {
 		add<T>(p: () => PromiseLike<T>) {
 			let node = { p } as Node<PromiseLike<T>>
 			let promise = new Promize((res, rej) => {
@@ -93,5 +96,7 @@ export let newQueue = (concurrency: number): Queue => {
 		},
 		active: () => active,
 		size: () => size,
-	}
+		all: <T>(fns: Array<PromiseLike<T> | (() => PromiseLike<T>)>): Promise<T[]> => 
+			Promize.all(fns.map(fn => queue.add(typeof fn === 'function' ? fn : () => fn))),
+	})
 }

index.async-storage.ts

@@ -1,6 +1,6 @@
 {
 	"name": "@henrygd/queue",
-	"version": "1.0.7",
+	"version": "1.1.0",
 	"license": "MIT",
 	"type": "module",
 	"description": "Tiny async queue with concurrency control. Like p-limit or fastq, but smaller and faster.",
@@ -61,7 +61,7 @@
 		"worker"
 	],
 	"devDependencies": {
-		"@types/bun": "^1.1.3",
+		"@types/bun": "latest",
 		"async": "^3.2.5",
 		"esbuild": "^0.21.4",
 		"fastq": "^1.17.1",