fix: use match and logger

juspay · Jan 16, 2024 · 1ca7d36 · 1ca7d36
1 parent 4783970
commit 1ca7d36
Showing 1 changed file with 20 additions and 14 deletions.
diff --git a/crates/router/src/utils/user/sample_data.rs b/crates/router/src/utils/user/sample_data.rs
@@ -52,7 +52,7 @@ pub async fn generate_sample_data(
 
     let business_label_default = merchant_parsed_details.first().map(|x| x.business.clone());
 
-    let profile_id = crate::core::utils::get_profile_id_from_business_details(
+    let profile_id = match crate::core::utils::get_profile_id_from_business_details(
         business_country_default,
         business_label_default.as_ref(),
         &merchant_from_db,
@@ -61,19 +61,25 @@ pub async fn generate_sample_data(
         false,
     )
     .await
-    .map(|id| id.clone())
-    .unwrap_or(
-        state
-            .store
-            .list_business_profile_by_merchant_id(&merchant_id)
-            .await
-            .change_context(SampleDataError::InternalServerError)
-            .attach_printable("Failed to get business profile")?
-            .first()
-            .ok_or(SampleDataError::InternalServerError)?
-            .profile_id
-            .clone(),
-    );
+    {
+        Ok(id) => id.clone(),
+        Err(_) => {
+            router_env::logger::error!(
+                "Profile ID not found in business details. Attempting to fetch from the database."
+            );
+
+            state
+                .store
+                .list_business_profile_by_merchant_id(&merchant_id)
+                .await
+                .change_context(SampleDataError::InternalServerError)
+                .attach_printable("Failed to get business profile")?
+                .first()
+                .ok_or(SampleDataError::InternalServerError)?
+                .profile_id
+                .clone()
+        }
+    };
 
     // 10 percent payments should be failed
     #[allow(clippy::as_conversions)]