142 lines
2.8 KiB
Lua
142 lines
2.8 KiB
Lua
local async = {}
|
|
|
|
---@class cmp.AsyncThrottle
|
|
---@field public running boolean
|
|
---@field public timeout integer
|
|
---@field public sync function(self: cmp.AsyncThrottle, timeout: integer|nil)
|
|
---@field public stop function
|
|
---@field public __call function
|
|
|
|
local timers = {}
|
|
|
|
vim.api.nvim_create_autocmd('VimLeavePre', {
|
|
callback = function()
|
|
for _, timer in pairs(timers) do
|
|
if timer and not timer:is_closing() then
|
|
timer:stop()
|
|
timer:close()
|
|
end
|
|
end
|
|
end
|
|
})
|
|
|
|
---@param fn function
|
|
---@param timeout integer
|
|
---@return cmp.AsyncThrottle
|
|
async.throttle = function(fn, timeout)
|
|
local time = nil
|
|
local timer = vim.loop.new_timer()
|
|
timers[#timers+1] = timer
|
|
return setmetatable({
|
|
running = false,
|
|
timeout = timeout,
|
|
sync = function(self, timeout_)
|
|
vim.wait(timeout_ or 1000, function()
|
|
return not self.running
|
|
end)
|
|
end,
|
|
stop = function()
|
|
time = nil
|
|
timer:stop()
|
|
end,
|
|
}, {
|
|
__call = function(self, ...)
|
|
local args = { ... }
|
|
|
|
if time == nil then
|
|
time = vim.loop.now()
|
|
end
|
|
|
|
self.running = true
|
|
timer:stop()
|
|
timer:start(math.max(1, self.timeout - (vim.loop.now() - time)), 0, function()
|
|
vim.schedule(function()
|
|
time = nil
|
|
fn(unpack(args))
|
|
self.running = false
|
|
end)
|
|
end)
|
|
end,
|
|
})
|
|
end
|
|
|
|
---Control async tasks.
|
|
async.step = function(...)
|
|
local tasks = { ... }
|
|
local next
|
|
next = function(...)
|
|
if #tasks > 0 then
|
|
table.remove(tasks, 1)(next, ...)
|
|
end
|
|
end
|
|
table.remove(tasks, 1)(next)
|
|
end
|
|
|
|
---Timeout callback function
|
|
---@param fn function
|
|
---@param timeout integer
|
|
---@return function
|
|
async.timeout = function(fn, timeout)
|
|
local timer
|
|
local done = false
|
|
local callback = function(...)
|
|
if not done then
|
|
done = true
|
|
timer:stop()
|
|
timer:close()
|
|
fn(...)
|
|
end
|
|
end
|
|
timer = vim.loop.new_timer()
|
|
timer:start(timeout, 0, function()
|
|
callback()
|
|
end)
|
|
return callback
|
|
end
|
|
|
|
---@alias cmp.AsyncDedup fun(callback: function): function
|
|
|
|
---Create deduplicated callback
|
|
---@return function
|
|
async.dedup = function()
|
|
local id = 0
|
|
return function(callback)
|
|
id = id + 1
|
|
|
|
local current = id
|
|
return function(...)
|
|
if current == id then
|
|
callback(...)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
---Convert async process as sync
|
|
async.sync = function(runner, timeout)
|
|
local done = false
|
|
runner(function()
|
|
done = true
|
|
end)
|
|
vim.wait(timeout, function()
|
|
return done
|
|
end, 10, false)
|
|
end
|
|
|
|
---Wait and callback for next safe state.
|
|
async.debounce_next_tick = function(callback)
|
|
local running = false
|
|
return function()
|
|
if running then
|
|
return
|
|
end
|
|
running = true
|
|
vim.schedule(function()
|
|
running = false
|
|
callback()
|
|
end)
|
|
end
|
|
end
|
|
|
|
return async
|