83 lines
2.8 KiB
JavaScript
83 lines
2.8 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.ChildPool = void 0;
|
|
const path = require("path");
|
|
const child_1 = require("./child");
|
|
const CHILD_KILL_TIMEOUT = 30000;
|
|
const supportCJS = () => {
|
|
return (typeof require === 'function' &&
|
|
typeof module === 'object' &&
|
|
typeof module.exports === 'object');
|
|
};
|
|
class ChildPool {
|
|
constructor({ mainFile = supportCJS()
|
|
? path.join(process.cwd(), 'dist/cjs/classes/main.js')
|
|
: path.join(process.cwd(), 'dist/esm/classes/main.js'), useWorkerThreads, workerForkOptions, workerThreadsOptions, }) {
|
|
this.retained = {};
|
|
this.free = {};
|
|
this.opts = {
|
|
mainFile,
|
|
useWorkerThreads,
|
|
workerForkOptions,
|
|
workerThreadsOptions,
|
|
};
|
|
}
|
|
async retain(processFile) {
|
|
let child = this.getFree(processFile).pop();
|
|
if (child) {
|
|
this.retained[child.pid] = child;
|
|
return child;
|
|
}
|
|
child = new child_1.Child(this.opts.mainFile, processFile, {
|
|
useWorkerThreads: this.opts.useWorkerThreads,
|
|
workerForkOptions: this.opts.workerForkOptions,
|
|
workerThreadsOptions: this.opts.workerThreadsOptions,
|
|
});
|
|
child.on('exit', this.remove.bind(this, child));
|
|
try {
|
|
await child.init();
|
|
// Check status here as well, in case the child exited before we could
|
|
// retain it.
|
|
if (child.exitCode !== null || child.signalCode !== null) {
|
|
throw new Error('Child exited before it could be retained');
|
|
}
|
|
this.retained[child.pid] = child;
|
|
return child;
|
|
}
|
|
catch (err) {
|
|
console.error(err);
|
|
this.release(child);
|
|
throw err;
|
|
}
|
|
}
|
|
release(child) {
|
|
delete this.retained[child.pid];
|
|
this.getFree(child.processFile).push(child);
|
|
}
|
|
remove(child) {
|
|
delete this.retained[child.pid];
|
|
const free = this.getFree(child.processFile);
|
|
const childIndex = free.indexOf(child);
|
|
if (childIndex > -1) {
|
|
free.splice(childIndex, 1);
|
|
}
|
|
}
|
|
async kill(child, signal = 'SIGKILL') {
|
|
this.remove(child);
|
|
return child.kill(signal, CHILD_KILL_TIMEOUT);
|
|
}
|
|
async clean() {
|
|
const children = Object.values(this.retained).concat(this.getAllFree());
|
|
this.retained = {};
|
|
this.free = {};
|
|
await Promise.all(children.map(c => this.kill(c, 'SIGTERM')));
|
|
}
|
|
getFree(id) {
|
|
return (this.free[id] = this.free[id] || []);
|
|
}
|
|
getAllFree() {
|
|
return Object.values(this.free).reduce((first, second) => first.concat(second), []);
|
|
}
|
|
}
|
|
exports.ChildPool = ChildPool;
|
|
//# sourceMappingURL=child-pool.js.map
|