From e6593986f1261d0af6dc83a32e2d0ac3af0f8c0d Mon Sep 17 00:00:00 2001
From: Eric Doughty-Papassideris <edoughtypapassideris@linagora.com>
Date: Sun, 8 Dec 2024 22:52:52 +0100
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20back:=20adding=20postgresql=20stati?=
 =?UTF-8?q?stics=20(#762)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../orm/connectors/postgres/postgres.ts        | 18 ++++++++++++++++--
 .../diagnostics/providers/database-service.ts  |  2 +-
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/tdrive/backend/node/src/core/platform/services/database/services/orm/connectors/postgres/postgres.ts b/tdrive/backend/node/src/core/platform/services/database/services/orm/connectors/postgres/postgres.ts
index f2a06f95..12dba80c 100644
--- a/tdrive/backend/node/src/core/platform/services/database/services/orm/connectors/postgres/postgres.ts
+++ b/tdrive/backend/node/src/core/platform/services/database/services/orm/connectors/postgres/postgres.ts
@@ -74,10 +74,24 @@ export class PostgresConnector extends AbstractConnector<PostgresConnectionOptio
     switch (depth) {
       case TServiceDiagnosticDepth.alive:
         return { ok: true, didConnect: await this.ping() };
-      case TServiceDiagnosticDepth.stats_basic:
       case TServiceDiagnosticDepth.stats_track:
+        return {
+          ok: true,
+          db: (
+            await this.client.query("select * from pg_stat_database where datname = $1", [
+              this.options.database,
+            ])
+          ).rows,
+        };
+      case TServiceDiagnosticDepth.stats_basic:
+        return { ok: true, warn: "pgsql_basic_has_nothing_more_than_track" };
       case TServiceDiagnosticDepth.stats_deep:
-        return { ok: true, warn: "unsupported_depth" };
+        return {
+          ok: true,
+          databases: (await this.client.query("select * from pg_stat_database")).rows,
+          tables: (await this.client.query("select * from pg_stat_user_tables")).rows,
+          indexes: (await this.client.query("select * from pg_stat_user_indexes")).rows,
+        };
 
       default:
         throw new Error(`Unexpected TServiceDiagnosticDepth: ${JSON.stringify(depth)}`);
diff --git a/tdrive/backend/node/src/core/platform/services/diagnostics/providers/database-service.ts b/tdrive/backend/node/src/core/platform/services/diagnostics/providers/database-service.ts
index ed403b49..0df00d3c 100644
--- a/tdrive/backend/node/src/core/platform/services/diagnostics/providers/database-service.ts
+++ b/tdrive/backend/node/src/core/platform/services/diagnostics/providers/database-service.ts
@@ -1,4 +1,4 @@
 import diagnostics from "../../../framework/api/diagnostics";
 import globalResolver from "../../../../../services/global-resolver";
 
-export default () => diagnostics.registerServiceProviders("db", () => globalResolver.database, {});
+export default () => diagnostics.registerServiceProviders("db", () => globalResolver.database);