summaryrefslogtreecommitdiff
path: root/script/brave/brave.lua
blob: c497c0935f5513c4317083a501b60ae39cb5b9f2 (plain)
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
local thread = require 'bee.thread'

---@class pub_brave
local m = {}
m.type = 'brave'
m.ability = {}
m.queue = {}

--- 注册成为勇者
function m.register(id)
    m.taskpad = thread.channel('taskpad' .. id)
    m.waiter  = thread.channel('waiter'  .. id)
    m.id      = id

    if #m.queue > 0 then
        for _, info in ipairs(m.queue) do
            m.waiter:push(info.name, info.params)
        end
    end
    m.queue = nil

    m.start()
end

--- 注册能力
function m.on(name, callback)
    m.ability[name] = callback
end

--- 报告
function m.push(name, params)
    if m.waiter then
        m.waiter:push(name, params)
    else
        m.queue[#m.queue+1] = {
            name   = name,
            params = params,
        }
    end
end

--- 开始找工作
function m.start()
    m.push('mem', collectgarbage 'count')
    while true do
        local suc, name, id, params = m.taskpad:pop()
        if not suc then
            -- 找不到工作的勇者,只好睡觉
            thread.sleep(0.01)
            goto CONTINUE
        end
        local ability = m.ability[name]
        -- TODO
        if not ability then
            m.waiter:push(id)
            log.error('Brave can not handle this work: ' .. name)
            goto CONTINUE
        end
        local ok, res = xpcall(ability, log.error, params)
        if ok then
            m.waiter:push(id, res)
        else
            m.waiter:push(id)
        end
        m.push('mem', collectgarbage 'count')
        ::CONTINUE::
    end
end

return m