modified delete function and added cli argument

Author: PoulavBhowmick03

beacon_node/src/cli.rs

@@ -817,6 +817,15 @@ pub fn cli_app() -> Command {
                 .action(ArgAction::Set)
                 .display_order(0)
         )
+        .arg(
+            Arg::new("state-cache-max-bytes")
+                .long("state-cache-max-bytes")
+                .value_name("BYTES")
+                .help("Specifies the maximum size of the state cache in bytes")
+                .default_value("536870912")
+                .action(ArgAction::Set)
+                .display_order(0)
+        )
         /*
          * Execution Layer Integration
          */

beacon_node/src/config.rs

@@ -377,6 +377,12 @@ pub fn get_config<E: EthSpec>(
             .map_err(|_| "state-cache-size is not a valid integer".to_string())?;
     }
 
+    if let Some(max_bytes) = cli_args.get_one::<String>("state-cache-max-bytes") {
+        client_config.store.max_state_cache_bytes = max_bytes
+            .parse()
+            .map_err(|_| "state-cache-max-bytes is not a valid integer".to_string())?;
+    }
+
     if let Some(historic_state_cache_size) =
         clap_utils::parse_optional(cli_args, "historic-state-cache-size")?
     {

beacon_node/store/src/state_cache.rs

@@ -340,14 +340,19 @@ impl<E: EthSpec> StateCache<E> {
     }
 
     pub fn delete_state(&mut self, state_root: &Hash256) {
+        self.cached_bytes = self.cached_bytes.saturating_sub(
+            self.states
+                .peek(state_root)
+                .map_or(0, |(_, state)| state.memory_size()),
+        );
         self.states.pop(state_root);
         self.block_map.delete(state_root);
     }
 
     pub fn delete_block_states(&mut self, block_root: &Hash256) {
         if let Some(slot_map) = self.block_map.delete_block_states(block_root) {
             for state_root in slot_map.slots.values() {
-                self.states.pop(state_root);
+                self.delete_state(state_root);
             }
         }
     }
@@ -410,6 +415,11 @@ impl<E: EthSpec> StateCache<E> {
             .collect::<Vec<_>>();
 
         for state_root in &state_roots_to_delete {
+            self.cached_bytes = self.cached_bytes.saturating_sub(
+                self.states
+                    .peek(state_root)
+                    .map_or(0, |(_, state)| state.memory_size()),
+            );
             self.delete_state(state_root);
         }