2016-12-28 14:49:51 -08:00
|
|
|
/**
|
2017-01-02 13:09:17 -08:00
|
|
|
* Misskey Entry Point!
|
2016-12-28 14:49:51 -08:00
|
|
|
*/
|
|
|
|
|
|
|
|
Error.stackTraceLimit = Infinity;
|
|
|
|
|
2018-08-06 05:35:49 -07:00
|
|
|
require('events').EventEmitter.defaultMaxListeners = 128;
|
|
|
|
|
2016-12-28 14:49:51 -08:00
|
|
|
import * as os from 'os';
|
|
|
|
import * as cluster from 'cluster';
|
2017-01-23 17:28:14 -08:00
|
|
|
import * as debug from 'debug';
|
2017-11-06 02:59:14 -08:00
|
|
|
import chalk from 'chalk';
|
2018-07-13 07:25:32 -07:00
|
|
|
import * as portscanner from 'portscanner';
|
2017-01-02 12:49:37 -08:00
|
|
|
import isRoot = require('is-root');
|
2017-06-08 09:03:54 -07:00
|
|
|
import Xev from 'xev';
|
2018-07-28 01:52:54 -07:00
|
|
|
import * as program from 'commander';
|
2018-11-18 20:39:10 -08:00
|
|
|
import * as sysUtils from 'systeminformation';
|
2018-11-10 21:27:00 -08:00
|
|
|
import mongo, { nativeDbConn } from './db/mongodb';
|
2017-04-04 17:58:29 -07:00
|
|
|
|
2018-07-07 03:19:00 -07:00
|
|
|
import Logger from './misc/logger';
|
2018-06-10 14:48:25 -07:00
|
|
|
import serverStats from './daemons/server-stats';
|
|
|
|
import notesStats from './daemons/notes-stats';
|
2018-04-01 21:15:53 -07:00
|
|
|
import loadConfig from './config/load';
|
|
|
|
import { Config } from './config/types';
|
2018-11-10 21:27:00 -08:00
|
|
|
import { lessThan } from './prelude/array';
|
2017-01-16 15:19:34 -08:00
|
|
|
|
2017-01-23 17:28:14 -08:00
|
|
|
const clusterLog = debug('misskey:cluster');
|
2017-06-08 09:03:54 -07:00
|
|
|
const ev = new Xev();
|
2017-01-23 17:28:14 -08:00
|
|
|
|
2018-08-16 11:51:42 -07:00
|
|
|
if (process.env.NODE_ENV != 'production' && process.env.DEBUG == null) {
|
2018-07-28 02:01:49 -07:00
|
|
|
debug.enable('misskey');
|
2018-04-05 02:08:51 -07:00
|
|
|
}
|
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
const pkg = require('../package.json');
|
|
|
|
|
2018-07-28 01:57:24 -07:00
|
|
|
//#region Command line argument definitions
|
2018-07-28 01:52:54 -07:00
|
|
|
program
|
|
|
|
.version(pkg.version)
|
|
|
|
.option('--no-daemons', 'Disable daemon processes (for debbuging)')
|
|
|
|
.option('--disable-clustering', 'Disable clustering')
|
|
|
|
.parse(process.argv);
|
2018-07-28 01:57:24 -07:00
|
|
|
//#endregion
|
2018-07-28 01:52:54 -07:00
|
|
|
|
2016-12-28 14:49:51 -08:00
|
|
|
/**
|
2017-02-26 23:11:49 -08:00
|
|
|
* Init process
|
2016-12-28 14:49:51 -08:00
|
|
|
*/
|
2017-01-23 17:28:14 -08:00
|
|
|
function main() {
|
2018-08-15 08:13:24 -07:00
|
|
|
process.title = `Misskey (${cluster.isMaster ? 'master' : 'worker'})`;
|
2018-07-28 01:57:24 -07:00
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
if (cluster.isMaster || program.disableClustering) {
|
2018-07-07 03:19:00 -07:00
|
|
|
masterMain();
|
2017-06-08 09:03:54 -07:00
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
if (cluster.isMaster) {
|
|
|
|
ev.mount();
|
|
|
|
}
|
|
|
|
|
2018-07-28 13:34:08 -07:00
|
|
|
if (program.daemons) {
|
2018-07-28 01:52:54 -07:00
|
|
|
serverStats();
|
|
|
|
notesStats();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cluster.isWorker || program.disableClustering) {
|
2018-07-07 03:19:00 -07:00
|
|
|
workerMain();
|
2016-12-28 14:49:51 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2017-02-26 23:11:49 -08:00
|
|
|
* Init master process
|
2016-12-28 14:49:51 -08:00
|
|
|
*/
|
2018-07-07 03:19:00 -07:00
|
|
|
async function masterMain() {
|
2017-04-04 17:58:29 -07:00
|
|
|
let config: Config;
|
2016-12-28 14:49:51 -08:00
|
|
|
|
|
|
|
try {
|
|
|
|
// initialize app
|
2017-04-04 17:58:29 -07:00
|
|
|
config = await init();
|
2016-12-28 14:49:51 -08:00
|
|
|
} catch (e) {
|
|
|
|
console.error(e);
|
2018-07-14 04:58:21 -07:00
|
|
|
Logger.error('Fatal error occurred during initialization');
|
2017-04-22 23:40:13 -07:00
|
|
|
process.exit(1);
|
2017-04-04 17:58:29 -07:00
|
|
|
}
|
|
|
|
|
2018-07-14 06:05:19 -07:00
|
|
|
Logger.succ('Misskey initialized');
|
2017-04-04 17:58:29 -07:00
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
if (!program.disableClustering) {
|
|
|
|
await spawnWorkers(config.clusterLimit);
|
|
|
|
}
|
|
|
|
|
2018-11-05 10:48:23 -08:00
|
|
|
Logger.succ(`Now listening on port ${config.port} on ${config.url}`);
|
2016-12-28 14:49:51 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2017-02-26 23:11:49 -08:00
|
|
|
* Init worker process
|
2016-12-28 14:49:51 -08:00
|
|
|
*/
|
2018-07-07 03:19:00 -07:00
|
|
|
async function workerMain() {
|
|
|
|
// start server
|
|
|
|
await require('./server').default();
|
2018-03-28 09:20:40 -07:00
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
if (cluster.isWorker) {
|
|
|
|
// Send a 'ready' message to parent process
|
|
|
|
process.send('ready');
|
|
|
|
}
|
2016-12-28 14:49:51 -08:00
|
|
|
}
|
|
|
|
|
2018-11-18 20:27:38 -08:00
|
|
|
const runningNodejsVersion = process.version.slice(1).split('.').map(x => parseInt(x, 10));
|
|
|
|
const requiredNodejsVersion = [10, 0, 0];
|
|
|
|
const satisfyNodejsVersion = !lessThan(runningNodejsVersion, requiredNodejsVersion);
|
|
|
|
|
2018-11-19 18:23:32 -08:00
|
|
|
function isWellKnownPort(port: number): boolean {
|
|
|
|
return port < 1024;
|
|
|
|
}
|
|
|
|
|
|
|
|
async function isPortAvailable(port: number): Promise<boolean> {
|
|
|
|
return await portscanner.checkPortStatus(port, '127.0.0.1') === 'closed';
|
|
|
|
}
|
|
|
|
|
2018-11-18 20:39:10 -08:00
|
|
|
async function showMachine() {
|
|
|
|
const logger = new Logger('Machine');
|
|
|
|
logger.info(`Hostname: ${os.hostname()}`);
|
|
|
|
logger.info(`Platform: ${process.platform}`);
|
|
|
|
logger.info(`Architecture: ${process.arch}`);
|
|
|
|
logger.info(`CPU: ${os.cpus().length} core`);
|
|
|
|
const mem = await sysUtils.mem();
|
|
|
|
const totalmem = (mem.total / 1024 / 1024 / 1024).toFixed(1);
|
|
|
|
const availmem = (mem.available / 1024 / 1024 / 1024).toFixed(1);
|
|
|
|
logger.info(`MEM: ${totalmem}GB (available: ${availmem}GB)`);
|
|
|
|
}
|
|
|
|
|
|
|
|
function showEnvironment(): void {
|
|
|
|
const env = process.env.NODE_ENV;
|
|
|
|
const logger = new Logger('Env');
|
|
|
|
logger.info(typeof env == 'undefined' ? 'NODE_ENV is not set' : `NODE_ENV: ${env}`);
|
|
|
|
|
|
|
|
if (env !== 'production') {
|
|
|
|
logger.warn('The environment is not in production mode');
|
|
|
|
logger.warn('Do not use for production purpose');
|
|
|
|
}
|
|
|
|
|
|
|
|
logger.info(`You ${isRoot() ? '' : 'do not '}have root privileges`);
|
|
|
|
}
|
|
|
|
|
2016-12-28 14:49:51 -08:00
|
|
|
/**
|
|
|
|
* Init app
|
|
|
|
*/
|
2017-04-04 17:58:29 -07:00
|
|
|
async function init(): Promise<Config> {
|
2016-12-29 06:09:21 -08:00
|
|
|
Logger.info('Welcome to Misskey!');
|
2018-10-09 11:37:51 -07:00
|
|
|
Logger.info(`<<< Misskey v${pkg.version} >>>`);
|
2016-12-28 14:49:51 -08:00
|
|
|
|
2018-11-18 20:27:38 -08:00
|
|
|
new Logger('Nodejs').info(`Version ${runningNodejsVersion.join('.')}`);
|
|
|
|
|
|
|
|
if (!satisfyNodejsVersion) {
|
|
|
|
new Logger('Nodejs').error(`Node.js version is less than ${requiredNodejsVersion.join('.')}. Please upgrade it.`);
|
2018-11-15 05:17:06 -08:00
|
|
|
process.exit(1);
|
|
|
|
}
|
|
|
|
|
2018-11-18 20:39:10 -08:00
|
|
|
await showMachine();
|
|
|
|
showEnvironment();
|
2016-12-30 10:14:38 -08:00
|
|
|
|
2017-05-24 04:50:17 -07:00
|
|
|
const configLogger = new Logger('Config');
|
2018-04-01 21:15:53 -07:00
|
|
|
let config;
|
2016-12-28 14:49:51 -08:00
|
|
|
|
2018-04-01 21:15:53 -07:00
|
|
|
try {
|
|
|
|
config = loadConfig();
|
|
|
|
} catch (exception) {
|
2018-07-14 04:58:21 -07:00
|
|
|
if (typeof exception === 'string') {
|
|
|
|
configLogger.error(exception);
|
|
|
|
process.exit(1);
|
|
|
|
}
|
2018-04-01 21:15:53 -07:00
|
|
|
if (exception.code === 'ENOENT') {
|
2018-07-14 04:58:21 -07:00
|
|
|
configLogger.error('Configuration file not found');
|
|
|
|
process.exit(1);
|
2018-04-01 21:15:53 -07:00
|
|
|
}
|
|
|
|
throw exception;
|
|
|
|
}
|
2017-01-16 15:19:34 -08:00
|
|
|
|
2018-07-14 06:05:19 -07:00
|
|
|
configLogger.succ('Loaded');
|
2016-12-28 14:49:51 -08:00
|
|
|
|
2018-11-15 10:25:35 -08:00
|
|
|
if (config.port == null) {
|
|
|
|
Logger.error('The port is not configured. Please configure port.');
|
|
|
|
process.exit(1);
|
|
|
|
}
|
|
|
|
|
2018-11-19 18:23:32 -08:00
|
|
|
if (process.platform === 'linux' && isWellKnownPort(config.port) && !isRoot()) {
|
|
|
|
Logger.error('You need root privileges to listen on well-known port on Linux');
|
2018-07-13 07:52:28 -07:00
|
|
|
process.exit(1);
|
2016-12-29 08:35:25 -08:00
|
|
|
}
|
|
|
|
|
2018-11-19 18:23:32 -08:00
|
|
|
if (!await isPortAvailable(config.port)) {
|
2018-07-13 08:05:49 -07:00
|
|
|
Logger.error(`Port ${config.port} is already in use`);
|
2018-07-13 07:52:28 -07:00
|
|
|
process.exit(1);
|
2016-12-28 14:49:51 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Try to connect to MongoDB
|
2018-07-28 01:52:54 -07:00
|
|
|
checkMongoDb(config);
|
|
|
|
|
|
|
|
return config;
|
|
|
|
}
|
|
|
|
|
|
|
|
function checkMongoDb(config: Config) {
|
2017-05-24 04:50:17 -07:00
|
|
|
const mongoDBLogger = new Logger('MongoDB');
|
2018-08-15 08:13:24 -07:00
|
|
|
const u = config.mongodb.user ? encodeURIComponent(config.mongodb.user) : null;
|
|
|
|
const p = config.mongodb.pass ? encodeURIComponent(config.mongodb.pass) : null;
|
|
|
|
const uri = `mongodb://${u && p ? `${u}:****@` : ''}${config.mongodb.host}:${config.mongodb.port}/${config.mongodb.db}`;
|
|
|
|
mongoDBLogger.info(`Connecting to ${uri}`);
|
2018-08-21 16:46:31 -07:00
|
|
|
|
|
|
|
mongo.then(() => {
|
2018-11-10 21:27:00 -08:00
|
|
|
nativeDbConn().then(db => db.admin().serverInfo()).then(x => x.version).then((version: string) => {
|
|
|
|
mongoDBLogger.info(`Version: ${version}`);
|
|
|
|
if (lessThan(version.split('.').map(x => parseInt(x, 10)), [3, 6])) {
|
|
|
|
mongoDBLogger.error(`MongoDB version is less than 3.6. Please upgrade it.`);
|
|
|
|
process.exit(1);
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
2018-08-21 16:46:31 -07:00
|
|
|
mongoDBLogger.succ('Connectivity confirmed');
|
|
|
|
})
|
2018-11-10 21:27:00 -08:00
|
|
|
.catch(err => {
|
|
|
|
mongoDBLogger.error(err.message);
|
|
|
|
});
|
2016-12-28 14:49:51 -08:00
|
|
|
}
|
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
function spawnWorkers(limit: number) {
|
2018-11-05 10:48:23 -08:00
|
|
|
Logger.info('Starting workers...');
|
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
return new Promise(res => {
|
|
|
|
// Count the machine's CPUs
|
|
|
|
const cpuCount = os.cpus().length;
|
2016-12-28 14:49:51 -08:00
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
const count = limit || cpuCount;
|
2018-11-05 10:48:23 -08:00
|
|
|
let started = 0;
|
2016-12-28 14:49:51 -08:00
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
// Create a worker for each CPU
|
|
|
|
for (let i = 0; i < count; i++) {
|
|
|
|
const worker = cluster.fork();
|
2018-11-05 10:48:23 -08:00
|
|
|
|
2018-07-28 01:52:54 -07:00
|
|
|
worker.on('message', message => {
|
2018-11-05 10:48:23 -08:00
|
|
|
if (message !== 'ready') return;
|
|
|
|
started++;
|
|
|
|
|
|
|
|
// When all workers started
|
|
|
|
if (started == count) {
|
|
|
|
Logger.succ('All workers started');
|
|
|
|
res();
|
2018-07-28 01:52:54 -07:00
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
2017-08-10 06:06:47 -07:00
|
|
|
});
|
2016-12-28 14:49:51 -08:00
|
|
|
}
|
|
|
|
|
2018-07-28 01:57:24 -07:00
|
|
|
//#region Events
|
|
|
|
|
2017-01-23 17:28:14 -08: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 15:04:17 -08:00
|
|
|
// Display detail of unhandled promise rejection
|
|
|
|
process.on('unhandledRejection', console.dir);
|
|
|
|
|
2018-05-20 03:24:54 -07:00
|
|
|
// Display detail of uncaught exception
|
|
|
|
process.on('uncaughtException', err => {
|
|
|
|
console.error(err);
|
|
|
|
});
|
|
|
|
|
2016-12-28 14:49:51 -08:00
|
|
|
// Dying away...
|
2018-05-20 03:24:54 -07:00
|
|
|
process.on('exit', code => {
|
2018-07-13 08:41:33 -07:00
|
|
|
Logger.info(`The process is going to exit with code ${code}`);
|
2016-12-28 14:49:51 -08:00
|
|
|
});
|
2018-07-28 01:57:24 -07:00
|
|
|
|
|
|
|
//#endregion
|
2018-11-18 19:58:58 -08:00
|
|
|
|
|
|
|
main();
|