107 lines
2.5 KiB
JavaScript
107 lines
2.5 KiB
JavaScript
var uuid = 0;
|
|
|
|
function stableSort(arr, compare) {
|
|
var wrapper = arr.map(function (item, idx) {
|
|
return { item: item, idx: idx };
|
|
});
|
|
|
|
wrapper.sort(function (a, b) {
|
|
return compare(a.item, b.item) || (a.idx - b.idx);
|
|
});
|
|
|
|
return wrapper.map(function (w) { return w.item });
|
|
}
|
|
|
|
function MemoryStore() {
|
|
this._queue = []; // Array of taskIds
|
|
this._tasks = {}; // Map of taskId => task
|
|
this._priorities = {}; // Map of taskId => priority
|
|
this._running = {}; // Map of lockId => taskIds
|
|
}
|
|
|
|
MemoryStore.prototype.connect = function (cb) {
|
|
cb(null, this._queue.length);
|
|
}
|
|
|
|
MemoryStore.prototype.getTask = function (taskId, cb) {
|
|
return cb(null, this._tasks[taskId]);
|
|
}
|
|
|
|
MemoryStore.prototype.deleteTask = function (taskId, cb) {
|
|
var self = this;
|
|
var hadTask = self._tasks[taskId];
|
|
delete self._tasks[taskId];
|
|
delete self._priorities[taskId];
|
|
if (hadTask) {
|
|
self._queue.splice(self._queue.indexOf(taskId), 1);
|
|
}
|
|
cb();
|
|
}
|
|
|
|
MemoryStore.prototype.putTask = function (taskId, task, priority, cb) {
|
|
var self = this;
|
|
var hadTask = self._tasks[taskId];
|
|
self._tasks[taskId] = task;
|
|
if (!hadTask) {
|
|
self._queue.push(taskId);
|
|
}
|
|
if (priority !== undefined) {
|
|
self._priorities[taskId] = priority;
|
|
self._queue = stableSort(self._queue, function (a, b) {
|
|
if (self._priorities[a] < self._priorities[b]) return 1;
|
|
if (self._priorities[a] > self._priorities[b]) return -1;
|
|
return 0;
|
|
})
|
|
}
|
|
cb();
|
|
}
|
|
|
|
MemoryStore.prototype.takeFirstN = function (n, cb) {
|
|
var self = this;
|
|
var lockId = uuid++;;
|
|
var taskIds = self._queue.splice(0, n);
|
|
var tasks = {};
|
|
taskIds.forEach(function (taskId) {
|
|
tasks[taskId] = self._tasks[taskId];
|
|
delete self._tasks[taskId];
|
|
})
|
|
if (taskIds.length > 0) {
|
|
self._running[lockId] = tasks;
|
|
}
|
|
cb(null, lockId);
|
|
}
|
|
|
|
MemoryStore.prototype.takeLastN = function (n, cb) {
|
|
var self = this;
|
|
var lockId = uuid++;
|
|
var taskIds = self._queue.splice(-n).reverse();
|
|
var tasks = {};
|
|
taskIds.forEach(function (taskId) {
|
|
tasks[taskId] = self._tasks[taskId];
|
|
delete self._tasks[taskId];
|
|
})
|
|
if (taskIds.length > 0) {
|
|
self._running[lockId] = tasks;
|
|
}
|
|
cb(null, lockId);
|
|
}
|
|
|
|
MemoryStore.prototype.getLock = function (lockId, cb) {
|
|
var self = this;
|
|
cb(null, self._running[lockId]);
|
|
}
|
|
|
|
MemoryStore.prototype.getRunningTasks = function (cb) {
|
|
var self = this;
|
|
cb(null, self._running);
|
|
}
|
|
|
|
MemoryStore.prototype.releaseLock = function (lockId, cb) {
|
|
var self = this;
|
|
delete self._running[lockId];
|
|
cb();
|
|
}
|
|
|
|
|
|
module.exports = MemoryStore;
|