1 files changed, 83 insertions, 0 deletions
diff --git a/src/start.ts b/src/start.ts
new file mode 100644
index 00000000..cf1a42a5
--- /dev/null
+++ b/src/start.ts
@@ -0,0 +1,83 @@
+// process.env.MONGOMS_DEBUG = "true";
+import { execSync } from "child_process";
+import cluster, { Worker } from "cluster";
+import { config } from "dotenv";
+import os from "os";
+import { bold, cyan, red, yellow } from "picocolors";
+import "reflect-metadata";
+import { initStats } from "./stats";
+config();
+
+// TODO: add socket event transmission
+let 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");
+}
|
