yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-12-12 07:48:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

MINOR: Add Try/Except to store procedure yielding (#18883)

Browse Source
This commit is contained in:
Mayur Singal 2024-12-04 21:36:53 +05:30 committed by GitHub
parent fe661a2f49
commit e44ca06173
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
ingestion/src/metadata/ingestion/source/database/stored_procedures_mixin.py
View File

@ -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 = {