1
0
mirror of https://github.com/hotomoe/hotomoe synced 2024-12-05 10:18:09 +09:00
hotomoe/src/index.ts

197 lines
4.2 KiB
TypeScript
Raw Normal View History

2016-12-29 07:49:51 +09:00
/**
2017-01-03 06:09:17 +09:00
* Misskey Entry Point!
2016-12-29 07:49:51 +09:00
*/
Error.stackTraceLimit = Infinity;
import * as os from 'os';
import * as cluster from 'cluster';
2017-01-24 10:28:14 +09:00
import * as debug from 'debug';
2017-11-06 19:59:14 +09:00
import chalk from 'chalk';
2018-07-13 23:25:32 +09:00
import * as portscanner from 'portscanner';
2017-01-03 05:49:37 +09:00
import isRoot = require('is-root');
2017-06-09 01:03:54 +09:00
import Xev from 'xev';
2017-04-05 09:58:29 +09:00
2018-07-07 19:19:00 +09:00
import Logger from './misc/logger';
import ProgressBar from './misc/cli/progressbar';
import EnvironmentInfo from './misc/environmentInfo';
import MachineInfo from './misc/machineInfo';
import DependencyInfo from './misc/dependencyInfo';
2018-06-11 06:48:25 +09:00
import serverStats from './daemons/server-stats';
import notesStats from './daemons/notes-stats';
2018-07-04 13:21:30 +09:00
import db from './db/mongodb';
2018-04-02 13:15:53 +09:00
import loadConfig from './config/load';
import { Config } from './config/types';
2017-01-17 08:19:34 +09:00
2017-01-24 10:28:14 +09:00
const clusterLog = debug('misskey:cluster');
2017-06-09 01:03:54 +09:00
const ev = new Xev();
2017-01-24 10:28:14 +09:00
2016-12-29 07:49:51 +09:00
process.title = 'Misskey';
2018-04-05 18:08:51 +09:00
if (process.env.NODE_ENV != 'production') {
process.env.DEBUG = 'misskey:*';
}
2018-04-02 18:59:48 +09:00
// https://github.com/Automattic/kue/issues/822
2018-04-07 06:25:58 +09:00
require('events').EventEmitter.prototype._maxListeners = 512;
2018-04-02 18:59:48 +09:00
2016-12-29 07:49:51 +09:00
// Start app
main();
/**
2017-02-27 16:11:49 +09:00
* Init process
2016-12-29 07:49:51 +09:00
*/
2017-01-24 10:28:14 +09:00
function main() {
2016-12-29 07:49:51 +09:00
if (cluster.isMaster) {
2018-07-07 19:19:00 +09:00
masterMain();
2017-06-09 01:03:54 +09:00
ev.mount();
serverStats();
2018-06-09 04:14:26 +09:00
notesStats();
2017-01-03 05:15:50 +09:00
} else {
2018-07-07 19:19:00 +09:00
workerMain();
2016-12-29 07:49:51 +09:00
}
}
/**
2017-02-27 16:11:49 +09:00
* Init master process
2016-12-29 07:49:51 +09:00
*/
2018-07-07 19:19:00 +09:00
async function masterMain() {
2017-04-05 09:58:29 +09:00
let config: Config;
2016-12-29 07:49:51 +09:00
try {
// initialize app
2017-04-05 09:58:29 +09:00
config = await init();
2016-12-29 07:49:51 +09:00
} catch (e) {
console.error(e);
2017-04-05 09:58:29 +09:00
Logger.error(chalk.red('Fatal error occurred during initializing :('));
2017-04-23 15:40:13 +09:00
process.exit(1);
2017-04-05 09:58:29 +09:00
}
Logger.info(chalk.green('Successfully initialized :)'));
2017-01-03 05:18:03 +09:00
spawnWorkers(() => {
2018-07-07 19:19:00 +09:00
Logger.info(chalk.bold.green(
`Now listening on port ${chalk.underline(config.port.toString())}`));
2017-08-11 22:14:55 +09:00
2018-07-07 19:19:00 +09:00
Logger.info(chalk.bold.green(config.url));
2018-03-29 01:20:40 +09:00
2018-07-07 19:19:00 +09:00
Logger.info(chalk.bold.green('Now processing jobs'));
2016-12-29 07:49:51 +09:00
});
}
/**
2017-02-27 16:11:49 +09:00
* Init worker process
2016-12-29 07:49:51 +09:00
*/
2018-07-07 19:19:00 +09:00
async function workerMain() {
// start server
await require('./server').default();
2018-03-29 01:20:40 +09:00
2018-07-07 19:19:00 +09:00
// start processor
require('./queue').default();
2018-03-29 01:20:40 +09:00
// Send a 'ready' message to parent process
process.send('ready');
2016-12-29 07:49:51 +09:00
}
/**
* Init app
*/
2017-04-05 09:58:29 +09:00
async function init(): Promise<Config> {
2016-12-29 23:09:21 +09:00
Logger.info('Welcome to Misskey!');
Logger.info('Initializing...');
2016-12-29 07:49:51 +09:00
2016-12-31 03:35:19 +09:00
EnvironmentInfo.show();
2016-12-31 03:19:59 +09:00
MachineInfo.show();
new DependencyInfo().showAll();
2016-12-31 03:14:38 +09:00
2017-05-24 20:50:17 +09:00
const configLogger = new Logger('Config');
2018-04-02 13:15:53 +09:00
let config;
2016-12-29 07:49:51 +09:00
2018-04-02 13:15:53 +09:00
try {
config = loadConfig();
} catch (exception) {
if (exception.code === 'ENOENT') {
throw 'Configuration not found - Please run "npm run config" command.';
}
throw exception;
}
2017-01-17 08:19:34 +09:00
2016-12-29 23:09:21 +09:00
configLogger.info('Successfully loaded');
2018-07-13 23:05:37 +09:00
configLogger.info(`Maintainer: ${config.maintainer.name}`);
2016-12-29 07:49:51 +09:00
2016-12-30 01:35:25 +09:00
if (process.platform === 'linux' && !isRoot() && config.port < 1024) {
2018-07-13 23:52:28 +09:00
Logger.error('You need root privileges to listen on port below 1024 on Linux');
process.exit(1);
2016-12-30 01:35:25 +09:00
}
2018-07-13 23:25:32 +09:00
if (await portscanner.checkPortStatus(config.port, '127.0.0.1') === 'open') {
2018-07-14 00:05:49 +09:00
Logger.error(`Port ${config.port} is already in use`);
2018-07-13 23:52:28 +09:00
process.exit(1);
2016-12-29 07:49:51 +09:00
}
// Try to connect to MongoDB
2017-05-24 20:50:17 +09:00
const mongoDBLogger = new Logger('MongoDB');
2017-04-23 16:40:49 +09:00
const db = require('./db/mongodb').default;
mongoDBLogger.info('Successfully connected');
db.close();
2016-12-29 07:49:51 +09:00
2017-04-05 09:58:29 +09:00
return config;
2016-12-29 07:49:51 +09:00
}
2017-04-23 15:55:37 +09:00
function spawnWorkers(onComplete: Function) {
2016-12-29 07:49:51 +09:00
// Count the machine's CPUs
const cpuCount = os.cpus().length;
const progress = new ProgressBar(cpuCount, 'Starting workers');
// Create a worker for each CPU
for (let i = 0; i < cpuCount; i++) {
const worker = cluster.fork();
worker.on('message', message => {
if (message === 'ready') {
progress.increment();
}
});
}
// On all workers started
2017-08-10 22:06:47 +09:00
progress.on('complete', () => {
onComplete();
});
2016-12-29 07:49:51 +09:00
}
2017-01-24 10:28:14 +09:00
// Listen new workers
cluster.on('fork', worker => {
clusterLog(`Process forked: [${worker.id}]`);
});
// Listen online workers
cluster.on('online', worker => {
clusterLog(`Process is now online: [${worker.id}]`);
});
// Listen for dying workers
cluster.on('exit', worker => {
// Replace the dead worker,
// we're not sentimental
clusterLog(chalk.red(`[${worker.id}] died :(`));
cluster.fork();
});
// Display detail of unhandled promise rejection
process.on('unhandledRejection', console.dir);
2018-05-20 19:24:54 +09:00
// Display detail of uncaught exception
process.on('uncaughtException', err => {
console.error(err);
});
2016-12-29 07:49:51 +09:00
// Dying away...
2018-05-20 19:24:54 +09:00
process.on('exit', code => {
2018-07-14 00:39:11 +09:00
Logger.info(`The process is going exit with code ${code}`);
2018-07-04 13:21:30 +09:00
db.close();
2016-12-29 07:49:51 +09:00
});