Software /
code /
prosody
Changeset
11484:a0120e935442
util.promise: Add join() convenience method
Usage: promise.join(p1, p2, function (result1, result2)
[...]
end)
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Thu, 25 Mar 2021 14:28:38 +0000 |
parents | 11483:24ce9d380475 |
children | 11485:7d42ed3a8a40 |
files | spec/util_promise_spec.lua util/promise.lua |
diffstat | 2 files changed, 37 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- a/spec/util_promise_spec.lua Thu Mar 25 13:51:16 2021 +0000 +++ b/spec/util_promise_spec.lua Thu Mar 25 14:28:38 2021 +0000 @@ -459,6 +459,32 @@ assert.same({ foo = true }, result); end); end); + describe("join()", function () + it("works", function () + local r1, r2; + local res1, res2; + local p1, p2 = promise.new(function (resolve) r1 = resolve end), promise.new(function (resolve) r2 = resolve end); + + local p = promise.join(p1, p2, function (_res1, _res2) + res1, res2 = _res1, _res2; + return promise.resolve("works"); + end); + + local result; + local cb = spy.new(function (v) + result = v; + end); + p:next(cb); + assert.spy(cb).was_called(0); + r2("yep"); + assert.spy(cb).was_called(0); + r1("nope"); + assert.spy(cb).was_called(1); + assert.same("works", result); + assert.equals("nope", res1); + assert.equals("yep", res2); + end); + end); it("promises may be resolved by other promises", function () local r1, r2; local p1, p2 = promise.new(function (resolve) r1 = resolve end), promise.new(function (resolve) r2 = resolve end);
--- a/util/promise.lua Thu Mar 25 13:51:16 2021 +0000 +++ b/util/promise.lua Thu Mar 25 14:28:38 2021 +0000 @@ -2,6 +2,7 @@ local promise_mt = { __name = "promise", __index = promise_methods }; local xpcall = require "util.xpcall".xpcall; +local unpack = table.unpack or unpack; --luacheck: ignore 113 function promise_mt:__tostring() return "promise (" .. (self._state or "invalid") .. ")"; @@ -137,6 +138,15 @@ end); end +local function join(...) + local promises, n = { ... }, select("#", ...); + local handler = promises[n]; + promises[n] = nil; + return all(promises):next(function (results) + return handler(unpack(results, 1, n - 1)); + end); +end + local function race(promises) return new(function (resolve, reject) for i = 1, #promises do @@ -180,6 +190,7 @@ return { new = new; resolve = resolve; + join = join; reject = reject; all = all; all_settled = all_settled;