diff options
| -rw-r--r-- | xmake/modules/async/jobgraph.lua | 21 |
1 files changed, 14 insertions, 7 deletions
diff --git a/xmake/modules/async/jobgraph.lua b/xmake/modules/async/jobgraph.lua index 9994ac1f3..ee3a6f274 100644 --- a/xmake/modules/async/jobgraph.lua +++ b/xmake/modules/async/jobgraph.lua @@ -26,7 +26,7 @@ import("core.base.hashset") -- define module local jobqueue = jobqueue or object {_init = {"_jobgraph", "_dag"}} -local jobgraph = jobgraph or object {_init = {"_name", "_jobs", "_size", "_dag"}} +local jobgraph = jobgraph or object {_init = {"_name", "_jobs", "_size", "_dag", "_groups"}} -- remove the finished job function jobqueue:remove(job) @@ -60,7 +60,7 @@ end -- -- @param name the job name -- @param run the job run command/script --- @param opt the job options +-- @param opt the job options, e.g. {group = "xxx"} -- function jobgraph:add(name, run, opt) local jobs = self._jobs @@ -68,6 +68,16 @@ function jobgraph:add(name, run, opt) local job = {name = name, run = run, opt = opt} jobs[name] = job self._size = self._size + 1 + + local group_name = opt.group + if group_name then + local groups = self._groups[group_name] + if not groups then + groups = {} + self._groups[group_name] = groups + end + table.insert(groups, job) + end end end @@ -98,11 +108,8 @@ function jobgraph:add_deps(...) end prev = curr end -end - --- add jog group -function jobgraph:add_group(name, callback) -- TODO + -- add groups jobs end -- build a job queue @@ -134,5 +141,5 @@ end -- new a jobgraph function new(name) - return jobgraph {name, {}, 0, graph.new(true)} + return jobgraph {name, {}, 0, graph.new(true), {}} end |