You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.4 KiB
Lua
69 lines
1.4 KiB
Lua
3 years ago
|
--[=[
|
||
|
|
@c Mutex
|
||
|
|
@t ui
|
||
|
|
@mt mem
|
||
|
|
@d Mutual exclusion class used to control Lua coroutine execution order.
|
||
|
|
]=]
|
||
|
|
|
||
|
|
local Deque = require('./Deque')
|
||
|
|
local timer = require('timer')
|
||
|
|
|
||
|
|
local yield = coroutine.yield
|
||
|
|
local resume = coroutine.resume
|
||
|
|
local running = coroutine.running
|
||
|
|
local setTimeout = timer.setTimeout
|
||
|
|
|
||
|
|
local Mutex = require('class')('Mutex', Deque)
|
||
|
|
|
||
|
|
function Mutex:__init()
|
||
|
|
Deque.__init(self)
|
||
|
|
self._active = false
|
||
|
|
end
|
||
|
|
|
||
|
|
--[=[
|
||
|
|
@m lock
|
||
|
|
@op prepend boolean
|
||
|
|
@r nil
|
||
|
|
@d If the mutex is not active (if a coroutine is not queued), this will activate
|
||
|
|
the mutex; otherwise, this will yield and queue the current coroutine.
|
||
|
|
]=]
|
||
|
|
function Mutex:lock(prepend)
|
||
|
|
if self._active then
|
||
|
|
if prepend then
|
||
|
|
return yield(self:pushLeft(running()))
|
||
|
|
else
|
||
|
|
return yield(self:pushRight(running()))
|
||
|
|
end
|
||
|
|
else
|
||
|
|
self._active = true
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
--[=[
|
||
|
|
@m unlock
|
||
|
|
@r nil
|
||
|
|
@d If the mutex is active (if a coroutine is queued), this will dequeue and
|
||
|
|
resume the next available coroutine; otherwise, this will deactivate the mutex.
|
||
|
|
]=]
|
||
|
|
function Mutex:unlock()
|
||
|
|
if self:getCount() > 0 then
|
||
|
|
return assert(resume(self:popLeft()))
|
||
|
|
else
|
||
|
|
self._active = false
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
--[=[
|
||
|
|
@m unlockAfter
|
||
|
|
@p delay number
|
||
|
|
@r uv_timer
|
||
|
|
@d Asynchronously unlocks the mutex after a specified time in milliseconds.
|
||
|
|
The relevant `uv_timer` object is returned.
|
||
|
|
]=]
|
||
|
|
local unlock = Mutex.unlock
|
||
|
|
function Mutex:unlockAfter(delay)
|
||
|
|
return setTimeout(delay, unlock, self)
|
||
|
|
end
|
||
|
|
|
||
|
|
return Mutex
|