| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- "use strict";
- Object.defineProperty(exports, "__esModule", { value: true });
- exports.promote = void 0;
- 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
- `;
- exports.promote = {
- name: 'promote',
- content,
- keys: 9,
- };
- //# sourceMappingURL=promote-9.js.map
|