nvim-java/lua/async/waits/wait_all.lua at v4.1.1 · nvim-java/nvim-java

Repository navigation

local co = coroutine

local function __join(thunks)

local len = #thunks

local done = 0

local acc = {}

local thunk = function(step)

if len == 0 then

return step()

end

for i, tk in ipairs(thunks) do

assert(type(tk) == 'function', 'thunk must be function')

local callback = function(...)

acc[i] = { ... }

done = done + 1

if done == len then

step(unpack(acc))

end

tk(callback)

end

return thunk

end

---Waits for list of async calls to be completed

---@param defer fun(callback: fun(result: any))

---@return any[]

local function wait_all(defer)

assert(type(defer) == 'table', 'type error :: expected table')

return co.yield(__join(defer))

end

return wait_all