OC-PsychOS2/module/sched.lua

82 lines
2.2 KiB
Lua
Raw Normal View History

do
local tTasks,nPid,nTimeout,cPid = {},1,0.25,0 -- table of tasks, next process ID, event timeout, current PID
function os.spawn(f,n) -- function string -- number -- creates a process from function *f* with name *n*
2019-10-27 23:34:02 +11:00
tTasks[nPid] = {
c=coroutine.create(function()
local rt = {pcall(f)}
if not rt[1] then
syslog(rt[2])
end
computer.pushSignal("process_finished",os.pid(),table.unpack(rt))
end), -- actual coroutine
2019-10-27 23:34:02 +11:00
n=n, -- process name
p=nPid, -- process PID
P=cPid, -- parent PID
2023-07-04 18:25:11 +10:00
t=0, -- CPU time
T=0, -- total uptime
2019-10-27 23:34:02 +11:00
e={} -- environment variables
}
if tTasks[cPid] then
for k,v in pairs(tTasks[cPid].e) do
tTasks[nPid].e[k] = tTasks[nPid].e[k] or v
end
2018-11-03 03:05:41 +11:00
end
nPid = nPid + 1
return nPid - 1
end
function os.kill(pid) -- number -- -- removes process *pid* from the task list
tTasks[pid] = nil
end
function os.pid() -- -- number -- returns the current process' PID
2019-10-27 23:34:02 +11:00
return cPid
end
function os.tasks() -- -- table -- returns a table of process IDs
2019-10-27 23:34:02 +11:00
local rt = {}
for k,v in pairs(tTasks) do
rt[#rt+1] = k
end
return rt
end
function os.taskInfo(pid) -- number -- table -- returns info on process *pid* as a table with name and parent values
pid = pid or os.pid()
if not tTasks[pid] then return false end
2023-07-04 18:25:11 +10:00
return {name=tTasks[pid].n,parent=tTasks[pid].P,cputime=tTasks[pid].t,iotime=tTasks[pid].T}
2019-10-27 23:34:02 +11:00
end
function os.sched() -- the actual scheduler function
os.sched = nil
2018-11-03 03:05:41 +11:00
while #tTasks > 0 do
local tEv = {computer.pullSignal(nTimeout)}
for k,v in pairs(tTasks) do
if coroutine.status(v.c) ~= "dead" then
cPid = k
2023-07-04 18:25:11 +10:00
local sT, sC = os.clock(), computer.uptime()
2018-11-03 03:05:41 +11:00
coroutine.resume(v.c,table.unpack(tEv))
2023-07-04 18:25:11 +10:00
v.t, v.T = v.t + os.clock() - sT, v.T + computer.uptime() - sC
2018-11-03 03:05:41 +11:00
else
tTasks[k] = nil
end
end
end
end
function os.setenv(k,v) -- set's the current process' environment variable *k* to *v*, which is passed to children
if tTasks[cPid] then
tTasks[cPid].e[k] = v
end
end
function os.getenv(k) -- gets a process' *k* environment variable
if tTasks[cPid] then
return tTasks[cPid].e[k]
end
end
function os.getTimeout()
return nTimeout
end
function os.setTimeout(n)
if type(n) == "number" and n >= 0 then
nTimeout = n
return true
end
return false
end
end