Fix AsyncResource propagation issue

Fixes: #200
sindresorhus · Oct 30, 2023 · 4e0de89 · 4e0de89
1 parent 38a6773
commit 4e0de89
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -1,4 +1,5 @@
 import Queue from 'yocto-queue';
+import { AsyncResource } from 'async_hooks';
 
 export default function pLimit(concurrency) {
 	if (!((Number.isInteger(concurrency) || concurrency === Number.POSITIVE_INFINITY) && concurrency > 0)) {
@@ -31,7 +32,9 @@ export default function pLimit(concurrency) {
 	};
 
 	const enqueue = (fn, resolve, args) => {
-		queue.enqueue(run.bind(undefined, fn, resolve, args));
+		queue.enqueue(
+			AsyncResource.bind(run.bind(undefined, fn, resolve, args))
+		);
 
 		(async () => {
 			// This function needs to wait until the next microtask before comparing