12345678910111213141516171819202122232425262728293031 |
- const cluster = require('cluster');
- const numCPUs = require('os').cpus().length;
- process.env.NODE_HANDLER = 'mysql-raw';
- if (process.env.TFB_TEST_NAME === 'nodejs-mongodb') {
- process.env.NODE_HANDLER = 'mongoose';
- } else if (process.env.TFB_TEST_NAME === 'nodejs-mongodb-raw') {
- process.env.NODE_HANDLER = 'mongodb-raw';
- } else if (process.env.TFB_TEST_NAME === 'nodejs-mysql') {
- process.env.NODE_HANDLER = 'sequelize';
- } else if (process.env.TFB_TEST_NAME === 'nodejs-postgres') {
- process.env.NODE_HANDLER = 'sequelize-postgres';
- }
- if (cluster.isPrimary) {
- console.log(`Primary ${process.pid} is running`);
- // Fork workers.
- for (let i = 0; i < numCPUs; i++) {
- cluster.fork();
- }
- cluster.on('exit', (worker, code, signal) => {
- console.log(`worker ${worker.process.pid} died`);
- process.exit(1);
- });
- } else {
- // Task for forked worker
- require('./create-server');
- }
|