improve ThroatQueueFunction api

HalleyAssist · Nov 29, 2022 · 2bd71b3 · 2bd71b3
1 parent 2bbf5ad
commit 2bd71b3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/ThroatQueueFunction.js b/lib/ThroatQueueFunction.js
@@ -6,9 +6,9 @@ function ThroatQueueFunction(n = 5){
     const running = []
     const ret = async function(what){
         if(what === null){
-            if(running.length === 0) return
+            if(running.length === 0) return running
             await Promise.all(running)
-            return
+            return running
         }
 
         // This shouldn't happen if we correctly await on the throat
@@ -37,6 +37,7 @@ function ThroatQueueFunction(n = 5){
         const idObj = {}
         const r = rFn()
         r.id = idObj
+        r.fn = what
         running.push(r)
         await r