Add a feature to randomly retrieve IDs from the ID pool.

Author: bxf12315

scenarios/full-20250604.json5

@@ -114,6 +114,48 @@
   ],
   "download_advisory": "24ae57c3-4b57-4f4e-82c1-83ae26059a89",
   "get_advisory": "24ae57c3-4b57-4f4e-82c1-83ae26059a89",
-  "put_advisory_lables": "urn:uuid:24ae57c3-4b57-4f4e-82c1-83ae26059a89",
-  "patch_advisory_lables": "urn:uuid:9ffb1a1b-14e0-4abb-943a-76906daa54aa"
+  "put_advisory_lables": [
+    "urn:uuid:24ae57c3-4b57-4f4e-82c1-83ae26059a89",
+    "urn:uuid:9ffb1a1b-14e0-4abb-943a-76906daa54aa",
+    "urn:uuid:7c932cc3-aa83-4bb4-bbaf-6030a198e271",
+    "urn:uuid:cac4862f-c87a-4a7d-8d67-f2e0db8f0273",
+    "urn:uuid:f80b5b7e-f2eb-4921-b254-a4873d69a9e9",
+    "urn:uuid:858a53a2-347a-491e-a86a-e24b9a4c1436",
+    "urn:uuid:15637501-2a0b-4076-aec3-d92fa2b959bf",
+    "urn:uuid:5364258d-418e-4cd8-add7-248207c4836f",
+    "urn:uuid:5066569d-c5a3-4d90-8cdc-d8680ee52db0",
+    "urn:uuid:869b1646-2f27-4b7a-b4e0-f6783651135d",
+    "urn:uuid:04313720-b5eb-40bc-b545-1f813f6e53b9",
+    "urn:uuid:2c6d41c1-f39d-47df-b37f-12b08af41577",
+    "urn:uuid:99bac490-3ff3-4c10-8466-2cc4947ac29c",
+    "urn:uuid:90a53b25-118d-4c2a-9b8d-a38ac1a85ab0",
+    "urn:uuid:4d3fd5be-eeb5-46f7-a55d-871be40c9837",
+    "urn:uuid:fd88f7cd-8b9e-4bba-b92b-726732545e82",
+    "urn:uuid:06a769a4-1f30-4fe5-9113-72080301e8ed",
+    "urn:uuid:871263d1-e7c3-4f4e-98b2-04aedeae048c",
+    "urn:uuid:35dc833f-fbb7-43f0-852a-9db14cd3a0f4",
+    "urn:uuid:acfbea2f-08d0-4f2f-aa22-12e635956dce"
+  ],
+  "patch_advisory_lables": [
+    "urn:uuid:8a28d2fc-3e91-4989-add8-b9dd78fcf37d",
+    "urn:uuid:9e3eee98-a032-424d-9d0d-1645faca9d24",
+    "urn:uuid:66dde1a5-9575-46a4-8f7a-f6ab9904f0d1",
+    "urn:uuid:2070a289-c6bb-4a6c-bdb2-6fc951743e55",
+    "urn:uuid:59751bb1-b205-4ea4-9832-a48ab69e39e6",
+    "urn:uuid:0dd5f452-af2b-4740-905e-a00f22fffa7e",
+    "urn:uuid:08467200-8c82-4d38-9920-3cbf6c2d0568",
+    "urn:uuid:be6b4acf-fe7b-46a2-9f4c-fdd497d43188",
+    "urn:uuid:6ddb2cc0-bb10-4f23-96bf-cc409263d80d",
+    "urn:uuid:3973cb57-325a-4574-a60c-1910a30b44ac",
+    "urn:uuid:074881e9-d328-402c-8bd8-13df42a590e5",
+    "urn:uuid:2e8008f1-e3d2-401d-8ef6-87bad7d328f0",
+    "urn:uuid:d30949a9-1f9b-4005-93a8-288a3578e7f7",
+    "urn:uuid:8aa4555d-3723-4c4f-aee8-69c4007ec5fd",
+    "urn:uuid:6076bbcb-b352-4bcb-b2e3-8889e0310724",
+    "urn:uuid:5390d139-c00c-4d3a-858a-833d4b571d9a",
+    "urn:uuid:f1479e4b-839e-4d53-929c-d08b6b907430",
+    "urn:uuid:96c1b100-ec62-46c0-9156-819c37bd0445",
+    "urn:uuid:148db4ee-55ca-4292-b55d-a0c0a9d83b9c",
+    "urn:uuid:e98e73ab-4cfe-4861-8b46-06df5a84198f"
+  ]
 }

src/main.rs

@@ -192,10 +192,20 @@ async fn main() -> Result<(), anyhow::Error> {
             tx!(s.post_vulnerability_analyze?(scenario.analyze_purl.clone()));
             tx!(s.get_purl_details?(scenario.get_purl_details.clone()));
             tx!(s.get_recommendations?(scenario.get_recommendations.clone()));
-            tx!(s.put_advisory_labels?(scenario.put_advisory_lables.clone()));
-            tx!(s.patch_advisory_labels?(
-                scenario.patch_advisory_lables.clone()
-            ));
+
+            // Register put Advisory labels transaction if pool is available
+            let put_advisory_labels_counter = Arc::new(std::sync::atomic::AtomicUsize::new(0));
+            if let Some(_advisory_ids) = scenario.put_advisory_lables.clone() {
+                tx!(s.put_advisory_labels?(scenario.put_advisory_lables.clone(),
+                 put_advisory_labels_counter.clone()),
+                 name: format!("put_advisory_labels_counter"));
+            }
+
+            // Register patch Advisory labels transaction if pool is available
+            let patch_advisory_labels_counter = Arc::new(std::sync::atomic::AtomicUsize::new(0));
+            if let Some(_advisory_ids) = scenario.patch_advisory_lables.clone() {
+                tx!(s.patch_advisory_labels?(scenario.patch_advisory_lables.clone(), patch_advisory_labels_counter.clone()),name: format!("patch_advisory_labels_counter"));
+            }
 
             tx!(s.download_advisory?(scenario.download_advisory.clone()));
             tx!(s.get_advisory?(scenario.get_advisory.clone()));
@@ -228,7 +238,8 @@ async fn main() -> Result<(), anyhow::Error> {
                 tx!(s.delete_sbom_from_pool_sequential?(
                             scenario.delete_sbom_pool.clone(),
                             delete_counter.clone()
-                        ), name: format!("delete_sbom_from_pool_sequential[{} SBOMs]", pool.len()))
+                        ),
+                         name: format!("delete_sbom_from_pool_sequential[{} SBOMs]", pool.len()))
             }
             s
         })

