node: Add max contracts option and value validations

Signed-off-by: Maksim Dimitrov <[email protected]>

Author: dimitrovmaksim

node/src/bin/manager.rs

@@ -565,8 +565,11 @@ pub enum CallCacheCommand {
         #[clap(long, conflicts_with_all = &["from", "to"])]
         remove_entire_cache: bool,
         /// Remove the cache for contracts that have not been accessed in the last <TTL_DAYS> days
-        #[clap(long, conflicts_with_all = &["from", "to", "remove-entire-cache"])]
+        #[clap(long, conflicts_with_all = &["from", "to", "remove-entire-cache"], value_parser = clap::value_parser!(i32).range(1..))]
         ttl_days: Option<i32>,
+        /// Limits the number of contracts to consider for cache removal when using --ttl_days
+        #[clap(long, conflicts_with_all = &["remove-entire-cache", "to", "from"], requires = "ttl_days", value_parser = clap::value_parser!(i64).range(1..))]
+        ttl_max_contracts: Option<i64>,
         /// Starting block number
         #[clap(long, short, conflicts_with = "remove-entire-cache", requires = "to")]
         from: Option<i32>,
@@ -1477,12 +1480,14 @@ async fn main() -> anyhow::Result<()> {
                             to,
                             remove_entire_cache,
                             ttl_days,
+                            ttl_max_contracts,
                         } => {
                             let chain_store = ctx.chain_store(&chain_name)?;
                             if let Some(ttl_days) = ttl_days {
                                 return commands::chain::clear_stale_call_cache(
                                     chain_store,
                                     ttl_days,
+                                    ttl_max_contracts,
                                 )
                                 .await;
                             }

node/src/manager/commands/chain.rs

@@ -84,12 +84,15 @@ pub async fn clear_call_cache(
 pub async fn clear_stale_call_cache(
     chain_store: Arc<ChainStore>,
     ttl_days: i32,
+    ttl_max_contracts: Option<i64>,
 ) -> Result<(), Error> {
     println!(
         "Removing stale entries from the call cache for `{}`",
         chain_store.chain
     );
-    chain_store.clear_stale_call_cache(ttl_days).await?;
+    chain_store
+        .clear_stale_call_cache(ttl_days, ttl_max_contracts)
+        .await?;
     Ok(())
 }