From ddd8c41864e65c2602792c83f044309928ef636b Mon Sep 17 00:00:00 2001 From: IceS2 Date: Tue, 20 Aug 2024 08:13:14 +0200 Subject: [PATCH] Fix DB2 Schema Trailing Whitespaces (#17475) --- .../metadata/ingestion/source/database/db2/metadata.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/ingestion/src/metadata/ingestion/source/database/db2/metadata.py b/ingestion/src/metadata/ingestion/source/database/db2/metadata.py index 0a5911b54a6..794cb3b8590 100644 --- a/ingestion/src/metadata/ingestion/source/database/db2/metadata.py +++ b/ingestion/src/metadata/ingestion/source/database/db2/metadata.py @@ -10,7 +10,7 @@ # limitations under the License. """Db2 source module""" import traceback -from typing import Optional +from typing import Iterable, Optional from ibm_db_sa.base import ischema_names from sqlalchemy.engine.reflection import Inspector @@ -52,6 +52,13 @@ class Db2Source(CommonDbSourceService): ) return cls(config, metadata) + def get_raw_database_schema_names(self) -> Iterable[str]: + if self.service_connection.__dict__.get("databaseSchema"): + yield self.service_connection.databaseSchema + else: + for schema_name in self.inspector.get_schema_names(): + yield schema_name.rstrip() + @staticmethod def get_table_description( schema_name: str, table_name: str, inspector: Inspector