fix(blockifier): add caching for native contracts (#2080)

Co-Authored-By: Bohdan Ohorodnii <[email protected]>

crates/blockifier/src/test_utils/contracts.rs

@@ -188,7 +188,8 @@ impl FeatureContract {
     pub fn get_runnable_class(&self) -> RunnableContractClass {
         #[cfg(feature = "cairo_native")]
         if CairoVersion::Native == self.cairo_version() {
-            let native_contract_class = NativeContractClassV1::from_file(&self.get_compiled_path());
+            let native_contract_class =
+                NativeContractClassV1::compile_or_get_cached(&self.get_compiled_path()).into();
             return RunnableContractClass::V1Native(native_contract_class);
         }
 

crates/blockifier/src/test_utils/struct_impls.rs

@@ -1,4 +1,8 @@
+#[cfg(feature = "cairo_native")]
+use std::collections::HashMap;
 use std::sync::Arc;
+#[cfg(feature = "cairo_native")]
+use std::sync::{LazyLock, RwLock};
 
 use cairo_lang_starknet_classes::casm_contract_class::CasmContractClass;
 #[cfg(feature = "cairo_native")]
@@ -238,6 +242,10 @@ impl BouncerWeights {
     }
 }
 
+#[cfg(feature = "cairo_native")]
+static COMPILED_NATIVE_CONTRACT_CACHE: LazyLock<RwLock<HashMap<String, NativeContractClassV1>>> =
+    LazyLock::new(|| RwLock::new(HashMap::new()));
+
 #[cfg(feature = "cairo_native")]
 impl NativeContractClassV1 {
     /// Convenience function to construct a NativeContractClassV1 from a raw contract class.
@@ -271,4 +279,18 @@ impl NativeContractClassV1 {
         let raw_contract_class = get_raw_contract_class(contract_path);
         Self::try_from_json_string(&raw_contract_class)
     }
+
+    /// Compile a contract from a file or get it from the cache.
+    pub fn compile_or_get_cached(path: &str) -> Self {
+        let cache = COMPILED_NATIVE_CONTRACT_CACHE.read().unwrap();
+        if let Some(cached_class) = cache.get(path) {
+            return cached_class.clone();
+        }
+        std::mem::drop(cache);
+
+        let class = NativeContractClassV1::from_file(path);
+        let mut cache = COMPILED_NATIVE_CONTRACT_CACHE.write().unwrap();
+        cache.insert(path.to_string(), class.clone());
+        class
+    }
 }