Files
scrypted/server/src/plugin/runtime/node-fork-worker.ts
Koushik Dutta a8b8aa03e0 restructure
2022-02-11 15:06:21 -08:00

55 lines
1.7 KiB
TypeScript

import { RuntimeWorkerOptions as RuntimeWorkerOptions } from "./runtime-worker";
import child_process from 'child_process';
import path from 'path';
import { RpcMessage, RpcPeer } from "../../rpc";
import { ChildProcessWorker } from "./child-process-worker";
import { getPluginNodePath } from "../plugin-npm-dependencies";
export class NodeForkWorker extends ChildProcessWorker {
constructor(pluginId: string, options: RuntimeWorkerOptions) {
super(pluginId, options);
const {env, pluginDebug} = options;
const execArgv: string[] = process.execArgv.slice();
if (pluginDebug) {
execArgv.push(`--inspect=0.0.0.0:${pluginDebug.inspectPort}`);
}
this.worker = child_process.fork(require.main.filename, ['child', this.pluginId], {
stdio: ['pipe', 'pipe', 'pipe', 'ipc'],
env: Object.assign({}, process.env, env, {
NODE_PATH: path.join(getPluginNodePath(this.pluginId), 'node_modules'),
}),
serialization: 'advanced',
execArgv,
});
this.setupWorker();
}
setupRpcPeer(peer: RpcPeer): void {
this.worker.on('message', message => peer.handleMessage(message as any));
peer.transportSafeArgumentTypes.add(Buffer.name);
}
send(message: RpcMessage, reject?: (e: Error) => void): void {
try {
if (!this.worker)
throw new Error('worked has been killed');
this.worker.send(message, undefined, e => {
if (e && reject)
reject(e);
});
}
catch (e) {
reject?.(e);
}
}
get pid() {
return this.worker.pid;
}
}