Bigquery View Lineage Fix (#6129)

This commit is contained in:
Mayur Singal 2022-07-17 23:06:24 +05:30 committed by GitHub
parent 4b5b184177
commit 6b10adc40f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions

View File

@ -29,6 +29,7 @@ from metadata.generated.schema.api.tags.createTag import CreateTagRequest
from metadata.generated.schema.api.tags.createTagCategory import (
CreateTagCategoryRequest,
)
from metadata.generated.schema.entity.data.table import TableType
from metadata.generated.schema.entity.services.connections.database.bigQueryConnection import (
BigQueryConnection,
)
@ -181,7 +182,7 @@ class BigquerySource(CommonDbSourceService):
def get_view_definition(
self, table_type: str, table_name: str, schema_name: str, inspector: Inspector
) -> Optional[str]:
if table_type == "View":
if table_type == TableType.View:
view_definition = ""
try:
view_definition = inspector.get_view_definition(

View File

@ -314,6 +314,7 @@ def get_lineage_by_query(
)
except Exception as err:
logger.debug(str(err))
logger.debug(f"Query : {query}")
logger.warning(f"Ingesting lineage failed")
@ -349,6 +350,6 @@ def get_lineage_via_table_entity(
query=query,
) or []
except Exception as e:
logger.error("Failed to create view lineage")
logger.warn("Failed to create view lineage")
logger.debug(f"Query : {query}")
logger.debug(traceback.format_exc())