yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-01 05:03:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix DB2 Schema Trailing Whitespaces (#17475)

Browse Source
This commit is contained in:
IceS2 2024-08-20 08:13:14 +02:00 committed by GitHub
parent 3ad096a889
commit ddd8c41864
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ingestion/src/metadata/ingestion/source/database/db2/metadata.py
View File

@ -10,7 +10,7 @@
# limitations under the License. # limitations under the License.
"""Db2 source module""" """Db2 source module"""
import traceback import traceback
from typing import Optional from typing import Iterable, Optional
from ibm_db_sa.base import ischema_names from ibm_db_sa.base import ischema_names
from sqlalchemy.engine.reflection import Inspector from sqlalchemy.engine.reflection import Inspector
@ -52,6 +52,13 @@ class Db2Source(CommonDbSourceService):
) )
return cls(config, metadata) 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 @staticmethod
def get_table_description( def get_table_description(
schema_name: str, table_name: str, inspector: Inspector schema_name: str, table_name: str, inspector: Inspector