@adpt/core
Version:
AdaptJS core library
82 lines • 3.38 kB
JavaScript
;
/*
* Copyright 2018-2019 Unbounded Systems, LLC
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
Object.defineProperty(exports, "__esModule", { value: true });
const tslib_1 = require("tslib");
const utils_1 = require("@adpt/utils");
const debug_1 = tslib_1.__importDefault(require("debug"));
// tslint:disable-next-line:no-var-requires
const forkRequire = require("@usys/fork-require");
const error_1 = require("../error");
const debug = debug_1.default("adapt:ops:fork");
function forkWithLogger(func, filename, funcName = func.name) {
if (debug.enabled && process.env.ADAPT_OP_FORKED) {
return async (opts, ...args) => {
debug(`${funcName}: child (PID ${process.pid})`);
return func(opts, ...args);
};
}
if (process.env.ADAPT_NO_FORK || process.env.ADAPT_OP_FORKED)
return func;
return async (options, ...args) => {
debug(`${funcName}: parent (PID ${process.pid})`);
const forked = forkRequire(filename, {
env: Object.assign({}, process.env, { ADAPT_OP_FORKED: true }),
stdio: ["inherit", "pipe", "pipe", "ipc"]
});
try {
// tslint:disable-next-line:prefer-const
let { client, logger } = options, opts = tslib_1.__rest(options, ["client", "logger"]);
if (!client) {
client = new utils_1.MessageStreamClient({
outStream: process.stdout,
errStream: process.stderr,
});
}
if (!utils_1.isMessageClient(client)) {
throw new Error(`API must be called with a MessageClient when using child process`);
}
if (!client.fromStream) {
throw new Error(`MessageClient does not support fromStream`);
}
client.fromStream(forked._childProcess.stdout);
if (client.errStream) {
// Ensure output from child goes to parent's output streams (which
// might not actually stream to real stdout/stderr file descriptors).
forked._childProcess.stderr.pipe(client.errStream, { end: false });
}
return await forked[funcName](opts, ...args);
}
finally {
forked._childProcess.kill();
}
};
}
function forkExports(mod, keys) {
const exps = mod.exports;
if (typeof exps !== "object")
throw new error_1.InternalError(`Invalid exports obj`);
if (!Array.isArray(keys))
keys = [keys];
for (const prop of keys) {
const val = exps[prop];
if (typeof val !== "function")
throw new error_1.InternalError(`Not a function property: ${prop}`);
exps[prop] = forkWithLogger(val, mod.filename);
}
}
exports.forkExports = forkExports;
//# sourceMappingURL=fork.js.map