aboutsummaryrefslogtreecommitdiffstats
path: root/util
diff options
context:
space:
mode:
authorMatthew Wild <mwild1@gmail.com>2021-03-25 13:51:16 +0000
committerMatthew Wild <mwild1@gmail.com>2021-03-25 13:51:16 +0000
commit3952fae8ce728d56b9ef2044e490d99a6c12f444 (patch)
treefc6be322593893ecf6b362a5836002395414fad1 /util
parent85d3eb829bb48512ab27a4c4971403e1f2284fb0 (diff)
downloadprosody-3952fae8ce728d56b9ef2044e490d99a6c12f444.tar.gz
prosody-3952fae8ce728d56b9ef2044e490d99a6c12f444.zip
util.promise: Add support for arbitrary keys in all()/all_settled()
Diffstat (limited to 'util')
-rw-r--r--util/promise.lua42
1 files changed, 27 insertions, 15 deletions
diff --git a/util/promise.lua b/util/promise.lua
index bcb7beda..98d313ba 100644
--- a/util/promise.lua
+++ b/util/promise.lua
@@ -91,37 +91,49 @@ end
local function all(promises)
return new(function (resolve, reject)
- local count, total, results = 0, #promises, {};
- for i = 1, total do
- promises[i]:next(function (v)
- results[i] = v;
- count = count + 1;
- if count == total then
+ local settled, results, loop_finished = 0, {}, false;
+ local total = 0;
+ for k, v in pairs(promises) do
+ total = total + 1;
+ v:next(function (value)
+ results[k] = value;
+ settled = settled + 1;
+ if settled == total and loop_finished then
resolve(results);
end
end, reject);
end
+ loop_finished = true;
+ if settled == total then
+ resolve(results);
+ end
end);
end
local function all_settled(promises)
return new(function (resolve)
- local count, total, results = 0, #promises, {};
- for i = 1, total do
- promises[i]:next(function (v)
- results[i] = { status = "fulfilled", value = v };
- count = count + 1;
- if count == total then
+ local settled, results, loop_finished = 0, {}, false;
+ local total = 0;
+ for k, v in pairs(promises) do
+ total = total + 1;
+ v:next(function (value)
+ results[k] = { status = "fulfilled", value = value };
+ settled = settled + 1;
+ if settled == total and loop_finished then
resolve(results);
end
end, function (e)
- results[i] = { status = "rejected", reason = e };
- count = count + 1;
- if count == total then
+ results[k] = { status = "rejected", reason = e };
+ settled = settled + 1;
+ if settled == total and loop_finished then
resolve(results);
end
end);
end
+ loop_finished = true;
+ if settled == total then
+ resolve(results);
+ end
end);
end