From 454b098ac52df75e5d5de5010f9edb42cf8b0d52 Mon Sep 17 00:00:00 2001
From: Ivan FB <128452529+Ivansete-status@users.noreply.github.com>
Date: Mon, 24 Nov 2025 10:16:37 +0100
Subject: [PATCH] new metric in postgres_driver to estimate payload stats
 (#3596)

---
 .../driver/postgres_driver/postgres_driver.nim         | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/waku/waku_archive/driver/postgres_driver/postgres_driver.nim b/waku/waku_archive/driver/postgres_driver/postgres_driver.nim
index 842d7cbc2..9b0e14c84 100644
--- a/waku/waku_archive/driver/postgres_driver/postgres_driver.nim
+++ b/waku/waku_archive/driver/postgres_driver/postgres_driver.nim
@@ -5,6 +5,7 @@ import
   stew/[byteutils, arrayops],
   results,
   chronos,
+  metrics,
   db_connector/[postgres, db_common],
   chronicles
 import
@@ -16,6 +17,9 @@ import
   ./postgres_healthcheck,
   ./partitions_manager
 
+declarePublicGauge postgres_payload_size_bytes,
+  "Payload size in bytes of correctly stored messages"
+
 type PostgresDriver* = ref object of ArchiveDriver
   ## Establish a separate pools for read/write operations
   writeConnPool: PgAsyncPool
@@ -333,7 +337,7 @@ method put*(
     return err("could not put msg in messages table: " & $error)
 
   ## Now add the row to messages_lookup
-  return await s.writeConnPool.runStmt(
+  let ret = await s.writeConnPool.runStmt(
     InsertRowInMessagesLookupStmtName,
     InsertRowInMessagesLookupStmtDefinition,
     @[messageHash, timestamp],
@@ -341,6 +345,10 @@ method put*(
     @[int32(0), int32(0)],
   )
 
+  if ret.isOk():
+    postgres_payload_size_bytes.set(message.payload.len)
+  return ret
+
 method getAllMessages*(
     s: PostgresDriver
 ): Future[ArchiveDriverResult[seq[ArchiveRow]]] {.async.} =