Added bundle files
This commit is contained in:
parent
809cfe948e
commit
106b176c77
4 changed files with 186 additions and 0 deletions
46
src/bundle/Server.ts
Normal file
46
src/bundle/Server.ts
Normal file
|
@ -0,0 +1,46 @@
|
||||||
|
process.on("unhandledRejection", console.error);
|
||||||
|
process.on("uncaughtException", console.error);
|
||||||
|
|
||||||
|
import http from "http";
|
||||||
|
import * as Api from "@valkyrie/api";
|
||||||
|
import * as Gateway from "@valkyrie/gateway";
|
||||||
|
import { CDNServer } from "@valkyrie/cdn";
|
||||||
|
import express from "express";
|
||||||
|
import { green, bold } from "picocolors";
|
||||||
|
import { Config, initDatabase, Sentry } from "@valkyrie/util";
|
||||||
|
|
||||||
|
const app = express();
|
||||||
|
const server = http.createServer();
|
||||||
|
const port = Number(process.env.PORT) || 3001;
|
||||||
|
const production = process.env.NODE_ENV == "development" ? false : true;
|
||||||
|
server.on("request", app);
|
||||||
|
|
||||||
|
const api = new Api.ValkyrieServer({ server, port, production, app });
|
||||||
|
const cdn = new CDNServer({ server, port, production, app });
|
||||||
|
const gateway = new Gateway.Server({ server, port, production });
|
||||||
|
|
||||||
|
process.on("SIGTERM", async () => {
|
||||||
|
console.log("Shutting down due to SIGTERM");
|
||||||
|
await gateway.stop();
|
||||||
|
await cdn.stop();
|
||||||
|
await api.stop();
|
||||||
|
server.close();
|
||||||
|
Sentry.close();
|
||||||
|
});
|
||||||
|
|
||||||
|
async function main() {
|
||||||
|
await initDatabase();
|
||||||
|
await Config.init();
|
||||||
|
await Sentry.init(app);
|
||||||
|
|
||||||
|
await new Promise((resolve) =>
|
||||||
|
server.listen({ port }, () => resolve(undefined)),
|
||||||
|
);
|
||||||
|
await Promise.all([api.start(), cdn.start(), gateway.start()]);
|
||||||
|
|
||||||
|
Sentry.errorHandler(app);
|
||||||
|
|
||||||
|
console.log(`[Server] ${green(`listening on port ${bold(port)}`)}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
main().catch(console.error);
|
4
src/bundle/index.ts
Normal file
4
src/bundle/index.ts
Normal file
|
@ -0,0 +1,4 @@
|
||||||
|
export * from "@valkyrie/api";
|
||||||
|
export * from "@valkyrie/util";
|
||||||
|
export * from "@valkyrie/gateway";
|
||||||
|
export * from "@valkyrie/cdn";
|
93
src/bundle/start.ts
Normal file
93
src/bundle/start.ts
Normal file
|
@ -0,0 +1,93 @@
|
||||||
|
// 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";
|
||||||
|
|
||||||
|
const cores = process.env.THREADS ? parseInt(process.env.THREADS) : 1;
|
||||||
|
|
||||||
|
function getCommitOrFail() {
|
||||||
|
try {
|
||||||
|
return execSync("git rev-parse HEAD").toString().trim();
|
||||||
|
} catch (e) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (cluster.isPrimary) {
|
||||||
|
const commit = getCommitOrFail();
|
||||||
|
|
||||||
|
console.log(
|
||||||
|
bold(`
|
||||||
|
██╗ ██╗ █████╗ ██╗ ██╗ ██╗██╗ ██╗██████╗ ██╗███████╗ ██████╗██╗ ██╗ █████╗ ████████╗
|
||||||
|
██║ ██║██╔══██╗██║ ██║ ██╔╝╚██╗ ██╔╝██╔══██╗██║██╔════╝██╔════╝██║ ██║██╔══██╗╚══██╔══╝
|
||||||
|
██║ ██║███████║██║ █████╔╝ ╚████╔╝ ██████╔╝██║█████╗ ██║ ███████║███████║ ██║
|
||||||
|
╚██╗ ██╔╝██╔══██║██║ ██╔═██╗ ╚██╔╝ ██╔══██╗██║██╔══╝ ██║ ██╔══██║██╔══██║ ██║
|
||||||
|
╚████╔╝ ██║ ██║███████╗██║ ██╗ ██║ ██║ ██║██║███████╗╚██████╗██║ ██║██║ ██║ ██║
|
||||||
|
╚═══╝ ╚═╝ ╚═╝╚══════╝╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═╝╚═╝╚══════╝ ╚═════╝╚═╝ ╚═╝╚═╝ ╚═╝ ╚═╝
|
||||||
|
|
||||||
|
valkyriechat-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
|
||||||
|
const 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) => {
|
||||||
|
for (const id in cluster.workers) {
|
||||||
|
const worker = cluster.workers[id];
|
||||||
|
if (worker === sender || !worker) continue;
|
||||||
|
worker.send(message);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
cluster.on("exit", (worker) => {
|
||||||
|
console.log(
|
||||||
|
`[Worker] ${red(
|
||||||
|
`died with PID: ${worker.process.pid} , restarting ...`,
|
||||||
|
)}`,
|
||||||
|
);
|
||||||
|
cluster.fork();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
require("./Server");
|
||||||
|
}
|
43
src/bundle/stats.ts
Normal file
43
src/bundle/stats.ts
Normal file
|
@ -0,0 +1,43 @@
|
||||||
|
import os from "os";
|
||||||
|
import osu from "node-os-utils";
|
||||||
|
import { red } from "picocolors";
|
||||||
|
|
||||||
|
export function initStats() {
|
||||||
|
console.log(`[Path] running in ${__dirname}`);
|
||||||
|
try {
|
||||||
|
console.log(`[CPU] ${osu.cpu.model()} Cores x${osu.cpu.count()}`);
|
||||||
|
} catch {
|
||||||
|
console.log("[CPU] Failed to get cpu model!");
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`[System] ${os.platform()} ${os.arch()}`);
|
||||||
|
console.log(`[Process] running with PID: ${process.pid}`);
|
||||||
|
if (process.getuid && process.getuid() === 0) {
|
||||||
|
console.warn(
|
||||||
|
red(
|
||||||
|
`[Process] Warning ValkyrieChat is running as root, this highly discouraged and might expose your system vulnerable to attackers.` +
|
||||||
|
`Please run ValkyrieChat as a user without root privileges.`,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: node-os-utils might have a memory leak, more investigation needed
|
||||||
|
// TODO: doesn't work if spawned with multiple threads
|
||||||
|
// setInterval(async () => {
|
||||||
|
// const [cpuUsed, memory, network] = await Promise.all([
|
||||||
|
// osu.cpu.usage(),
|
||||||
|
// osu.mem.info(),
|
||||||
|
// osu.netstat.inOut(),
|
||||||
|
// ]);
|
||||||
|
// var networkUsage = "";
|
||||||
|
// if (typeof network === "object") {
|
||||||
|
// networkUsage = `| [Network]: in ${network.total.inputMb}mb | out ${network.total.outputMb}mb`;
|
||||||
|
// }
|
||||||
|
|
||||||
|
// console.log(
|
||||||
|
// `[CPU] ${cpuUsed.toPrecision(3)}% | [Memory] ${Math.round(
|
||||||
|
// process.memoryUsage().rss / 1024 / 1024
|
||||||
|
// )}mb/${memory.totalMemMb.toFixed(0)}mb ${networkUsage}`
|
||||||
|
// );
|
||||||
|
// }, 1000 * 60 * 5);
|
||||||
|
}
|
Loading…
Reference in a new issue