add jobpool

1 files changed, 122 insertions, 0 deletions

diff --git a/xmake/modules/private/async/jobpool.lua b/xmake/modules/private/async/jobpool.lua
new file mode 100644
index 000000000..b0dc56454
--- /dev/null
+++ b/xmake/modules/private/async/jobpool.lua
@@ -0,0 +1,122 @@
+--!A cross-platform build utility based on Lua
+--
+-- Licensed under the Apache License, Version 2.0 (the "License");
+-- you may not use this file except in compliance with the License.
+-- You may obtain a copy of the License at
+--
+--     http://www.apache.org/licenses/LICENSE-2.0
+--
+-- Unless required by applicable law or agreed to in writing, software
+-- distributed under the License is distributed on an "AS IS" BASIS,
+-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+-- See the License for the specific language governing permissions and
+-- limitations under the License.
+--
+-- Copyright (C) 2015-2020, TBOOX Open Source Group.
+--
+-- @author      ruki
+-- @file        jobpool.lua
+--
+
+-- imports
+import("core.base.dlist")
+import("core.base.object")
+
+-- define module
+local jobpool = jobpool or object {_init = {"_count", "_rootjob", "_leafjobs"}}
+
+-- get jobs count
+function jobpool:count()
+    return self._count
+end
+
+-- get root job
+function jobpool:rootjob()
+    return self._rootjob
+end
+
+-- add job to the given job node
+--
+-- @param rootjob   the root job node
+-- @param name      the job name
+-- @param run       the run command/script
+--
+function jobpool:addjob(name, run, rootjob)
+    rootjob = rootjob or self:rootjob()
+    local job = {name = name, run = run, _parent = rootjob}
+    rootjob._deps = rootjob._deps or dlist:new()
+    rootjob._deps:push(job)
+    self._count = self._count + 1
+    return job
+end
+
+-- pop job without deps at leaf node 
+function jobpool:popjob()
+
+    -- no jobs?
+    if self:count() == 0 then
+        return 
+    end
+
+    -- init leaf jobs first
+    local leafjobs = self._leafjobs
+    if #leafjobs == 0 then
+        self:_genleafjobs(self:rootjob(), leafjobs)
+    end
+
+    -- pop a job from the leaf jobs
+    if #leafjobs > 0 then
+
+        -- update jobs count
+        self._count = self._count - 1
+
+        -- get job
+        local job = leafjobs[#leafjobs]
+        table.remove(leafjobs, #leafjobs)
+
+        -- remove this job from the parent node
+        local priority = job._priority or 0
+        local parent = assert(job._parent, "invalid job without parent node!")
+        parent._priority = math.max(parent._priority or 0, priority + 1)
+        parent._deps:remove(job)
+        if parent._deps:empty() and self._count > 0 then
+            table.insert(leafjobs, 1, parent)
+        end
+        return job, priority
+    end
+end
+
+-- generate all leaf jobs from the given job
+function jobpool:_genleafjobs(job, leafjobs)
+    local deps = job._deps
+    if deps and not deps:empty() then
+        for dep in deps:items() do
+            self:_genleafjobs(dep, leafjobs)
+        end
+    else
+        table.insert(leafjobs, job)
+    end
+end
+
+-- generate jobs tree for the given job
+function jobpool:_gentree(job)
+    local tree = {job.name}
+    local deps = job._deps
+    if deps and not deps:empty() then
+        tree[2] = {}
+        for dep in deps:items() do
+            table.insert(tree[2], self:_gentree(dep))
+        end
+    end
+    return tree
+end
+
+-- tostring
+function jobpool:__tostring()
+    return string.serialize(self:_gentree(self:rootjob()), {indent = 2})
+end
+
+-- new a jobpool
+function new()
+    return jobpool {0, {name = "root"}, {}}
+end