From 3f9364d8a483ac4ae527ca6cf5a13f5d5b47ea16 Mon Sep 17 00:00:00 2001 From: Mayur Singal <39544459+ulixius9@users.noreply.github.com> Date: Thu, 26 Jan 2023 21:23:41 +0530 Subject: [PATCH] Fix #9923: Fix usage for tables with same name (#9925) * Fix #9923: Fix usage for tables with same name * Fix #9923: Fix usage for tables with same name * Update ingestion/src/metadata/ingestion/bulksink/metadata_usage.py Co-authored-by: Nahuel --- .../src/metadata/ingestion/bulksink/metadata_usage.py | 2 +- ingestion/src/metadata/ingestion/lineage/sql_lineage.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/ingestion/src/metadata/ingestion/bulksink/metadata_usage.py b/ingestion/src/metadata/ingestion/bulksink/metadata_usage.py index a7bb91f326b..10731b7d049 100644 --- a/ingestion/src/metadata/ingestion/bulksink/metadata_usage.py +++ b/ingestion/src/metadata/ingestion/bulksink/metadata_usage.py @@ -102,7 +102,7 @@ class MetadataUsageBulkSink(BulkSink): self.table_usage_map[table_entity.id.__root__] = { "table_entity": table_entity, "usage_count": table_usage.count, - "sql_queries": table_usage.sqlQueries, + "sql_queries": list(table_usage.sqlQueries or []), "usage_date": table_usage.date, "database": table_usage.databaseName, "database_schema": table_usage.databaseSchema, diff --git a/ingestion/src/metadata/ingestion/lineage/sql_lineage.py b/ingestion/src/metadata/ingestion/lineage/sql_lineage.py index a0b0c765772..6aa38141470 100644 --- a/ingestion/src/metadata/ingestion/lineage/sql_lineage.py +++ b/ingestion/src/metadata/ingestion/lineage/sql_lineage.py @@ -136,8 +136,8 @@ def get_table_entities_from_query( table_entities = search_table_entities( metadata=metadata, service_name=service_name, - database=database_name, - database_schema=database_schema, + database=database_query, + database_schema=schema_query, table=table, ) @@ -147,8 +147,8 @@ def get_table_entities_from_query( table_entities = search_table_entities( metadata=metadata, service_name=service_name, - database=database_query, - database_schema=schema_query, + database=database_name, + database_schema=database_schema, table=table, )