util.async: Optionally allow too many 'done' callbacks

Sometimes, like in mod_c2s and mod_s2s during shutdown, all you want is
to wait for the first done() and not complicate things.

1 files changed, 2 insertions, 2 deletions

diff --git a/util/async.lua b/util/async.lua
index ece589cb..2830238f 100644
--- a/util/async.lua
+++ b/util/async.lua
@@ -73,7 +73,7 @@ local function runner_continue(thread)
 	return true;
 end
 
-local function waiter(num)
+local function waiter(num, allow_many)
 	local thread = checkthread();
 	num = num or 1;
 	local waiting;
@@ -85,7 +85,7 @@ local function waiter(num)
 		num = num - 1;
 		if num == 0 and waiting then
 			runner_continue(thread);
-		elseif num < 0 then
+		elseif not allow_many and num < 0 then
 			error("done() called too many times");
 		end
 	end;