107 lines
3.4 KiB
JavaScript
107 lines
3.4 KiB
JavaScript
const content = `--[[
|
|
Promotes a job that is currently "delayed" to the "waiting" state
|
|
Input:
|
|
KEYS[1] 'delayed'
|
|
KEYS[2] 'wait'
|
|
KEYS[3] 'paused'
|
|
KEYS[4] 'meta'
|
|
KEYS[5] 'prioritized'
|
|
KEYS[6] 'active'
|
|
KEYS[7] 'pc' priority counter
|
|
KEYS[8] 'event stream'
|
|
KEYS[9] 'marker'
|
|
ARGV[1] queue.toKey('')
|
|
ARGV[2] jobId
|
|
Output:
|
|
0 - OK
|
|
-3 - Job not in delayed zset.
|
|
Events:
|
|
'waiting'
|
|
]]
|
|
local rcall = redis.call
|
|
local jobId = ARGV[2]
|
|
-- Includes
|
|
--[[
|
|
Function to add job in target list and add marker if needed.
|
|
]]
|
|
-- Includes
|
|
--[[
|
|
Add marker if needed when a job is available.
|
|
]]
|
|
local function addBaseMarkerIfNeeded(markerKey, isPausedOrMaxed)
|
|
if not isPausedOrMaxed then
|
|
rcall("ZADD", markerKey, 0, "0")
|
|
end
|
|
end
|
|
local function addJobInTargetList(targetKey, markerKey, pushCmd, isPausedOrMaxed, jobId)
|
|
rcall(pushCmd, targetKey, jobId)
|
|
addBaseMarkerIfNeeded(markerKey, isPausedOrMaxed)
|
|
end
|
|
--[[
|
|
Function to add job considering priority.
|
|
]]
|
|
-- Includes
|
|
--[[
|
|
Function to get priority score.
|
|
]]
|
|
local function getPriorityScore(priority, priorityCounterKey)
|
|
local prioCounter = rcall("INCR", priorityCounterKey)
|
|
return priority * 0x100000000 + prioCounter % 0x100000000
|
|
end
|
|
local function addJobWithPriority(markerKey, prioritizedKey, priority, jobId, priorityCounterKey,
|
|
isPausedOrMaxed)
|
|
local score = getPriorityScore(priority, priorityCounterKey)
|
|
rcall("ZADD", prioritizedKey, score, jobId)
|
|
addBaseMarkerIfNeeded(markerKey, isPausedOrMaxed)
|
|
end
|
|
--[[
|
|
Function to check for the meta.paused key to decide if we are paused or not
|
|
(since an empty list and !EXISTS are not really the same).
|
|
]]
|
|
local function getTargetQueueList(queueMetaKey, activeKey, waitKey, pausedKey)
|
|
local queueAttributes = rcall("HMGET", queueMetaKey, "paused", "concurrency", "max", "duration")
|
|
if queueAttributes[1] then
|
|
return pausedKey, true, queueAttributes[3], queueAttributes[4]
|
|
else
|
|
if queueAttributes[2] then
|
|
local activeCount = rcall("LLEN", activeKey)
|
|
if activeCount >= tonumber(queueAttributes[2]) then
|
|
return waitKey, true, queueAttributes[3], queueAttributes[4]
|
|
else
|
|
return waitKey, false, queueAttributes[3], queueAttributes[4]
|
|
end
|
|
end
|
|
end
|
|
return waitKey, false, queueAttributes[3], queueAttributes[4]
|
|
end
|
|
if rcall("ZREM", KEYS[1], jobId) == 1 then
|
|
local jobKey = ARGV[1] .. jobId
|
|
local priority = tonumber(rcall("HGET", jobKey, "priority")) or 0
|
|
local metaKey = KEYS[4]
|
|
local markerKey = KEYS[9]
|
|
-- Remove delayed "marker" from the wait list if there is any.
|
|
-- Since we are adding a job we do not need the marker anymore.
|
|
-- Markers in waitlist DEPRECATED in v5: Remove in v6.
|
|
local target, isPausedOrMaxed = getTargetQueueList(metaKey, KEYS[6], KEYS[2], KEYS[3])
|
|
local marker = rcall("LINDEX", target, 0)
|
|
if marker and string.sub(marker, 1, 2) == "0:" then rcall("LPOP", target) end
|
|
if priority == 0 then
|
|
-- LIFO or FIFO
|
|
addJobInTargetList(target, markerKey, "LPUSH", isPausedOrMaxed, jobId)
|
|
else
|
|
addJobWithPriority(markerKey, KEYS[5], priority, jobId, KEYS[7], isPausedOrMaxed)
|
|
end
|
|
rcall("XADD", KEYS[8], "*", "event", "waiting", "jobId", jobId, "prev",
|
|
"delayed");
|
|
rcall("HSET", jobKey, "delay", 0)
|
|
return 0
|
|
else
|
|
return -3
|
|
end
|
|
`;
|
|
export const promote = {
|
|
name: 'promote',
|
|
content,
|
|
keys: 9,
|
|
};
|
|
//# sourceMappingURL=promote-9.js.map
|