36 lines
746 B
JavaScript
36 lines
746 B
JavaScript
const cp = require('child_process');
|
|
const path = require('path');
|
|
|
|
const TYPE = process.env.TYPE;
|
|
const MODE = process.env.MODE;
|
|
|
|
if (!TYPE) {
|
|
console.error(`
|
|
Usage: TYPE={type} node start.js
|
|
TYPE=node node start.js
|
|
`.trim());
|
|
|
|
process.exit(1);
|
|
}
|
|
|
|
const handler = ({
|
|
node: () => {
|
|
return cp.exec('node src/index.js', {
|
|
cwd: __dirname
|
|
})
|
|
},
|
|
artillery: () => {
|
|
const conf = path.join(__dirname, '../artillery/artillery-${MODE}.yml');
|
|
return cp.exec(`../node_modules/.bin/artillery run ${conf}`)
|
|
}
|
|
})[TYPE];
|
|
|
|
if (!handler) {
|
|
console.error(`No handler for ${TYPE}`);
|
|
process.exit(1);
|
|
}
|
|
|
|
handler().stdout.pipe(process.stdout);
|
|
handler().stderr.pipe(process.stderr);
|
|
process.stdin.pipe(handler().stdin);
|