diff options
Diffstat (limited to 'src/bundle/start.ts')
-rw-r--r-- | src/bundle/start.ts | 96 |
1 files changed, 96 insertions, 0 deletions
diff --git a/src/bundle/start.ts b/src/bundle/start.ts new file mode 100644 index 00000000..2a1e6520 --- /dev/null +++ b/src/bundle/start.ts @@ -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"); +} |