Add metrics for mempool size changes (#220)

sei-protocol · Apr 16, 2024 · 4b01444 · 4b01444
1 parent 23751c8
commit 4b01444
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/internal/mempool/mempool.go b/internal/mempool/mempool.go
@@ -889,6 +889,7 @@ func (txmp *TxMempool) insertTx(wtx *WrappedTx, updatePriorityIndex bool) bool {
 	gossipEl := txmp.gossipIndex.PushBack(wtx)
 	wtx.gossipEl = gossipEl
 
+	txmp.metrics.InsertedTxs.Add(1)
 	atomic.AddInt64(&txmp.sizeBytes, int64(wtx.Size()))
 	return true
 }
@@ -911,6 +912,7 @@ func (txmp *TxMempool) removeTx(wtx *WrappedTx, removeFromCache bool, shouldReen
 	txmp.gossipIndex.Remove(wtx.gossipEl)
 	wtx.gossipEl.DetachPrev()
 
+	txmp.metrics.RemovedTxs.Add(1)
 	atomic.AddInt64(&txmp.sizeBytes, int64(-wtx.Size()))
 
 	wtx.removeHandler(removeFromCache)

diff --git a/internal/mempool/metrics.gen.go b/internal/mempool/metrics.gen.go
diff --git a/internal/mempool/metrics.go b/internal/mempool/metrics.go
@@ -49,4 +49,10 @@ type Metrics struct {
 
 	// Number of times transactions are rechecked in the mempool.
 	RecheckTimes metrics.Counter
+
+	// Number of removed tx from mempool
+	RemovedTxs metrics.Counter
+
+	// Number of txs inserted to mempool
+	InsertedTxs metrics.Counter
 }