From d6e4169faf6510025bdd7406c923e7bcd4fbc1fb Mon Sep 17 00:00:00 2001 From: vanshika18 <52087403+vanshika18@users.noreply.github.com> Date: Thu, 17 Aug 2023 11:21:12 +0530 Subject: [PATCH] Fixed #11519 Support Usage queries contain \n characters (#12676) --- .../src/metadata/ingestion/source/database/usage_source.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ingestion/src/metadata/ingestion/source/database/usage_source.py b/ingestion/src/metadata/ingestion/source/database/usage_source.py index 3f2ac07432c..73e37128296 100644 --- a/ingestion/src/metadata/ingestion/source/database/usage_source.py +++ b/ingestion/src/metadata/ingestion/source/database/usage_source.py @@ -78,6 +78,9 @@ class UsageSource(QueryParserSource, ABC): else: yield from self.yield_table_queries() + def format_query(self, query: str) -> str: + return query.replace("\\n", "\n") + def yield_table_queries(self): """ Given an Engine, iterate over the day range and @@ -103,7 +106,7 @@ class UsageSource(QueryParserSource, ABC): try: queries.append( TableQuery( - query=row["query_text"], + query=self.format_query(row["query_text"]), userName=row["user_name"], startTime=str(row["start_time"]), endTime=str(row["end_time"]),