diff options
author | 最萌小汐 <sumneko@hotmail.com> | 2019-11-23 00:05:30 +0800 |
---|---|---|
committer | 最萌小汐 <sumneko@hotmail.com> | 2019-11-23 00:05:30 +0800 |
commit | 6da2b175e20ed3c03b0dfcfc9046de1e0e5d4444 (patch) | |
tree | fdc22d78150fd1c5edc46732c8b151ccfefb519f /script-beta/await.lua | |
parent | d0ff66c9abe9d6abbca12fd811e0c3cb69c1033a (diff) | |
download | lua-language-server-6da2b175e20ed3c03b0dfcfc9046de1e0e5d4444.zip |
正路目录
Diffstat (limited to 'script-beta/await.lua')
-rw-r--r-- | script-beta/await.lua | 100 |
1 files changed, 100 insertions, 0 deletions
diff --git a/script-beta/await.lua b/script-beta/await.lua new file mode 100644 index 00000000..5a960e96 --- /dev/null +++ b/script-beta/await.lua @@ -0,0 +1,100 @@ +local timer = require 'timer' + +---@class await +local m = {} +m.type = 'await' + +m.coTracker = setmetatable({}, { __mode = 'k' }) +m.delayQueue = {} +m.delayQueueIndex = 1 + +--- 设置错误处理器 +---@param errHandle function {comment = '当有错误发生时,会以错误堆栈为参数调用该函数'} +function m.setErrorHandle(errHandle) + m.errorHandle = errHandle +end + +function m.checkResult(co, ...) + local suc, err = ... + if not suc and m.errorHandle then + m.errorHandle(debug.traceback(co, err)) + end + return ... +end + +--- 创建一个任务 +function m.create(callback, ...) + local co = coroutine.create(callback) + m.coTracker[co] = true + return m.checkResult(co, coroutine.resume(co, ...)) +end + +--- 休眠一段时间 +---@param time number +function m.sleep(time, getVersion) + if not coroutine.isyieldable() then + if m.errorHandle then + m.errorHandle(debug.traceback('Cannot yield')) + end + return + end + local version = getVersion and getVersion() + local co = coroutine.running() + timer.wait(time, function () + if version == (getVersion and getVersion()) then + return m.checkResult(co, coroutine.resume(co)) + else + coroutine.close(co) + end + end) + return coroutine.yield(getVersion) +end + +--- 等待直到唤醒 +---@param callback function +function m.wait(callback, ...) + if not coroutine.isyieldable() then + return + end + local co = coroutine.running() + callback(function (...) + return m.checkResult(co, coroutine.resume(co, ...)) + end) + return coroutine.yield(...) +end + +--- 延迟 +function m.delay(getVersion) + if not coroutine.isyieldable() then + return + end + local co = coroutine.running() + local version = getVersion and getVersion() + m.delayQueue[#m.delayQueue+1] = function () + if version == (getVersion and getVersion()) then + return m.checkResult(co, coroutine.resume(co)) + else + coroutine.close(co) + end + end + return coroutine.yield() +end + +--- 步进 +function m.step() + local waker = m.delayQueue[m.delayQueueIndex] + if waker then + m.delayQueue[m.delayQueueIndex] = false + m.delayQueueIndex = m.delayQueueIndex + 1 + waker() + return true + else + for i = 1, #m.delayQueue do + m.delayQueue[i] = nil + end + m.delayQueueIndex = 1 + return false + end +end + +return m |