128 lines
4.2 KiB
JavaScript
128 lines
4.2 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.changePriority = void 0;
|
|
const content = `--[[
|
|
Change job priority
|
|
Input:
|
|
KEYS[1] 'wait',
|
|
KEYS[2] 'paused'
|
|
KEYS[3] 'meta'
|
|
KEYS[4] 'prioritized'
|
|
KEYS[5] 'active'
|
|
KEYS[6] 'pc' priority counter
|
|
KEYS[7] 'marker'
|
|
ARGV[1] priority value
|
|
ARGV[2] prefix key
|
|
ARGV[3] job id
|
|
ARGV[4] lifo
|
|
Output:
|
|
0 - OK
|
|
-1 - Missing job
|
|
]]
|
|
local jobId = ARGV[3]
|
|
local jobKey = ARGV[2] .. jobId
|
|
local priority = tonumber(ARGV[1])
|
|
local rcall = redis.call
|
|
-- 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
|
|
--[[
|
|
Function to push back job considering priority in front of same prioritized jobs.
|
|
]]
|
|
local function pushBackJobWithPriority(prioritizedKey, priority, jobId)
|
|
-- in order to put it at front of same prioritized jobs
|
|
-- we consider prioritized counter as 0
|
|
local score = priority * 0x100000000
|
|
rcall("ZADD", prioritizedKey, score, jobId)
|
|
end
|
|
local function reAddJobWithNewPriority( prioritizedKey, markerKey, targetKey,
|
|
priorityCounter, lifo, priority, jobId, isPausedOrMaxed)
|
|
if priority == 0 then
|
|
local pushCmd = lifo and 'RPUSH' or 'LPUSH'
|
|
addJobInTargetList(targetKey, markerKey, pushCmd, isPausedOrMaxed, jobId)
|
|
else
|
|
if lifo then
|
|
pushBackJobWithPriority(prioritizedKey, priority, jobId)
|
|
else
|
|
addJobWithPriority(markerKey, prioritizedKey, priority, jobId,
|
|
priorityCounter, isPausedOrMaxed)
|
|
end
|
|
end
|
|
end
|
|
if rcall("EXISTS", jobKey) == 1 then
|
|
local metaKey = KEYS[3]
|
|
local target, isPausedOrMaxed = getTargetQueueList(metaKey, KEYS[5], KEYS[1], KEYS[2])
|
|
local prioritizedKey = KEYS[4]
|
|
local priorityCounterKey = KEYS[6]
|
|
local markerKey = KEYS[7]
|
|
-- Re-add with the new priority
|
|
if rcall("ZREM", prioritizedKey, jobId) > 0 then
|
|
reAddJobWithNewPriority( prioritizedKey, markerKey, target,
|
|
priorityCounterKey, ARGV[4] == '1', priority, jobId, isPausedOrMaxed)
|
|
elseif rcall("LREM", target, -1, jobId) > 0 then
|
|
reAddJobWithNewPriority( prioritizedKey, markerKey, target,
|
|
priorityCounterKey, ARGV[4] == '1', priority, jobId, isPausedOrMaxed)
|
|
end
|
|
rcall("HSET", jobKey, "priority", priority)
|
|
return 0
|
|
else
|
|
return -1
|
|
end
|
|
`;
|
|
exports.changePriority = {
|
|
name: 'changePriority',
|
|
content,
|
|
keys: 7,
|
|
};
|
|
//# sourceMappingURL=changePriority-7.js.map
|