add cache for file reader

Author: bxf12315

src/restapi.rs

@@ -1,12 +1,43 @@
 use crate::utils::DisplayVec;
 use goose::goose::{GooseUser, TransactionError, TransactionResult};
 use serde_json::json;
+use std::collections::HashMap;
 use std::sync::{
     Arc,
     atomic::{AtomicUsize, Ordering},
 };
+use tokio::sync::{OnceCell, RwLock};
 use urlencoding::encode;
 
+/// Cache for advisory files
+static FILE_CACHE: OnceCell<RwLock<HashMap<String, Vec<u8>>>> = OnceCell::const_new();
+
+/// Get advisory file cache
+async fn get_file_cache() -> &'static RwLock<HashMap<String, Vec<u8>>> {
+    FILE_CACHE
+        .get_or_init(|| async { RwLock::new(HashMap::new()) })
+        .await
+}
+
+/// Upload advisory data and extract advisory ID from response
+async fn upload_advisory_and_extract_id(
+    file_bytes: Vec<u8>,
+    user: &mut GooseUser,
+) -> Result<String, Box<TransactionError>> {
+    let response = user.post("/api/v2/advisory", file_bytes).await?;
+    let v = response.response?.json::<serde_json::Value>().await?;
+    let advisory_id = v["id"]
+        .as_str()
+        .ok_or_else(|| {
+            Box::new(TransactionError::Custom(
+                "Missing advisory ID in response".to_string(),
+            ))
+        })?
+        .to_string();
+
+    Ok(advisory_id)
+}
+
 pub async fn get_advisory(id: String, user: &mut GooseUser) -> TransactionResult {
     let uri = format!("/api/v2/advisory/{}", encode(&format!("urn:uuid:{}", id)));
 
@@ -43,26 +74,33 @@ pub async fn upload_advisory_and_get_id(
     advisory_file: String,
     user: &mut GooseUser,
 ) -> Result<String, Box<TransactionError>> {
+    // Check cache first
+    let cache = get_file_cache().await;
+
+    {
+        let read_guard = cache.read().await;
+        if let Some(cached_bytes) = read_guard.get(&advisory_file) {
+            let file_bytes = cached_bytes.clone();
+            // Found in cache, upload directly
+            drop(read_guard);
+
+            return upload_advisory_and_extract_id(file_bytes, user).await;
+        }
+    }
+
     let file_bytes = tokio::fs::read(&advisory_file).await.map_err(|e| {
         Box::new(TransactionError::Custom(format!(
             "Failed to read file {}: {}",
             advisory_file, e
         )))
     })?;
 
-    let response = user.post("/api/v2/advisory", file_bytes).await?;
-    let v = response.response?.json::<serde_json::Value>().await?;
-
-    let advisory_id = v["id"]
-        .as_str()
-        .ok_or_else(|| {
-            Box::new(TransactionError::Custom(
-                "Missing advisory ID in response".to_string(),
-            ))
-        })?
-        .to_string();
+    {
+        let mut write_guard = cache.write().await;
+        write_guard.insert(advisory_file.clone(), file_bytes.clone());
+    }
 
-    Ok(advisory_id)
+    upload_advisory_and_extract_id(file_bytes, user).await
 }
 
 /// Delete advisory by ID