From ac8a1187678b8580c2ff26181344d6647fb46303 Mon Sep 17 00:00:00 2001
From: mat <github@matdoes.dev>
Date: Thu, 30 Dec 2021 22:33:55 -0600
Subject: change some metrics

---
 src/database.ts |  4 ++++
 src/metrics.ts  | 25 +++++++++++++++++--------
 2 files changed, 21 insertions(+), 8 deletions(-)

(limited to 'src')

diff --git a/src/database.ts b/src/database.ts
index 9ef812d..571ec5e 100644
--- a/src/database.ts
+++ b/src/database.ts
@@ -807,6 +807,10 @@ export async function fetchServerStatus() {
 	return await database.admin().serverStatus()
 }
 
+export async function fetchServerStats() {
+	return await database.stats()
+}
+
 // make sure it's not in a test
 console.log('global.isTest', globalThis.isTest)
 if (!globalThis.isTest) {
diff --git a/src/metrics.ts b/src/metrics.ts
index 9db3ec3..12f833e 100644
--- a/src/metrics.ts
+++ b/src/metrics.ts
@@ -1,5 +1,5 @@
 import { collectDefaultMetrics, Gauge, register } from 'prom-client'
-import { fetchServerStatus } from './database.js'
+import { fetchServerStats, fetchServerStatus } from './database.js'
 import { getKeyUsage } from './hypixelApi.js'
 export { register } from 'prom-client'
 
@@ -26,8 +26,8 @@ const connectionsCounter = new Gauge({
 		connectionsCounter.set(status.connections.current)
 		networkBytesInCounter.set(status.network.bytesIn)
 		networkBytesOutCounter.set(status.network.bytesOut)
-		queryOpCounter.set(status.opcounters.query)
-		updateOpCounter.set(status.opcounters.update)
+		readCounter.set(status.opLatencies.reads.ops)
+		writeCounter.set(status.opLatencies.writes.ops)
 	}
 })
 const networkBytesInCounter = new Gauge({
@@ -40,15 +40,24 @@ const networkBytesOutCounter = new Gauge({
 	help: 'Number of bytes sent by the database.',
 	registers: [ register ],
 })
-const queryOpCounter = new Gauge({
-	name: 'mongodb_query_op_count',
-	help: 'Number of queries executed by the database.',
+const readCounter = new Gauge({
+	name: 'mongodb_read_ops',
+	help: 'Number of read operations by the database.',
 	registers: [ register ],
 })
-const updateOpCounter = new Gauge({
+const writeCounter = new Gauge({
 	name: 'mongodb_update_op_count',
-	help: 'Number of updates executed by the database.',
+	help: 'Number of write operations by the database.',
 	registers: [ register ],
 })
+const dbSizeCounter = new Gauge({
+	name: 'mongodb_db_size',
+	help: 'Size of the database in bytes.',
+	registers: [ register ],
+	async collect() {
+		let stats = await fetchServerStats()
 
+		dbSizeCounter.set(stats.dataSize + stats.indexSize)
+	}
+})
 
-- 
cgit