feat(status): Add more stats

SideProjectGuys · Dec 17, 2019 · ff878e5 · ff878e5
1 parent 3610038
commit ff878e5
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 14 deletions.
diff --git a/src/client.ts b/src/client.ts
@@ -105,9 +105,14 @@ export class IMClient extends Client {
 	public gatewayConnected: boolean;
 	public activityInterval: NodeJS.Timer;
 	public voiceConnections: LavaPlayerManager;
-	public eventsRaised: number;
-	public warningsRaised: number;
-	public errorsRaised: number;
+	public stats: {
+		wsEvents: number;
+		wsWarnings: number;
+		wsErrors: number;
+		cmdProcessed: number;
+		cmdErrors: number;
+		cmdHttpErrors: Map<number, number>;
+	};
 
 	private counts: {
 		cachedAt: Moment;
@@ -135,7 +140,14 @@ export class IMClient extends Client {
 		});
 
 		this.startedAt = moment();
-		this.eventsRaised = 0;
+		this.stats = {
+			wsEvents: 0,
+			wsWarnings: 0,
+			wsErrors: 0,
+			cmdProcessed: 0,
+			cmdErrors: 0,
+			cmdHttpErrors: new Map()
+		};
 		this.counts = {
 			cachedAt: moment.unix(0),
 			guilds: 0,
@@ -604,15 +616,15 @@ export class IMClient extends Client {
 
 	private async onWarn(warn: string) {
 		console.error('DISCORD WARNING:', warn);
-		this.warningsRaised++;
+		this.stats.wsWarnings++;
 	}
 
 	private async onError(error: Error) {
 		console.error('DISCORD ERROR:', error);
-		this.errorsRaised++;
+		this.stats.wsErrors++;
 	}
 
 	private async onRawWS() {
-		this.eventsRaised++;
+		this.stats.wsEvents++;
 	}
 }
diff --git a/src/framework/services/Commands.ts b/src/framework/services/Commands.ts
@@ -455,6 +455,8 @@ export class CommandsService {
 			i++;
 		}
 
+		this.client.stats.cmdProcessed++;
+
 		// Run command
 		let error: any = null;
 		try {
@@ -467,6 +469,12 @@ export class CommandsService {
 		const execTime = Date.now() - start;
 
 		if (error) {
+			this.client.stats.cmdErrors++;
+			if (error.code) {
+				const num = this.client.stats.cmdHttpErrors.get(error.code) || 0;
+				this.client.stats.cmdHttpErrors.set(error.code, num + 1);
+			}
+
 			console.error(error);
 
 			withScope(scope => {

diff --git a/src/framework/services/RabbitMq.ts b/src/framework/services/RabbitMq.ts
@@ -167,13 +167,14 @@ export class RabbitMqService {
 				connections: this.client.music.getMusicConnectionGuildIds()
 			},
 			cache: this.getCacheSizes(),
-			events: {
-				total: this.client.eventsRaised,
-				warnings: this.client.warningsRaised,
-				errors: this.client.errorsRaised
-			},
-			requests: {
-				queued
+			stats: {
+				wsEvents: this.client.stats.wsEvents,
+				wsWarnings: this.client.stats.wsWarnings,
+				wsErrors: this.client.stats.wsErrors,
+				cmdProcessed: this.client.stats.cmdProcessed,
+				cmdErrors: this.client.stats.cmdErrors,
+				cmdHttpErrors: [...this.client.stats.cmdHttpErrors.entries()].map(([code, count]) => ({ code, count })),
+				httpRequestsQueued: queued
 			}
 		});
 	}