mirror of
https://github.com/spacebarchat/server.git
synced 2026-10-15 19:50:56 +00:00
Refactor to mono-repo + upgrade packages
This commit is contained in:
@@ -0,0 +1,96 @@
|
||||
// process.env.MONGOMS_DEBUG = "true";
|
||||
require('module-alias/register');
|
||||
import "reflect-metadata";
|
||||
import cluster, { Worker } from "cluster";
|
||||
import os from "os";
|
||||
import { red, bold, yellow, cyan } from "picocolors";
|
||||
import { initStats } from "./stats";
|
||||
import { config } from "dotenv";
|
||||
config();
|
||||
import { execSync } from "child_process";
|
||||
|
||||
// TODO: add socket event transmission
|
||||
var cores = 1;
|
||||
try {
|
||||
cores = Number(process.env.THREADS) || os.cpus().length;
|
||||
} catch {
|
||||
console.log("[API] Failed to get thread count! Using 1...");
|
||||
}
|
||||
|
||||
if (cluster.isMaster) {
|
||||
function getCommitOrFail() {
|
||||
try {
|
||||
return execSync("git rev-parse HEAD").toString().trim();
|
||||
} catch (e) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
const commit = getCommitOrFail();
|
||||
|
||||
console.log(
|
||||
bold(`
|
||||
███████╗ ██████╗ ███████╗███████╗ ██████╗ ██████╗ ██████╗ ██████╗
|
||||
██╔════╝██╔═══██╗██╔════╝██╔════╝██╔════╝██╔═══██╗██╔══██╗██╔══██╗
|
||||
█████╗ ██║ ██║███████╗███████╗██║ ██║ ██║██████╔╝██║ ██║
|
||||
██╔══╝ ██║ ██║╚════██║╚════██║██║ ██║ ██║██╔══██╗██║ ██║
|
||||
██║ ╚██████╔╝███████║███████║╚██████╗╚██████╔╝██║ ██║██████╔╝
|
||||
╚═╝ ╚═════╝ ╚══════╝╚══════╝ ╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚═════╝
|
||||
|
||||
fosscord-server | ${yellow(
|
||||
`Pre-release (${commit !== null
|
||||
? commit.slice(0, 7)
|
||||
: "Unknown (Git cannot be found)"
|
||||
})`
|
||||
)}
|
||||
|
||||
Commit Hash: ${commit !== null
|
||||
? `${cyan(commit)} (${yellow(commit.slice(0, 7))})`
|
||||
: "Unknown (Git cannot be found)"
|
||||
}
|
||||
Cores: ${cyan(os.cpus().length)} (Using ${cores} thread(s).)
|
||||
`)
|
||||
);
|
||||
|
||||
if (commit == null) {
|
||||
console.log(yellow(`Warning: Git is not installed or not in PATH.`));
|
||||
}
|
||||
|
||||
initStats();
|
||||
|
||||
console.log(`[Process] starting with ${cores} threads`);
|
||||
|
||||
if (cores === 1) {
|
||||
require("./Server");
|
||||
} else {
|
||||
process.env.EVENT_TRANSMISSION = "process";
|
||||
|
||||
// Fork workers.
|
||||
for (let i = 0; i < cores; i++) {
|
||||
// Delay each worker start if using sqlite database to prevent locking it
|
||||
let delay = process.env.DATABASE?.includes("://") ? 0 : i * 1000;
|
||||
setTimeout(() => {
|
||||
cluster.fork();
|
||||
console.log(`[Process] worker ${cyan(i)} started.`);
|
||||
}, delay);
|
||||
}
|
||||
|
||||
cluster.on("message", (sender: Worker, message: any) => {
|
||||
for (const id in cluster.workers) {
|
||||
const worker = cluster.workers[id];
|
||||
if (worker === sender || !worker) continue;
|
||||
worker.send(message);
|
||||
}
|
||||
});
|
||||
|
||||
cluster.on("exit", (worker: any, code: any, signal: any) => {
|
||||
console.log(
|
||||
`[Worker] ${red(
|
||||
`died with PID: ${worker.process.pid} , restarting ...`
|
||||
)}`
|
||||
);
|
||||
cluster.fork();
|
||||
});
|
||||
}
|
||||
} else {
|
||||
require("./Server");
|
||||
}
|
||||
Reference in New Issue
Block a user