From 10db78618fb070dfab3e891804a5d04de6168f74 Mon Sep 17 00:00:00 2001
From: Roman Krasiuk <rokrassyuk@gmail.com>
Date: Fri, 30 Jun 2023 16:48:37 +0300
Subject: [PATCH] chore(rpc): add cached bytes metric (#3502)

---
 crates/rpc/rpc/src/eth/cache.rs | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/crates/rpc/rpc/src/eth/cache.rs b/crates/rpc/rpc/src/eth/cache.rs
index af0858cb4b..ffa0e4ec44 100644
--- a/crates/rpc/rpc/src/eth/cache.rs
+++ b/crates/rpc/rpc/src/eth/cache.rs
@@ -315,9 +315,9 @@ where
     }
 
     fn update_cached_metrics(&self) {
-        self.full_block_cache.update_cached_count();
-        self.receipts_cache.update_cached_count();
-        self.evm_env_cache.update_cached_count();
+        self.full_block_cache.update_cached_metrics();
+        self.receipts_cache.update_cached_metrics();
+        self.evm_env_cache.update_cached_metrics();
     }
 }
 
@@ -507,8 +507,9 @@ where
         }
     }
 
-    fn update_cached_count(&self) {
+    fn update_cached_metrics(&self) {
         self.metrics.cached_count.set(self.cache.len() as f64);
+        self.metrics.cached_bytes.set(self.cache.memory_usage() as f64);
     }
 }
 
@@ -580,6 +581,8 @@ where
 struct CacheMetrics {
     /// The number of entities in the cache.
     cached_count: Gauge,
+    /// The memory usage of the cache in bytes.
+    cached_bytes: Gauge,
     /// The number of queued consumers.
     queued_consumers_count: Gauge,
 }