1 files changed, 95 insertions, 0 deletions
diff --git a/packages/backend/src/daemons/ServerStatsService.ts b/packages/backend/src/daemons/ServerStatsService.ts
new file mode 100644
index 0000000000..3a6d408431
--- /dev/null
+++ b/packages/backend/src/daemons/ServerStatsService.ts
@@ -0,0 +1,95 @@
+import { Inject, Injectable } from '@nestjs/common';
+import si from 'systeminformation';
+import Xev from 'xev';
+import * as osUtils from 'os-utils';
+import { DI } from '@/di-symbols.js';
+import type { OnApplicationShutdown } from '@nestjs/common';
+
+const ev = new Xev();
+
+const interval = 2000;
+
+const roundCpu = (num: number) => Math.round(num * 1000) / 1000;
+const round = (num: number) => Math.round(num * 10) / 10;
+
+@Injectable()
+export class ServerStatsService implements OnApplicationShutdown {
+	#intervalId: NodeJS.Timer;
+
+	constructor(
+	) {
+	}
+
+	/**
+	 * Report server stats regularly
+	 */
+	public start(): void {
+		const log = [] as any[];
+
+		ev.on('requestServerStatsLog', x => {
+			ev.emit(`serverStatsLog:${x.id}`, log.slice(0, x.length ?? 50));
+		});
+
+		const tick = async () => {
+			const cpu = await cpuUsage();
+			const memStats = await mem();
+			const netStats = await net();
+			const fsStats = await fs();
+
+			const stats = {
+				cpu: roundCpu(cpu),
+				mem: {
+					used: round(memStats.used - memStats.buffers - memStats.cached),
+					active: round(memStats.active),
+				},
+				net: {
+					rx: round(Math.max(0, netStats.rx_sec)),
+					tx: round(Math.max(0, netStats.tx_sec)),
+				},
+				fs: {
+					r: round(Math.max(0, fsStats.rIO_sec ?? 0)),
+					w: round(Math.max(0, fsStats.wIO_sec ?? 0)),
+				},
+			};
+			ev.emit('serverStats', stats);
+			log.unshift(stats);
+			if (log.length > 200) log.pop();
+		};
+
+		tick();
+
+		this.#intervalId = setInterval(tick, interval);
+	}
+
+	public onApplicationShutdown(signal?: string | undefined) {
+		clearInterval(this.#intervalId);
+	}
+}
+
+// CPU STAT
+function cpuUsage(): Promise<number> {
+	return new Promise((res, rej) => {
+		osUtils.cpuUsage((cpuUsage) => {
+			res(cpuUsage);
+		});
+	});
+}
+
+// MEMORY STAT
+async function mem() {
+	const data = await si.mem();
+	return data;
+}
+
+// NETWORK STAT
+async function net() {
+	const iface = await si.networkInterfaceDefault();
+	const data = await si.networkStats(iface);
+	return data[0];
+}
+
+// FS STAT
+async function fs() {
+	const data = await si.disksIO().catch(() => ({ rIO_sec: 0, wIO_sec: 0 }));
+	return data ?? { rIO_sec: 0, wIO_sec: 0 };
+}