mirror of
https://iceshrimp.dev/crimekillz/iceshrimp-161sh.git
synced 2024-11-24 13:09:07 +01:00
90 lines
2.1 KiB
TypeScript
90 lines
2.1 KiB
TypeScript
import cluster from "node:cluster";
|
|
import chalk from "chalk";
|
|
import Xev from "xev";
|
|
|
|
import Logger from "@/services/logger.js";
|
|
import { envOption } from "../env.js";
|
|
|
|
// for typeorm
|
|
import "reflect-metadata";
|
|
import { masterMain } from "./master.js";
|
|
import { workerMain } from "./worker.js";
|
|
import os from "node:os";
|
|
|
|
const logger = new Logger("core", "cyan");
|
|
const clusterLogger = logger.createSubLogger("cluster", "orange", false);
|
|
const ev = new Xev();
|
|
|
|
/**
|
|
* Init process
|
|
*/
|
|
export default async function () {
|
|
process.title = `Iceshrimp (${cluster.isPrimary ? "master" : "worker"})`;
|
|
|
|
if (cluster.isPrimary || envOption.disableClustering) {
|
|
await masterMain();
|
|
if (cluster.isPrimary) {
|
|
ev.mount();
|
|
}
|
|
}
|
|
|
|
if (cluster.isWorker || envOption.disableClustering) {
|
|
await workerMain();
|
|
}
|
|
|
|
if (cluster.isPrimary) {
|
|
// Leave the master process with a marginally lower priority but not too low.
|
|
os.setPriority(2);
|
|
}
|
|
if (cluster.isWorker) {
|
|
// Set workers to a much lower priority so that the master process will be
|
|
// able to respond to api calls even if the workers gank everything.
|
|
os.setPriority(10);
|
|
}
|
|
|
|
// For when Iceshrimp is started in a child process during unit testing.
|
|
// Otherwise, process.send cannot be used, so start it.
|
|
if (process.send) {
|
|
process.send("ok");
|
|
}
|
|
}
|
|
|
|
//#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", console.dir);
|
|
}
|
|
|
|
// Display detail of uncaught exception
|
|
process.on("uncaughtException", (err) => {
|
|
try {
|
|
logger.error(err);
|
|
} catch {}
|
|
});
|
|
|
|
// Dying away...
|
|
process.on("exit", (code) => {
|
|
logger.info(`The process is going to exit with code ${code}`);
|
|
});
|
|
|
|
//#endregion
|