1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
local thread = require 'bee.thread'
local utility = require 'utility'
local task = require 'task'
local braveTemplate = [[
package.path = %q
package.cpath = %q
local brave = require 'pub.brave'
brave:register(%d)
]]
---@class pub_client
local m = {}
m.type = 'pub.client'
m.braves = {}
--- 招募勇者，勇者会从公告板上领取任务，完成任务后到看板娘处交付任务
---@param num integer
function m:recruitBraves(num)
for _ = 1, num do
local id = #self.braves + 1
log.info('Create pub brave:', id)
thread.newchannel('taskpad' .. id)
thread.newchannel('waiter' .. id)
self.braves[id] = {
id = id,
taskpad = thread.channel('taskpad' .. id),
waiter = thread.channel('waiter' .. id),
thread = thread.thread(braveTemplate:format(
package.path,
package.cpath,
id
)),
taskList = {},
counter = utility.counter(),
currentTask = nil,
}
end
end
--- 勇者是否有空
function m:isIdle(brave)
return brave.currentTask == nil and not next(brave.taskList)
end
--- 给勇者推送任务
function m:pushTask(brave, name, ...)
local taskID = brave.counter()
local co = coroutine.running()
brave.taskpad:push(name, taskID, ...)
brave.taskList[taskID] = co
return coroutine.yield(co)
end
--- 从勇者处接收任务反馈
function m:popTask(brave, id, ...)
local co = brave.taskList[id]
if not co then
log.warn(('Brave pushed unknown task result: [%d] => [%d]'):format(brave.id, id))
return
end
brave.taskList[id] = nil
coroutine.resume(co, ...)
end
--- 发布任务
---@parma name string
function m:task(name, ...)
local _, main = coroutine.running()
if main then
error('不能在主协程中发布任务')
end
for _, brave in ipairs(self.braves) do
if self:isIdle(brave) then
return self:pushTask(brave, name, ...)
end
end
end
return m
|
