summaryrefslogtreecommitdiff
path: root/script/pub/pub.lua
diff options
context:
space:
mode:
author最萌小汐 <sumneko@hotmail.com>2021-02-09 15:33:08 +0800
committer最萌小汐 <sumneko@hotmail.com>2021-02-09 15:33:08 +0800
commit438fa5d57a8799a45f99afaa499fb658d414a54d (patch)
tree8e873e861a799f2c6fe795f96ae804067b7d4742 /script/pub/pub.lua
parentbc5e23e547f37eef194a7d6cdfbde14b73472bd2 (diff)
downloadlua-language-server-438fa5d57a8799a45f99afaa499fb658d414a54d.zip
share channel
Diffstat (limited to 'script/pub/pub.lua')
-rw-r--r--script/pub/pub.lua142
1 files changed, 39 insertions, 103 deletions
diff --git a/script/pub/pub.lua b/script/pub/pub.lua
index ad1cd749..c145e982 100644
--- a/script/pub/pub.lua
+++ b/script/pub/pub.lua
@@ -1,9 +1,13 @@
local thread = require 'bee.thread'
local utility = require 'utility'
local await = require 'await'
-local timer = require 'timer'
+
+thread.newchannel 'taskpad'
+thread.newchannel 'waiter'
local errLog = thread.channel 'errlog'
+local taskPad = thread.channel 'taskpad'
+local waiter = thread.channel 'waiter'
local type = type
local counter = utility.counter()
@@ -24,11 +28,12 @@ brave.register(%d)
]]
---@class pub
-local m = {}
-m.type = 'pub'
-m.braves = {}
-m.ability = {}
+local m = {}
+m.type = 'pub'
+m.braves = {}
+m.ability = {}
m.taskQueue = {}
+m.taskMap = {}
--- 注册酒馆的功能
function m.on(name, callback)
@@ -41,12 +46,8 @@ function m.recruitBraves(num)
for _ = 1, num do
local id = #m.braves + 1
log.info('Create brave:', id)
- thread.newchannel('taskpad' .. id)
- thread.newchannel('waiter' .. id)
m.braves[id] = {
id = id,
- taskpad = thread.channel('taskpad' .. id),
- waiter = thread.channel('waiter' .. id),
thread = thread.thread(braveTemplate:format(
package.path,
package.cpath,
@@ -63,32 +64,24 @@ function m.recruitBraves(num)
end
end
---- 勇者是否有空
-function m.isIdle(brave)
- return next(brave.taskMap) == nil
-end
-
--- 给勇者推送任务
-function m.pushTask(brave, info)
+function m.pushTask(info)
if info.removed then
return false
end
- brave.taskpad:push(info.name, info.id, info.params)
- brave.taskMap[info.id] = info
- --log.info(('Push task %q(%d) to # %d, queue length %d'):format(info.name, info.id, brave.id, #m.taskQueue))
+ taskPad:push(info.name, info.id, info.params)
+ m.taskMap[info.id] = info
return true
end
--- 从勇者处接收任务反馈
function m.popTask(brave, id, result)
- local info = brave.taskMap[id]
+ local info = m.taskMap[id]
if not info then
log.warn(('Brave pushed unknown task result: # %d => [%d]'):format(brave.id, id))
return
end
- brave.taskMap[id] = nil
- --log.info(('Pop task %q(%d) from # %d'):format(info.name, info.id, brave.id))
- m.checkWaitingTask(brave)
+ m.taskMap[id] = nil
if not info.removed then
info.removed = true
if info.callback then
@@ -116,25 +109,13 @@ function m.awaitTask(name, params)
name = name,
params = params,
}
- for _, brave in ipairs(m.braves) do
- if m.isIdle(brave) then
- if m.pushTask(brave, info) then
- return await.wait(function (waker)
- info.callback = waker
- end)
- else
- return nil
- end
- end
+ if m.pushTask(info) then
+ return await.wait(function (waker)
+ info.callback = waker
+ end)
+ else
+ return false
end
- -- 如果所有勇者都在战斗,那么把任务缓存到队列里
- -- 当有勇者提交任务反馈后,尝试把按顺序将堆积任务
- -- 交给该勇者
- m.taskQueue[#m.taskQueue+1] = info
- --log.info(('Add task %q(%d) in queue, length %d.'):format(name, info.id, #m.taskQueue))
- return await.wait(function (waker)
- info.callback = waker
- end)
end
--- 发布同步任务,如果任务进入了队列,会返回执行器
@@ -149,77 +130,32 @@ function m.task(name, params, callback)
params = params,
callback = callback,
}
- for _, brave in ipairs(m.braves) do
- if m.isIdle(brave) then
- m.pushTask(brave, info)
- return nil
- end
- end
- -- 如果所有勇者都在战斗,那么把任务缓存到队列里
- -- 当有勇者提交任务反馈后,尝试把按顺序将堆积任务
- -- 交给该勇者
- m.taskQueue[#m.taskQueue+1] = info
- --log.info(('Add task %q(%d) in queue, length %d.'):format(name, info.id, #m.taskQueue))
- return info
-end
-
---- 插队
-function m.jumpQueue(info)
- for i = 2, #m.taskQueue do
- if m.taskQueue[i] == info then
- m.taskQueue[i] = nil
- table.move(m.taskQueue, 1, i - 1, 2)
- m.taskQueue[1] = info
- return
- end
- end
-end
-
---- 移除任务
-function m.remove(info)
- info.removed = true
- for i = 1, #m.taskQueue do
- if m.taskQueue[i] == info then
- table.remove(m.taskQueue[i], i)
- return
- end
- end
-end
-
---- 检查堆积任务
-function m.checkWaitingTask(brave)
- if #m.taskQueue == 0 then
- return
- end
- -- 如果勇者还有其他活要忙,那么让他继续忙去吧
- if next(brave.taskMap) then
- return
- end
- while #m.taskQueue > 0 do
- local info = table.remove(m.taskQueue, 1)
- if m.pushTask(brave, info) then
- break
- end
- end
+ return m.pushTask(info)
end
--- 接收反馈
----|返回接收到的反馈数量
+--- 返回接收到的反馈数量
---@return integer
-function m.recieve()
- for _, brave in ipairs(m.braves) do
+function m.recieve(block)
+ if block then
+ local id, name, result = waiter:bpop()
+ if type(name) == 'string' then
+ m.popReport(m.braves[id], name, result)
+ else
+ m.popTask(m.braves[id], name, result)
+ end
+ else
while true do
- local suc, id, result = brave.waiter:pop()
+ local suc, id, name, result = waiter:pop()
if not suc then
- goto CONTINUE
+ break
end
- if type(id) == 'string' then
- m.popReport(brave, id, result)
+ if type(name) == 'string' then
+ m.popReport(m.braves[id], name, result)
else
- m.popTask(brave, id, result)
+ m.popTask(m.braves[id], name, result)
end
end
- ::CONTINUE::
end
end
@@ -234,9 +170,9 @@ function m.checkDead()
end
end
-function m.step()
+function m.step(block)
m.checkDead()
- m.recieve()
+ m.recieve(block)
end
return m