yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-11-19 14:24:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ingest): add db name to postgres URNs (#2401)

Browse Source
This commit is contained in:
Harshal Sheth 2021-04-14 15:04:13 -07:00 committed by GitHub
parent 2af4603e49
commit 9c2a30c3a1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
metadata-ingestion/src/datahub/ingestion/source/postgres.py
View File

@ -15,6 +15,12 @@ class PostgresConfig(BasicSQLAlchemyConfig):
# defaults # defaults
scheme = "postgresql+psycopg2" scheme = "postgresql+psycopg2"
def get_identifier(self, schema: str, table: str) -> str:
regular = f"{schema}.{table}"
if self.database:
return f"{self.database}.{regular}"
return regular
class PostgresSource(SQLAlchemySource): class PostgresSource(SQLAlchemySource):
def __init__(self, config, ctx): def __init__(self, config, ctx):