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.

146 lines
3.2 KiB

2 years ago
'use strict';
function throatInternal(size) {
var queue = new Queue();
var s = size | 0;
function run(fn, self, args) {
if ((s | 0) !== 0) {
s = (s | 0) - 1;
return new Promise(function (resolve) {
resolve(fn.apply(self, args));
}).then(onFulfill, onReject);
}
return new Promise(function (resolve) {
queue.push(new Delayed(resolve, fn, self, args));
}).then(runDelayed);
}
function runDelayed(d) {
try {
return Promise.resolve(d.fn.apply(d.self, d.args)).then(
onFulfill,
onReject
);
} catch (ex) {
onReject(ex);
}
}
function onFulfill(result) {
release();
return result;
}
function onReject(error) {
release();
throw error;
}
function release() {
var next = queue.shift();
if (next) {
next.resolve(next);
} else {
s = (s | 0) + 1;
}
}
return run;
}
function earlyBound(size, fn) {
const run = throatInternal(size | 0);
return function () {
var args = new Array(arguments.length);
for (var i = 0; i < arguments.length; i++) {
args[i] = arguments[i];
}
return run(fn, this, args);
};
}
function lateBound(size) {
const run = throatInternal(size | 0);
return function (fn) {
if (typeof fn !== 'function') {
throw new TypeError(
'Expected throat fn to be a function but got ' + typeof fn
);
}
var args = new Array(arguments.length - 1);
for (var i = 1; i < arguments.length; i++) {
args[i - 1] = arguments[i];
}
return run(fn, this, args);
};
}
module.exports = function throat(size, fn) {
if (typeof size === 'function') {
var temp = fn;
fn = size;
size = temp;
}
if (typeof size !== 'number') {
throw new TypeError(
'Expected throat size to be a number but got ' + typeof size
);
}
if (fn !== undefined && typeof fn !== 'function') {
throw new TypeError(
'Expected throat fn to be a function but got ' + typeof fn
);
}
if (typeof fn === 'function') {
return earlyBound(size | 0, fn);
} else {
return lateBound(size | 0);
}
};
module.exports.default = module.exports;
function Delayed(resolve, fn, self, args) {
this.resolve = resolve;
this.fn = fn;
this.self = self || null;
this.args = args;
}
var blockSize = 64;
function Queue() {
this._s1 = [];
this._s2 = [];
this._shiftBlock = this._pushBlock = new Array(blockSize);
this._pushIndex = 0;
this._shiftIndex = 0;
}
Queue.prototype.push = function (value) {
if (this._pushIndex === blockSize) {
this._pushIndex = 0;
this._s1[this._s1.length] = this._pushBlock = new Array(blockSize);
}
this._pushBlock[this._pushIndex++] = value;
};
Queue.prototype.shift = function () {
if (this._shiftIndex === blockSize) {
this._shiftIndex = 0;
var s2 = this._s2;
if (s2.length === 0) {
var s1 = this._s1;
if (s1.length === 0) {
return undefined;
}
this._s1 = s2;
s2 = this._s2 = s1.reverse();
}
this._shiftBlock = s2.pop();
}
if (
this._pushBlock === this._shiftBlock &&
this._pushIndex === this._shiftIndex
) {
return undefined;
}
var result = this._shiftBlock[this._shiftIndex];
this._shiftBlock[this._shiftIndex++] = null;
return result;
};