From e44ca0617364747ef254a786074f789c362371ed Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Wed, 4 Dec 2024 21:36:53 +0530
Subject: [PATCH] MINOR: Add Try/Except to store procedure yielding (#18883)

---
 .../database/stored_procedures_mixin.py       | 28 +++++++++++++------
 1 file changed, 20 insertions(+), 8 deletions(-)

diff --git a/ingestion/src/metadata/ingestion/source/database/stored_procedures_mixin.py b/ingestion/src/metadata/ingestion/source/database/stored_procedures_mixin.py
index 5b2aa6119c1..5473f8b1700 100644
--- a/ingestion/src/metadata/ingestion/source/database/stored_procedures_mixin.py
+++ b/ingestion/src/metadata/ingestion/source/database/stored_procedures_mixin.py
@@ -217,14 +217,26 @@ class StoredProcedureLineageMixin(ABC):
         self, procedure_and_query: ProcedureAndQuery
     ) -> Iterable[Either[Union[AddLineageRequest, CreateQueryRequest]]]:
 
-        yield from self._yield_procedure_lineage(
-            query_by_procedure=procedure_and_query.query_by_procedure,
-            procedure=procedure_and_query.procedure,
-        )
-        yield from self.yield_procedure_query(
-            query_by_procedure=procedure_and_query.query_by_procedure,
-            procedure=procedure_and_query.procedure,
-        )
+        try:
+            yield from self._yield_procedure_lineage(
+                query_by_procedure=procedure_and_query.query_by_procedure,
+                procedure=procedure_and_query.procedure,
+            )
+        except Exception as exc:
+            logger.debug(traceback.format_exc())
+            logger.warning(
+                f"Could not get lineage for store procedure '{procedure_and_query.procedure.fullyQualifiedName}' due to [{exc}]."
+            )
+        try:
+            yield from self.yield_procedure_query(
+                query_by_procedure=procedure_and_query.query_by_procedure,
+                procedure=procedure_and_query.procedure,
+            )
+        except Exception as exc:
+            logger.debug(traceback.format_exc())
+            logger.warning(
+                f"Could not get query for store procedure '{procedure_and_query.procedure.fullyQualifiedName}' due to [{exc}]."
+            )
 
     def procedure_lineage_generator(self) -> Iterable[ProcedureAndQuery]:
         query = {