src/restapi.rs

@@ -97,9 +97,13 @@ async fn send_advisory_label_request(
     Ok(())
 }
 
-pub async fn put_advisory_labels(advisory_id: String, user: &mut GooseUser) -> TransactionResult {
+pub async fn put_advisory_labels(
+    advisory_ids: Vec<String>,
+    counter: Arc<AtomicUsize>,
+    user: &mut GooseUser,
+) -> TransactionResult {
     send_advisory_label_request(
-        advisory_id,
+        advisory_ids[counter.fetch_add(1, Ordering::Relaxed) % advisory_ids.len()].clone(),
         user,
         GooseMethod::Put,
         "It's a put request",
@@ -108,9 +112,13 @@ pub async fn put_advisory_labels(advisory_id: String, user: &mut GooseUser) -> T
     .await
 }
 
-pub async fn patch_advisory_labels(advisory_id: String, user: &mut GooseUser) -> TransactionResult {
+pub async fn patch_advisory_labels(
+    advisory_ids: Vec<String>,
+    counter: Arc<AtomicUsize>,
+    user: &mut GooseUser,
+) -> TransactionResult {
     send_advisory_label_request(
-        advisory_id,
+        advisory_ids[counter.fetch_add(1, Ordering::Relaxed) % advisory_ids.len()].clone(),
         user,
         GooseMethod::Patch,
         "It's a patch request",

src/scenario/mod.rs

@@ -103,10 +103,11 @@ pub(crate) struct Scenario {
     #[serde(with = "required")]
     pub get_advisory: Option<String>,
 
-    pub put_advisory_lables: Option<String>,
+    #[serde(default, skip_serializing_if = "Option::is_none")]
+    pub put_advisory_lables: Option<Vec<String>>,
 
-    #[serde(with = "required")]
-    pub patch_advisory_lables: Option<String>,
+    #[serde(default, skip_serializing_if = "Option::is_none")]
+    pub patch_advisory_lables: Option<Vec<String>>,
 }
 
 impl Scenario {
@@ -149,11 +150,22 @@ impl Scenario {
         let download_advisory = Some(loader.download_advisory().await?);
         let get_advisory = Some(loader.download_advisory().await?);
 
-        let put_advisory_lables = Some(format!("urn:uuid:{}", loader.put_advisory_lables().await?));
-        let patch_advisory_lables = Some(format!(
-            "urn:uuid:{}",
-            loader.patch_advisory_lables().await?
-        ));
+        let put_advisory_lables = Some(
+            loader
+                .put_advisory_lables()
+                .await?
+                .iter()
+                .map(|sbom_id| format!("urn:uuid:{sbom_id}"))
+                .collect(),
+        );
+        let patch_advisory_lables = Some(
+            loader
+                .patch_advisory_lables()
+                .await?
+                .iter()
+                .map(|sbom_id| format!("urn:uuid:{sbom_id}"))
+                .collect(),
+        );
 
         Ok(Self {
             get_sbom: large_sbom_digest.clone(),
@@ -389,27 +401,31 @@ FROM public.advisory order by modified desc limit 1;"#,
         )
         .await
     }
-    
-    /// A advisory ID for put labels
-    pub async fn put_advisory_lables(&self) -> anyhow::Result<String> {
-        self.find(
-            r#"
-SELECT id::text as result
-FROM public.advisory  where labels is not null order by modified desc limit 1;
-"#,
-        )
-        .await
+
+    /// Advisory IDs for put labels
+    pub async fn put_advisory_lables(&self) -> anyhow::Result<Vec<String>> {
+        let mut db = crate::db::connect(&self.db).await?;
+
+        let rows = sqlx::query("SELECT id::text as id FROM public.advisory  where labels is not null order by modified desc limit 20;")
+            .fetch_all(&mut db)
+            .await?;
+        Ok(rows
+            .into_iter()
+            .map(|row| row.get::<String, _>("id"))
+            .collect())
     }
 
-    /// A advisory ID for patch labels
-    pub async fn patch_advisory_lables(&self) -> anyhow::Result<String> {
-        self.find(
-            r#"
-SELECT id::text as result
-FROM public.advisory  where labels is not null order by modified desc OFFSET 1 limit 1;
-"#,
-        )
-        .await
+    /// Advisory IDs for patch labels
+    pub async fn patch_advisory_lables(&self) -> anyhow::Result<Vec<String>> {
+        let mut db = crate::db::connect(&self.db).await?;
+
+        let rows = sqlx::query("SELECT id::text as id FROM public.advisory  where labels is not null order by modified desc OFFSET 20 limit 20;")
+            .fetch_all(&mut db)
+            .await?;
+        Ok(rows
+            .into_iter()
+            .map(|row| row.get::<String, _>("id"))
+            .collect())
     }
 }