yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-12-27 09:58:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(airflow): respect port parameter if provided (#7945)

Browse Source
This commit is contained in:
Harshal Sheth 2023-05-03 04:58:22 +05:30 committed by GitHub
parent bf86235e26
commit 6833494347
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_provider/hooks/datahub.py
View File

@ -58,6 +58,12 @@ class DatahubRestHook(BaseHook):
host = conn.host
if host is None:
raise AirflowException("host parameter is required")
if conn.port:
if ":" in host:
raise AirflowException(
"host parameter should not contain a port number if the port is specified separately"
)
host = f"{host}:{conn.port}"
password = conn.password
timeout_sec = conn.extra_dejson.get("timeout_sec")
return (host, password, timeout_sec)