yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-10-23 14:54:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Inspector (#15997)

Browse Source
This commit is contained in:
Ayush Shah 2024-04-23 16:32:30 +05:30 committed by GitHub
parent a4b856956d
commit cf914dea94
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ingestion/src/metadata/ingestion/source/database/hive/metadata.py
View File

@ -15,7 +15,6 @@ Hive source methods.
from typing import Optional, Tuple from typing import Optional, Tuple
from pyhive.sqlalchemy_hive import HiveDialect from pyhive.sqlalchemy_hive import HiveDialect
from sqlalchemy.inspection import inspect
from metadata.generated.schema.entity.services.connections.database.hiveConnection import ( from metadata.generated.schema.entity.services.connections.database.hiveConnection import (
HiveConnection, HiveConnection,
@ -95,4 +94,5 @@ class HiveSource(CommonDbSourceService):
self.engine = get_metastore_connection( self.engine = get_metastore_connection(
self.service_connection.metastoreConnection self.service_connection.metastoreConnection
) )
self.inspector = inspect(self.engine) self._connection_map = {} # Lazy init as well
self._inspector_map = {}

5
ingestion/src/metadata/ingestion/source/database/presto/metadata.py
View File

@ -18,7 +18,7 @@ from copy import deepcopy
from typing import Iterable, Optional from typing import Iterable, Optional
from pyhive.sqlalchemy_presto import PrestoDialect, _type_map from pyhive.sqlalchemy_presto import PrestoDialect, _type_map
from sqlalchemy import inspect, types, util from sqlalchemy import types, util
from sqlalchemy.engine import reflection from sqlalchemy.engine import reflection
from metadata.generated.schema.entity.data.database import Database from metadata.generated.schema.entity.data.database import Database
@ -140,7 +140,8 @@ class PrestoSource(CommonDbSourceService):
new_service_connection = deepcopy(self.service_connection) new_service_connection = deepcopy(self.service_connection)
new_service_connection.catalog = database_name new_service_connection.catalog = database_name
self.engine = get_connection(new_service_connection) self.engine = get_connection(new_service_connection)
self.inspector = inspect(self.engine) self._connection_map = {} # Lazy init as well
self._inspector_map = {}
def get_database_names(self) -> Iterable[str]: def get_database_names(self) -> Iterable[str]:
configured_catalog = self.service_connection.catalog configured_catalog = self.service_connection.catalog

5
ingestion/src/metadata/ingestion/source/database/saphana/metadata.py
View File

@ -13,8 +13,6 @@ SAP Hana source module
""" """
from typing import Iterable, Optional from typing import Iterable, Optional
from sqlalchemy import inspect
from metadata.generated.schema.entity.services.connections.database.sapHanaConnection import ( from metadata.generated.schema.entity.services.connections.database.sapHanaConnection import (
SapHanaConnection, SapHanaConnection,
) )
@ -51,7 +49,8 @@ class SaphanaSource(CommonDbSourceService):
""" """
Check if the db is configured, or query the name Check if the db is configured, or query the name
""" """
self.inspector = inspect(self.engine) self._connection_map = {} # Lazy init as well
self._inspector_map = {}
if getattr(self.service_connection.connection, "database"): if getattr(self.service_connection.connection, "database"):
yield self.service_connection.connection.database yield self.service_connection.connection.database