You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.5 KiB
58 lines
1.5 KiB
3 years ago
|
"use strict";
|
||
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
|
const utils_1 = require("../utils");
|
||
|
|
const Deque = require("denque");
|
||
|
|
const debug = utils_1.Debug("delayqueue");
|
||
|
|
/**
|
||
|
|
* Queue that runs items after specified duration
|
||
|
|
*
|
||
|
|
* @export
|
||
|
|
* @class DelayQueue
|
||
|
|
*/
|
||
|
|
class DelayQueue {
|
||
|
|
constructor() {
|
||
|
|
this.queues = {};
|
||
|
|
this.timeouts = {};
|
||
|
|
}
|
||
|
|
/**
|
||
|
|
* Add a new item to the queue
|
||
|
|
*
|
||
|
|
* @param {string} bucket bucket name
|
||
|
|
* @param {Function} item function that will run later
|
||
|
|
* @param {IDelayQueueOptions} options
|
||
|
|
* @memberof DelayQueue
|
||
|
|
*/
|
||
|
|
push(bucket, item, options) {
|
||
|
|
const callback = options.callback || process.nextTick;
|
||
|
|
if (!this.queues[bucket]) {
|
||
|
|
this.queues[bucket] = new Deque();
|
||
|
|
}
|
||
|
|
const queue = this.queues[bucket];
|
||
|
|
queue.push(item);
|
||
|
|
if (!this.timeouts[bucket]) {
|
||
|
|
this.timeouts[bucket] = setTimeout(() => {
|
||
|
|
callback(() => {
|
||
|
|
this.timeouts[bucket] = null;
|
||
|
|
this.execute(bucket);
|
||
|
|
});
|
||
|
|
}, options.timeout);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
execute(bucket) {
|
||
|
|
const queue = this.queues[bucket];
|
||
|
|
if (!queue) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const { length } = queue;
|
||
|
|
if (!length) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
debug("send %d commands in %s queue", length, bucket);
|
||
|
|
this.queues[bucket] = null;
|
||
|
|
while (queue.length > 0) {
|
||
|
|
queue.shift()();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
exports.default = DelayQueue;
|