updated whenDone, onceDone to use wrapped callback

kksharma1618 · Feb 1, 2015 · dd079a7 · dd079a7
1 parent 6a19a39
commit dd079a7
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/lib/taskgroup.coffee b/src/lib/taskgroup.coffee
@@ -61,7 +61,11 @@ class Interface extends EventEmitter
 	whenDone: (listener) ->
 		# check if we have a listener
 		if typeof listener is 'function'
-			@on('done', listener.bind(@))
+			queuedListener = (args...) ->
+				process.nextTick () ->
+					listener(args...)
+
+			@on('done', queuedListener.bind(@))
 
 		# Chain
 		@
@@ -72,7 +76,11 @@ class Interface extends EventEmitter
 	onceDone: (listener) ->
 		# check if we have a listener
 		if typeof listener is 'function'
-			@once('done', listener)
+			queuedListener = (args...) ->
+				process.nextTick () ->
+					listener(args...)
+
+			@once('done', queuedListener)
 
 		# Chain
 		@
@@ -1280,4 +1288,4 @@ class TaskGroup extends Interface
 		@
 
 # Export
-module.exports = {Task,TaskGroup}
+module.exports = {Task,TaskGroup}