/* eslint-disable import/no-dynamic-require */ /* eslint-disable no-await-in-loop */ /* eslint-disable no-eval */ const cluster = require('cluster'); const numCPUs = require('os').cpus().length; async function terminate() { for (const task of global.onDestory) { // eslint-disable-next-line no-await-in-loop await task(); } process.exit(0); } async function entry(config) { if (config.entry) { // TODO newProcess const loader = require(`./entry/${config.entry}`); await loader(entry); } } async function stopWorker() { cluster.disconnect(); } async function startWorker() { for (let i = 0; i < numCPUs; i++) { cluster.fork(); } } async function messageHandler(worker, msg) { if (msg.event && msg.event === 'restart') { console.log('Restarting'); await stopWorker(); console.log('Worker stopped'); await startWorker(); } } async function load() { global.nodeModules = { bson: require('bson'), 'js-yaml': require('js-yaml'), mongodb: require('mongodb'), }; global.Hydro = { config: {}, handler: {}, service: {}, model: {}, script: {}, lib: {}, wiki: {}, template: {}, ui: {}, }; global.onDestory = []; Error.stackTraceLimit = 50; process.on('unhandledRejection', (e) => console.error(e)); process.on('SIGINT', terminate); if (cluster.isMaster) { console.log(`Master ${process.pid} Starting`); process.stdin.setEncoding('utf8'); process.stdin.on('data', async (input) => { try { const t = eval(input.toString().trim()); if (t instanceof Promise) console.log(await t); else console.log(t); } catch (e) { console.warn(e); } }); await entry({ entry: 'master' }); cluster.on('exit', (worker, code, signal) => { console.log(`Worker ${worker.process.pid} exit: ${code} ${signal}`); }); cluster.on('disconnect', (worker) => { console.log(`Worker ${worker.process.pid} disconnected`); }); cluster.on('listening', (worker, address) => { console.log(`Worker ${worker.process.pid} listening at `, address); }); cluster.on('online', (worker) => { console.log(`Worker ${worker.process.pid} is online`); }); cluster.on('message', messageHandler); await startWorker(); } else { console.log(`Worker ${process.pid} Starting`); await entry({ entry: 'worker' }); console.log(`Worker ${process.pid} Started`); } if (global.gc) global.gc(); } if (!module.parent) { load().catch((e) => { console.error(e); process.exit(1); }); }