2016-12-28 16:49:51 -06:00
|
|
|
/**
|
2017-01-02 15:09:17 -06:00
|
|
|
* Misskey Entry Point!
|
2016-12-28 16:49:51 -06:00
|
|
|
*/
|
|
|
|
|
|
|
|
Error.stackTraceLimit = Infinity;
|
|
|
|
|
|
|
|
import * as os from 'os';
|
|
|
|
import * as cluster from 'cluster';
|
2017-01-23 19:28:14 -06:00
|
|
|
import * as debug from 'debug';
|
2017-11-06 04:59:14 -06:00
|
|
|
import chalk from 'chalk';
|
2018-07-13 09:25:32 -05:00
|
|
|
import * as portscanner from 'portscanner';
|
2017-01-02 14:49:37 -06:00
|
|
|
import isRoot = require('is-root');
|
2017-06-08 11:03:54 -05:00
|
|
|
import Xev from 'xev';
|
2017-04-04 19:58:29 -05:00
|
|
|
|
2018-07-07 05:19:00 -05: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-10 16:48:25 -05:00
|
|
|
import serverStats from './daemons/server-stats';
|
|
|
|
import notesStats from './daemons/notes-stats';
|
2018-04-01 23:15:53 -05:00
|
|
|
import loadConfig from './config/load';
|
|
|
|
import { Config } from './config/types';
|
2017-01-16 17:19:34 -06:00
|
|
|
|
2017-01-23 19:28:14 -06:00
|
|
|
const clusterLog = debug('misskey:cluster');
|
2017-06-08 11:03:54 -05:00
|
|
|
const ev = new Xev();
|
2017-01-23 19:28:14 -06:00
|
|
|
|
2016-12-28 16:49:51 -06:00
|
|
|
process.title = 'Misskey';
|
|
|
|
|
2018-04-05 04:08:51 -05:00
|
|
|
if (process.env.NODE_ENV != 'production') {
|
|
|
|
process.env.DEBUG = 'misskey:*';
|
|
|
|
}
|
|
|
|
|
2018-04-02 04:59:48 -05:00
|
|
|
// https://github.com/Automattic/kue/issues/822
|
2018-04-06 16:25:58 -05:00
|
|
|
require('events').EventEmitter.prototype._maxListeners = 512;
|
2018-04-02 04:59:48 -05:00
|
|
|
|
2016-12-28 16:49:51 -06:00
|
|
|
// Start app
|
|
|
|
main();
|
|
|
|
|
|
|
|
/**
|
2017-02-27 01:11:49 -06:00
|
|
|
* Init process
|
2016-12-28 16:49:51 -06:00
|
|
|
*/
|
2017-01-23 19:28:14 -06:00
|
|
|
function main() {
|
2016-12-28 16:49:51 -06:00
|
|
|
if (cluster.isMaster) {
|
2018-07-07 05:19:00 -05:00
|
|
|
masterMain();
|
2017-06-08 11:03:54 -05:00
|
|
|
|
|
|
|
ev.mount();
|
2018-06-08 11:45:25 -05:00
|
|
|
serverStats();
|
2018-06-08 14:14:26 -05:00
|
|
|
notesStats();
|
2017-01-02 14:15:50 -06:00
|
|
|
} else {
|
2018-07-07 05:19:00 -05:00
|
|
|
workerMain();
|
2016-12-28 16:49:51 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2017-02-27 01:11:49 -06:00
|
|
|
* Init master process
|
2016-12-28 16:49:51 -06:00
|
|
|
*/
|
2018-07-07 05:19:00 -05:00
|
|
|
async function masterMain() {
|
2017-04-04 19:58:29 -05:00
|
|
|
let config: Config;
|
2016-12-28 16:49:51 -06:00
|
|
|
|
|
|
|
try {
|
|
|
|
// initialize app
|
2017-04-04 19:58:29 -05:00
|
|
|
config = await init();
|
2016-12-28 16:49:51 -06:00
|
|
|
} catch (e) {
|
|
|
|
console.error(e);
|
2018-07-14 06:58:21 -05:00
|
|
|
Logger.error('Fatal error occurred during initialization');
|
2017-04-23 01:40:13 -05:00
|
|
|
process.exit(1);
|
2017-04-04 19:58:29 -05:00
|
|
|
}
|
|
|
|
|
2018-07-14 06:58:21 -05:00
|
|
|
Logger.succ('Successfully initialized');
|
2017-04-04 19:58:29 -05:00
|
|
|
|
2017-01-02 14:18:03 -06:00
|
|
|
spawnWorkers(() => {
|
2018-07-14 06:58:21 -05:00
|
|
|
Logger.info(chalk.bold.green(`Now listening on port ${chalk.underline(config.port.toString())}`));
|
2018-07-07 05:19:00 -05:00
|
|
|
Logger.info(chalk.bold.green(config.url));
|
|
|
|
Logger.info(chalk.bold.green('Now processing jobs'));
|
2016-12-28 16:49:51 -06:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2017-02-27 01:11:49 -06:00
|
|
|
* Init worker process
|
2016-12-28 16:49:51 -06:00
|
|
|
*/
|
2018-07-07 05:19:00 -05:00
|
|
|
async function workerMain() {
|
|
|
|
// start server
|
|
|
|
await require('./server').default();
|
2018-03-28 11:20:40 -05:00
|
|
|
|
2018-07-07 05:19:00 -05:00
|
|
|
// start processor
|
|
|
|
require('./queue').default();
|
2018-03-28 11:20:40 -05:00
|
|
|
|
|
|
|
// Send a 'ready' message to parent process
|
|
|
|
process.send('ready');
|
2016-12-28 16:49:51 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Init app
|
|
|
|
*/
|
2017-04-04 19:58:29 -05:00
|
|
|
async function init(): Promise<Config> {
|
2016-12-29 08:09:21 -06:00
|
|
|
Logger.info('Welcome to Misskey!');
|
2016-12-28 16:49:51 -06:00
|
|
|
|
2016-12-30 12:35:19 -06:00
|
|
|
EnvironmentInfo.show();
|
2016-12-30 12:19:59 -06:00
|
|
|
MachineInfo.show();
|
2016-12-30 12:24:07 -06:00
|
|
|
new DependencyInfo().showAll();
|
2016-12-30 12:14:38 -06:00
|
|
|
|
2017-05-24 06:50:17 -05:00
|
|
|
const configLogger = new Logger('Config');
|
2018-04-01 23:15:53 -05:00
|
|
|
let config;
|
2016-12-28 16:49:51 -06:00
|
|
|
|
2018-04-01 23:15:53 -05:00
|
|
|
try {
|
|
|
|
config = loadConfig();
|
|
|
|
} catch (exception) {
|
2018-07-14 06:58:21 -05:00
|
|
|
if (typeof exception === 'string') {
|
|
|
|
configLogger.error(exception);
|
|
|
|
process.exit(1);
|
|
|
|
}
|
2018-04-01 23:15:53 -05:00
|
|
|
if (exception.code === 'ENOENT') {
|
2018-07-14 06:58:21 -05:00
|
|
|
configLogger.error('Configuration file not found');
|
|
|
|
process.exit(1);
|
2018-04-01 23:15:53 -05:00
|
|
|
}
|
|
|
|
throw exception;
|
|
|
|
}
|
2017-01-16 17:19:34 -06:00
|
|
|
|
2018-07-14 06:58:21 -05:00
|
|
|
configLogger.succ('Successfully loaded');
|
2018-07-13 09:05:37 -05:00
|
|
|
configLogger.info(`Maintainer: ${config.maintainer.name}`);
|
2016-12-28 16:49:51 -06:00
|
|
|
|
2016-12-29 10:35:25 -06:00
|
|
|
if (process.platform === 'linux' && !isRoot() && config.port < 1024) {
|
2018-07-13 09:52:28 -05:00
|
|
|
Logger.error('You need root privileges to listen on port below 1024 on Linux');
|
|
|
|
process.exit(1);
|
2016-12-29 10:35:25 -06:00
|
|
|
}
|
|
|
|
|
2018-07-13 09:25:32 -05:00
|
|
|
if (await portscanner.checkPortStatus(config.port, '127.0.0.1') === 'open') {
|
2018-07-13 10:05:49 -05:00
|
|
|
Logger.error(`Port ${config.port} is already in use`);
|
2018-07-13 09:52:28 -05:00
|
|
|
process.exit(1);
|
2016-12-28 16:49:51 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
// Try to connect to MongoDB
|
2017-05-24 06:50:17 -05:00
|
|
|
const mongoDBLogger = new Logger('MongoDB');
|
2017-04-23 02:40:49 -05:00
|
|
|
const db = require('./db/mongodb').default;
|
2018-07-14 06:58:21 -05:00
|
|
|
mongoDBLogger.succ('Successfully connected');
|
2017-04-23 02:40:49 -05:00
|
|
|
db.close();
|
2016-12-28 16:49:51 -06:00
|
|
|
|
2017-04-04 19:58:29 -05:00
|
|
|
return config;
|
2016-12-28 16:49:51 -06:00
|
|
|
}
|
|
|
|
|
2017-04-23 01:55:37 -05:00
|
|
|
function spawnWorkers(onComplete: Function) {
|
2016-12-28 16:49:51 -06: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 08:06:47 -05:00
|
|
|
progress.on('complete', () => {
|
|
|
|
onComplete();
|
|
|
|
});
|
2016-12-28 16:49:51 -06:00
|
|
|
}
|
|
|
|
|
2017-01-23 19:28:14 -06: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();
|
|
|
|
});
|
|
|
|
|
2017-01-18 17:04:17 -06:00
|
|
|
// Display detail of unhandled promise rejection
|
|
|
|
process.on('unhandledRejection', console.dir);
|
|
|
|
|
2018-05-20 05:24:54 -05:00
|
|
|
// Display detail of uncaught exception
|
|
|
|
process.on('uncaughtException', err => {
|
|
|
|
console.error(err);
|
|
|
|
});
|
|
|
|
|
2016-12-28 16:49:51 -06:00
|
|
|
// Dying away...
|
2018-05-20 05:24:54 -05:00
|
|
|
process.on('exit', code => {
|
2018-07-13 10:41:33 -05:00
|
|
|
Logger.info(`The process is going to exit with code ${code}`);
|
2016-12-28 16:49:51 -06:00
|
|
|
});
|