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.
151 lines
6.5 KiB
151 lines
6.5 KiB
2 years ago
|
"use strict";
|
||
|
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
|
||
|
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
|
||
|
return new (P || (P = Promise))(function (resolve, reject) {
|
||
|
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
|
||
|
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
|
||
|
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
|
||
|
step((generator = generator.apply(thisArg, _arguments || [])).next());
|
||
|
});
|
||
|
};
|
||
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
const child_process_1 = require("child_process");
|
||
|
const RpcIpcMessagePortClosedError_1 = require("./error/RpcIpcMessagePortClosedError");
|
||
|
function createRpcIpcMessagePort(process) {
|
||
|
const messageListeners = new Set();
|
||
|
const errorListeners = new Set();
|
||
|
let closedError;
|
||
|
const handleExit = (code, signal) => __awaiter(this, void 0, void 0, function* () {
|
||
|
closedError = new RpcIpcMessagePortClosedError_1.RpcIpcMessagePortClosedError(code
|
||
|
? `Process ${process.pid} exited with code "${code}" [${signal}]`
|
||
|
: `Process ${process.pid} exited [${signal}].`, code, signal);
|
||
|
errorListeners.forEach((listener) => {
|
||
|
if (closedError) {
|
||
|
listener(closedError);
|
||
|
}
|
||
|
});
|
||
|
yield port.close();
|
||
|
});
|
||
|
const handleMessage = (message) => {
|
||
|
messageListeners.forEach((listener) => {
|
||
|
listener(message);
|
||
|
});
|
||
|
};
|
||
|
process.on('message', handleMessage);
|
||
|
process.on('exit', handleExit);
|
||
|
const port = {
|
||
|
dispatchMessage: (message) => __awaiter(this, void 0, void 0, function* () {
|
||
|
return new Promise((resolve, reject) => {
|
||
|
if (!process.connected) {
|
||
|
reject(closedError ||
|
||
|
new RpcIpcMessagePortClosedError_1.RpcIpcMessagePortClosedError(`Process ${process.pid} doesn't have open IPC channels`));
|
||
|
}
|
||
|
if (process.send) {
|
||
|
process.send(Object.assign(Object.assign({}, message), { source: process.pid }), undefined, undefined, (sendError) => {
|
||
|
if (sendError) {
|
||
|
if (!closedError) {
|
||
|
closedError = new RpcIpcMessagePortClosedError_1.RpcIpcMessagePortClosedError(`Cannot send the message - the message port has been closed for the process ${process.pid}.`);
|
||
|
}
|
||
|
reject(closedError);
|
||
|
}
|
||
|
else {
|
||
|
resolve();
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
else {
|
||
|
reject(new RpcIpcMessagePortClosedError_1.RpcIpcMessagePortClosedError(`Process ${process.pid} doesn't have IPC channels`));
|
||
|
}
|
||
|
});
|
||
|
}),
|
||
|
addMessageListener: (listener) => {
|
||
|
messageListeners.add(listener);
|
||
|
},
|
||
|
removeMessageListener: (listener) => {
|
||
|
messageListeners.delete(listener);
|
||
|
},
|
||
|
addErrorListener: (listener) => {
|
||
|
errorListeners.add(listener);
|
||
|
},
|
||
|
removeErrorListener: (listener) => {
|
||
|
errorListeners.delete(listener);
|
||
|
},
|
||
|
isOpen: () => !!process.connected,
|
||
|
open: () => __awaiter(this, void 0, void 0, function* () {
|
||
|
if (!process.connected || closedError) {
|
||
|
throw (closedError ||
|
||
|
new RpcIpcMessagePortClosedError_1.RpcIpcMessagePortClosedError(`Cannot open closed IPC channel for process ${process.pid}.`));
|
||
|
}
|
||
|
}),
|
||
|
close: () => __awaiter(this, void 0, void 0, function* () {
|
||
|
process.off('message', handleMessage);
|
||
|
process.off('exit', handleExit);
|
||
|
messageListeners.clear();
|
||
|
errorListeners.clear();
|
||
|
if (process.disconnect && process.connected) {
|
||
|
process.disconnect();
|
||
|
}
|
||
|
}),
|
||
|
};
|
||
|
return port;
|
||
|
}
|
||
|
exports.createRpcIpcMessagePort = createRpcIpcMessagePort;
|
||
|
function createRpcIpcForkedProcessMessagePort(filePath, memoryLimit = 2048, autoRecreate = true) {
|
||
|
function createChildProcess() {
|
||
|
return child_process_1.fork(filePath, [], {
|
||
|
execArgv: [`--max-old-space-size=${memoryLimit}`],
|
||
|
stdio: ['inherit', 'inherit', 'inherit', 'ipc'],
|
||
|
});
|
||
|
}
|
||
|
const messageListeners = new Set();
|
||
|
const errorListeners = new Set();
|
||
|
let childProcess = createChildProcess();
|
||
|
let port = createRpcIpcMessagePort(childProcess);
|
||
|
return {
|
||
|
dispatchMessage: (message) => port.dispatchMessage(message),
|
||
|
addMessageListener: (listener) => {
|
||
|
messageListeners.add(listener);
|
||
|
return port.addMessageListener(listener);
|
||
|
},
|
||
|
removeMessageListener: (listener) => {
|
||
|
messageListeners.delete(listener);
|
||
|
return port.removeMessageListener(listener);
|
||
|
},
|
||
|
addErrorListener: (listener) => {
|
||
|
errorListeners.add(listener);
|
||
|
return port.addErrorListener(listener);
|
||
|
},
|
||
|
removeErrorListener: (listener) => {
|
||
|
errorListeners.delete(listener);
|
||
|
return port.removeErrorListener(listener);
|
||
|
},
|
||
|
isOpen: () => port.isOpen(),
|
||
|
open: () => __awaiter(this, void 0, void 0, function* () {
|
||
|
if (!port.isOpen() && autoRecreate) {
|
||
|
// recreate the process and add existing message listeners
|
||
|
childProcess = createChildProcess();
|
||
|
port = createRpcIpcMessagePort(childProcess);
|
||
|
messageListeners.forEach((listener) => {
|
||
|
port.addMessageListener(listener);
|
||
|
});
|
||
|
errorListeners.forEach((listener) => {
|
||
|
port.addErrorListener(listener);
|
||
|
});
|
||
|
}
|
||
|
else {
|
||
|
return port.open();
|
||
|
}
|
||
|
}),
|
||
|
close: () => __awaiter(this, void 0, void 0, function* () {
|
||
|
yield port.close();
|
||
|
messageListeners.clear();
|
||
|
errorListeners.clear();
|
||
|
if (childProcess) {
|
||
|
childProcess.kill('SIGTERM');
|
||
|
childProcess = undefined;
|
||
|
}
|
||
|
}),
|
||
|
};
|
||
|
}
|
||
|
exports.createRpcIpcForkedProcessMessagePort = createRpcIpcForkedProcessMessagePort;
|