From 9561f12172b40dcdbee8e7ded081d5bc311d110c Mon Sep 17 00:00:00 2001
From: Lorenzo Delgado <lorenzo@status.im>
Date: Fri, 9 Sep 2022 16:31:33 +0200
Subject: [PATCH] feat(store): add support for getting stored messages count

---
 waku/v2/node/storage/message/message_store.nim             | 2 ++
 waku/v2/node/storage/message/sqlite_store/sqlite_store.nim | 2 ++
 waku/v2/node/storage/message/waku_store_queue.nim          | 3 +++
 3 files changed, 7 insertions(+)

diff --git a/waku/v2/node/storage/message/message_store.nim b/waku/v2/node/storage/message/message_store.nim
index 99c3ff434..a970c250f 100644
--- a/waku/v2/node/storage/message/message_store.nim
+++ b/waku/v2/node/storage/message/message_store.nim
@@ -45,3 +45,5 @@ method getMessagesByHistoryQuery*(
   maxPageSize = StoreMaxPageSize,
   ascendingOrder = true
 ): MessageStoreResult[MessageStorePage] {.base.} = discard
+
+method getMessagesCount*(ms: MessageStore): int64 {.base.} = discard
diff --git a/waku/v2/node/storage/message/sqlite_store/sqlite_store.nim b/waku/v2/node/storage/message/sqlite_store/sqlite_store.nim
index 53f86c661..2154bfd6a 100644
--- a/waku/v2/node/storage/message/sqlite_store/sqlite_store.nim
+++ b/waku/v2/node/storage/message/sqlite_store/sqlite_store.nim
@@ -156,6 +156,8 @@ method getMessagesByHistoryQuery*(
 
   ok((messages, some(pagingInfo)))
 
+method getMessagesCount*(s: SqliteStore): int64 =
+  int64(s.numMessages)
 
 proc close*(s: SqliteStore) = 
   ## Close the database connection
diff --git a/waku/v2/node/storage/message/waku_store_queue.nim b/waku/v2/node/storage/message/waku_store_queue.nim
index 20d2460c5..266168d0e 100644
--- a/waku/v2/node/storage/message/waku_store_queue.nim
+++ b/waku/v2/node/storage/message/waku_store_queue.nim
@@ -425,3 +425,6 @@ method getMessagesByHistoryQuery*(
     return ok((messages, none(PagingInfo)))
   
   ok((messages, some(pagingInfo)))
+
+method getMessagesCount*(storeQueue: StoreQueueRef): int64 =
+  int64(storeQueue.len())