108 lines
4.1 KiB
JavaScript
108 lines
4.1 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.TaskInfo = exports.FixedQueue = exports.ArrayTaskQueue = void 0;
|
|
exports.isTaskQueue = isTaskQueue;
|
|
const node_perf_hooks_1 = require("node:perf_hooks");
|
|
const node_async_hooks_1 = require("node:async_hooks");
|
|
const common_1 = require("../common");
|
|
const symbols_1 = require("../symbols");
|
|
var array_queue_1 = require("./array_queue");
|
|
Object.defineProperty(exports, "ArrayTaskQueue", { enumerable: true, get: function () { return array_queue_1.ArrayTaskQueue; } });
|
|
var fixed_queue_1 = require("./fixed_queue");
|
|
Object.defineProperty(exports, "FixedQueue", { enumerable: true, get: function () { return fixed_queue_1.FixedQueue; } });
|
|
/**
|
|
* Verifies if a given TaskQueue is valid
|
|
*
|
|
* @export
|
|
* @param {*} value
|
|
* @return {*} {boolean}
|
|
*/
|
|
function isTaskQueue(value) {
|
|
return (typeof value === 'object' &&
|
|
value !== null &&
|
|
'size' in value &&
|
|
typeof value.shift === 'function' &&
|
|
typeof value.remove === 'function' &&
|
|
typeof value.push === 'function');
|
|
}
|
|
let taskIdCounter = 0;
|
|
// Extend AsyncResource so that async relations between posting a task and
|
|
// receiving its result are visible to diagnostic tools.
|
|
class TaskInfo extends node_async_hooks_1.AsyncResource {
|
|
constructor(task, transferList, filename, name, callback, abortSignal, triggerAsyncId) {
|
|
super('Piscina.Task', { requireManualDestroy: true, triggerAsyncId });
|
|
// abortListener : (() => void) | null = null;
|
|
this.workerInfo = null;
|
|
this.aborted = false;
|
|
this._abortListener = null;
|
|
this.callback = callback;
|
|
this.task = task;
|
|
this.transferList = transferList;
|
|
// If the task is a Transferable returned by
|
|
// Piscina.move(), then add it to the transferList
|
|
// automatically
|
|
if ((0, common_1.isMovable)(task)) {
|
|
// This condition should never be hit but typescript
|
|
// complains if we dont do the check.
|
|
/* istanbul ignore if */
|
|
if (this.transferList == null) {
|
|
this.transferList = [];
|
|
}
|
|
this.transferList =
|
|
this.transferList.concat(task[symbols_1.kTransferable]);
|
|
this.task = task[symbols_1.kValue];
|
|
}
|
|
this.filename = filename;
|
|
this.name = name;
|
|
// TODO: This should not be global
|
|
this.taskId = taskIdCounter++;
|
|
this.abortSignal = abortSignal;
|
|
this.created = node_perf_hooks_1.performance.now();
|
|
this.started = 0;
|
|
}
|
|
// TODO: improve this handling - ideally should be extended
|
|
set abortListener(value) {
|
|
this._abortListener = () => {
|
|
this.aborted = true;
|
|
value();
|
|
};
|
|
}
|
|
get abortListener() {
|
|
return this._abortListener;
|
|
}
|
|
releaseTask() {
|
|
const ret = this.task;
|
|
this.task = null;
|
|
return ret;
|
|
}
|
|
done(err, result) {
|
|
this.runInAsyncScope(this.callback, null, err, result);
|
|
this.emitDestroy(); // `TaskInfo`s are used only once.
|
|
// If an abort signal was used, remove the listener from it when
|
|
// done to make sure we do not accidentally leak.
|
|
if (this.abortSignal && this.abortListener) {
|
|
if ('removeEventListener' in this.abortSignal && this.abortListener) {
|
|
this.abortSignal.removeEventListener('abort', this.abortListener);
|
|
}
|
|
else {
|
|
this.abortSignal.off('abort', this.abortListener);
|
|
}
|
|
}
|
|
}
|
|
get [symbols_1.kQueueOptions]() {
|
|
var _a, _b;
|
|
return (_b = (_a = this.task) === null || _a === void 0 ? void 0 : _a[symbols_1.kQueueOptions]) !== null && _b !== void 0 ? _b : null;
|
|
}
|
|
get interface() {
|
|
return {
|
|
taskId: this.taskId,
|
|
filename: this.filename,
|
|
name: this.name,
|
|
created: this.created,
|
|
isAbortable: this.abortSignal !== null,
|
|
[symbols_1.kQueueOptions]: this[symbols_1.kQueueOptions]
|
|
};
|
|
}
|
|
}
|
|
exports.TaskInfo = TaskInfo;
|
|
//# sourceMappingURL=index.js.map
|