mirror of
https://git.boykissers.com/pawkey/pawkey-sk.git
synced 2025-12-20 04:04:16 +00:00
125 lines
3.6 KiB
TypeScript
125 lines
3.6 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
/**
|
|
* Misskey Entry Point!
|
|
*/
|
|
|
|
import cluster from 'node:cluster';
|
|
import { EventEmitter } from 'node:events';
|
|
import { inspect } from 'node:util';
|
|
import chalk from 'chalk';
|
|
import Xev from 'xev';
|
|
import Logger from '@/logger.js';
|
|
import { envOption } from '../env.js';
|
|
import { masterMain } from './master.js';
|
|
import { workerMain } from './worker.js';
|
|
import { readyRef } from './ready.js';
|
|
|
|
import 'reflect-metadata';
|
|
|
|
process.title = `Misskey (${cluster.isPrimary ? 'master' : 'worker'})`;
|
|
|
|
Error.stackTraceLimit = Infinity;
|
|
EventEmitter.defaultMaxListeners = 128;
|
|
|
|
const logger = new Logger('core', 'cyan');
|
|
const clusterLogger = logger.createSubLogger('cluster', 'orange');
|
|
const ev = new Xev();
|
|
|
|
// We wrap this in a main function, that gets called,
|
|
// because not all platforms support top level await :/
|
|
|
|
async function main() {
|
|
//#region Events
|
|
// Listen new workers
|
|
cluster.on('fork', worker => {
|
|
clusterLogger.debug(`Process forked: [${worker.id}]`);
|
|
});
|
|
|
|
// Listen online workers
|
|
cluster.on('online', worker => {
|
|
clusterLogger.debug(`Process is now online: [${worker.id}]`);
|
|
});
|
|
|
|
// Listen for dying workers
|
|
cluster.on('exit', worker => {
|
|
// Replace the dead worker,
|
|
// we're not sentimental
|
|
clusterLogger.error(chalk.red(`[${worker.id}] died :(`));
|
|
cluster.fork();
|
|
});
|
|
|
|
// Display detail of unhandled promise rejection
|
|
if (!envOption.quiet) {
|
|
process.on('unhandledRejection', e => {
|
|
logger.error('Unhandled rejection:', inspect(e));
|
|
});
|
|
}
|
|
|
|
process.on('uncaughtException', (err) => {
|
|
// Workaround for https://github.com/node-fetch/node-fetch/issues/954
|
|
if (String(err).match(/^TypeError: .+ is an? url with embedded credentials.$/)) {
|
|
logger.debug('Suppressed node-fetch issue#954, but the current job may fail.');
|
|
return;
|
|
}
|
|
|
|
// Workaround for https://github.com/node-fetch/node-fetch/issues/1845
|
|
if (String(err) === 'TypeError: Cannot read properties of undefined (reading \'body\')') {
|
|
logger.debug('Suppressed node-fetch issue#1845, but the current job may fail.');
|
|
return;
|
|
}
|
|
|
|
// Throw all other errors to avoid inconsistent state.
|
|
// (per NodeJS docs, it's unsafe to suppress arbitrary errors in an uncaughtException handler.)
|
|
throw err;
|
|
});
|
|
|
|
// Display detail of uncaught exception
|
|
process.on('uncaughtExceptionMonitor', (err, origin) => {
|
|
logger.error(`Uncaught exception (${origin}):`, err);
|
|
});
|
|
|
|
// Dying away...
|
|
process.on('disconnect', () => {
|
|
logger.warn('IPC channel disconnected! The process may soon die.');
|
|
});
|
|
process.on('beforeExit', code => {
|
|
logger.warn(`Event loop died! Process will exit with code ${code}.`);
|
|
});
|
|
process.on('exit', code => {
|
|
logger.info(`The process is going to exit with code ${code}`);
|
|
});
|
|
//#endregion
|
|
|
|
if (!envOption.disableClustering) {
|
|
if (cluster.isPrimary) {
|
|
logger.info(`Start main process... pid: ${process.pid}`);
|
|
await masterMain();
|
|
ev.mount();
|
|
} else if (cluster.isWorker) {
|
|
logger.info(`Start worker process... pid: ${process.pid}`);
|
|
await workerMain();
|
|
} else {
|
|
throw new Error('Unknown process type');
|
|
}
|
|
} else {
|
|
// 非clusterの場合はMasterのみが起動するため、Workerの処理は行わない(cluster.isWorker === trueの状態でこのブロックに来ることはない)
|
|
logger.info(`Start main process... pid: ${process.pid}`);
|
|
await masterMain();
|
|
ev.mount();
|
|
}
|
|
|
|
readyRef.value = true;
|
|
|
|
// ユニットテスト時にMisskeyが子プロセスで起動された時のため
|
|
// それ以外のときは process.send は使えないので弾く
|
|
if (process.send) {
|
|
process.send('ok');
|
|
}
|
|
}
|
|
|
|
main();
|