From 7c5b02c5a1998d13aeb62db60df213fd9d8b837a Mon Sep 17 00:00:00 2001
From: Jack Robison <jackrobison@lbry.io>
Date: Tue, 12 Apr 2022 15:39:02 -0400
Subject: [PATCH] mempool metrics

---
 scribe/hub/mempool.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/scribe/hub/mempool.py b/scribe/hub/mempool.py
index e98b632..7422896 100644
--- a/scribe/hub/mempool.py
+++ b/scribe/hub/mempool.py
@@ -4,7 +4,7 @@ import attr
 import typing
 import logging
 from collections import defaultdict
-from prometheus_client import Histogram
+from prometheus_client import Histogram, Gauge
 import rocksdb.errors
 from scribe import PROMETHEUS_NAMESPACE
 from scribe.common import HISTOGRAM_BUCKETS
@@ -39,6 +39,10 @@ mempool_process_time_metric = Histogram(
     "processed_mempool", "Time to process mempool and notify touched addresses",
     namespace=NAMESPACE, buckets=HISTOGRAM_BUCKETS
 )
+mempool_tx_count_metric = Gauge("mempool_tx_count", "Transactions in mempool", namespace=NAMESPACE)
+mempool_touched_address_count_metric = Gauge(
+    "mempool_touched_address_count", "Count of addresses touched by transactions in mempool", namespace=NAMESPACE
+)
 
 
 class HubMemPool:
@@ -134,6 +138,9 @@ class HubMemPool:
             for hashX, value in itertools.chain(tx.prevouts, tx.out_pairs):
                 self.touched_hashXs[hashX].add(tx_hash)
                 touched_hashXs.add(hashX)
+
+        mempool_tx_count_metric.set(len(self.txs))
+        mempool_touched_address_count_metric.set(len(self.touched_hashXs))
         return touched_hashXs
 
     def transaction_summaries(self, hashX):